Merge pull request #88 from sergioteula/dev-add-tests

Added basic tests

Author: sergioteula

tests/test_api_get_items.py

@@ -0,0 +1,14 @@
+import unittest
+from unittest import mock
+
+from amazon_paapi import AmazonApi
+from amazon_paapi.helpers import requests
+
+
+class TestGetItems(unittest.TestCase):
+    @mock.patch.object(requests, "get_items_response")
+    def test_get_items(self, mocked_get_items_response):
+        mocked_get_items_response.return_value = []
+        amazon = AmazonApi("key", "secret", "tag", "ES")
+        response = amazon.get_items("ABCDEFGHIJ")
+        self.assertTrue(isinstance(response, list))

tests/test_helpers_arguments.py

@@ -1,17 +1,30 @@
-from amazon_paapi.exceptions import AsinNotFoundException, InvalidArgumentException
+import unittest
+
+from amazon_paapi.errors import AsinNotFoundException, InvalidArgumentException
 from amazon_paapi.helpers.arguments import get_items_ids
-import pytest
 
 
-def test_get_items_ids():
-    amazon_url = 'https://www.amazon.es/gp/product/B07PHPXHQS'
-    assert get_items_ids('B01N5IB20Q') == ['B01N5IB20Q']
-    assert get_items_ids('B01N5IB20Q,B01N5IB20Q,B01N5IB20Q') == ['B01N5IB20Q','B01N5IB20Q','B01N5IB20Q']
-    assert get_items_ids(['B01N5IB20Q','B01N5IB20Q','B01N5IB20Q']) == ['B01N5IB20Q','B01N5IB20Q','B01N5IB20Q']
-    assert get_items_ids(amazon_url) == ['B07PHPXHQS']
-    assert get_items_ids([amazon_url,amazon_url]) == ['B07PHPXHQS', 'B07PHPXHQS']
+class TestHelpersArguments(unittest.TestCase):
+    def test_get_items_ids(self):
+        amazon_url = "https://www.amazon.es/gp/product/B07PHPXHQS"
+        self.assertEqual(get_items_ids("B01N5IB20Q"), ["B01N5IB20Q"])
+        self.assertEqual(
+            get_items_ids("B01N5IB20Q,B01N5IB20Q,B01N5IB20Q"),
+            ["B01N5IB20Q", "B01N5IB20Q", "B01N5IB20Q"],
+        )
+        self.assertEqual(
+            get_items_ids(["B01N5IB20Q", "B01N5IB20Q", "B01N5IB20Q"]),
+            ["B01N5IB20Q", "B01N5IB20Q", "B01N5IB20Q"],
+        )
+        self.assertEqual(get_items_ids(amazon_url), ["B07PHPXHQS"])
+        self.assertEqual(
+            get_items_ids([amazon_url, amazon_url]), ["B07PHPXHQS", "B07PHPXHQS"]
+        )
+
+    def test_get_items_ids_asin_not_found(self):
+        with self.assertRaises(AsinNotFoundException):
+            get_items_ids("https://www.amazon.es/gp/")
 
-    with pytest.raises(AsinNotFoundException):
-        get_items_ids('https://www.amazon.es/gp/')
-    with pytest.raises(InvalidArgumentException):
-        get_items_ids(34)
+    def test_get_items_ids_invalid_argument(self):
+        with self.assertRaises(InvalidArgumentException):
+            get_items_ids(34)

tests/test_helpers_generators.py

@@ -0,0 +1,16 @@
+import unittest
+
+from amazon_paapi.helpers.generators import get_list_chunks
+
+
+class TestHelpersGenerators(unittest.TestCase):
+    def test_get_list_chunks(self):
+        mocked_iterable = get_list_chunks(list(range(15)), 10)
+        chunk = next(mocked_iterable)
+        self.assertEqual(len(chunk), 10)
+        chunk = next(mocked_iterable)
+        self.assertEqual(len(chunk), 5)
+
+        mocked_iterable = get_list_chunks(list(range(5)), 10)
+        chunk = next(mocked_iterable)
+        self.assertEqual(len(chunk), 5)

tests/test_helpers_items.py

@@ -0,0 +1,43 @@
+import unittest
+from operator import imod
+from unittest import mock
+
+from amazon_paapi.helpers.items import sort_items
+
+
+class MockedItem(mock.MagicMock):
+    def __init__(self, asin):
+        super().__init__()
+        self.asin = asin
+
+
+class TestHelpersItems(unittest.TestCase):
+    def setUp(self):
+        self.mocked_items = [
+            MockedItem("B"),
+            MockedItem("C"),
+            MockedItem("A"),
+            MockedItem("D"),
+        ]
+        self.mocked_items_ids = ["B", "A", "D", "C", "E", "A"]
+
+    def test_sort_items(self):
+        sorted_items = sort_items(self.mocked_items, self.mocked_items_ids, False)
+        self.assertEqual(sorted_items[0].asin, "B")
+        self.assertEqual(sorted_items[1].asin, "A")
+        self.assertEqual(sorted_items[2].asin, "D")
+        self.assertEqual(sorted_items[3].asin, "C")
+
+    def test_sort_items_include_unavailable(self):
+        sorted_items = sort_items(self.mocked_items, self.mocked_items_ids, True)
+        self.assertEqual(sorted_items[4].asin, "E")
+        self.assertEqual(len(sorted_items), 6)
+
+    def test_sort_items_not_include_unavailable(self):
+        sorted_items = sort_items(self.mocked_items, self.mocked_items_ids, False)
+        self.assertEqual(sorted_items[4].asin, "A")
+        self.assertEqual(len(sorted_items), 5)
+
+    def test_sort_items_include_repeated(self):
+        sorted_items = sort_items(self.mocked_items, self.mocked_items_ids, True)
+        self.assertEqual(sorted_items[1].asin, sorted_items[5].asin)

tests/test_tools.py

@@ -1,16 +1,35 @@
-from amazon_paapi.exceptions import AsinNotFoundException
+import unittest
+
+from amazon_paapi.errors import AsinNotFoundException
 from amazon_paapi.tools import get_asin
-import pytest
 
 
-def test_get_asin():
-    assert get_asin('B01N5IB20Q') == 'B01N5IB20Q'
-    assert get_asin('https://www.amazon.es/gp/product/B07PHPXHQS') == 'B07PHPXHQS'
-    assert get_asin('https://www.amazon.es/gp/product/B07PHPXHQS?pf_rd_r=3FXDZDV1W6KY83KEE2Z4&pf_rd_p=c6fa5af0-ec7c-40de-8332-fd1421de4244&pd_rd_r=58786171-de0f-4fe1-a2df-ee335d6715ee&pd_rd_w=KND7A&pd_rd_wg=kIr5z&ref_=pd_gw_unk') == 'B07PHPXHQS'
-    assert get_asin('https://www.amazon.es/dp/B07PKW4CKF') == 'B07PKW4CKF'
-    assert get_asin('https://www.amazon.es/dp/B07PKW4CKF?_encoding=UTF8&ref_=pocs_dp_m_sp_multi_c_more_nooffers_B08D1G2XVX') == 'B07PKW4CKF'
+class TestTools(unittest.TestCase):
+    def test_get_asin(self):
+        self.assertEqual(get_asin("B01N5IB20Q"), "B01N5IB20Q")
+        self.assertEqual(
+            get_asin("https://www.amazon.es/gp/product/B07PHPXHQS"), "B07PHPXHQS"
+        )
+        self.assertEqual(
+            get_asin(
+                "https://www.amazon.es/gp/product/B07PHPXHQS?pf_rd_r3FXDZDV1W6KY83KEE2Z"
+                "4&pf_rd_p=c6fa5af0-ec7c-40de-8332-fd1421de4244&pd_rd_r=58786171-de0f-4"
+                "fe1-a2df-ee335d6715ee&pd_rd_w=KND7A&pd_rd_wg=kIr5z&ref_=pd_gw_unk"
+            ),
+            "B07PHPXHQS",
+        )
+        self.assertEqual(get_asin("https://www.amazon.es/dp/B07PKW4CKF"), "B07PKW4CKF")
+        self.assertEqual(
+            get_asin(
+                "https://www.amazon.es/dp/B07PKW4CKF?_encoding=UTF8&ref_=pocs_dp_m_sp_m"
+                "ulti_c_more_nooffers_B08D1G2XVX"
+            ),
+            "B07PKW4CKF",
+        )
+
+    def test_asin_not_found(self):
+        with self.assertRaises(AsinNotFoundException):
+            get_asin("https://www.amazon.es/gp/")
 
-    with pytest.raises(AsinNotFoundException):
-        get_asin('https://www.amazon.es/gp/')
-    with pytest.raises(AsinNotFoundException):
-        get_asin('this is not even a URL')
+        with self.assertRaises(AsinNotFoundException):
+            get_asin("this is not even a URL")