Add item_count parameter functionality to search

postalservice/frilservice.py

@@ -25,14 +25,16 @@
 class FrilService(BaseService):
 
     async def fetch_data_async(self, params: dict) -> httpx.Response:
+        self.item_count = params.get("item_count", 36)
         url = self.get_search_params(params)
         res = await fetch_async(url)
         return res
 
     def fetch_data(self, params: dict) -> httpx.Response:
-            url = self.get_search_params(params)
-            res = httpx.get(url)
-            return res
+        self.item_count = params.get("item_count", 36)
+        url = self.get_search_params(params)
+        res = httpx.get(url)
+        return res
 
     async def parse_response_async(self, response: httpx.Response) -> str:
         soup = bs4.BeautifulSoup(response.text, "lxml")
@@ -50,7 +52,7 @@ def parse_response(self, response: httpx.Response) -> str:
 
     def get_base_details(self, results) -> list:
         cleaned_items_list = []
-        for item in results[:10]:
+        for item in results[:self.item_count]:
             id = item.select_one(".link_search_image")["href"].split("/")[-1]
             temp = {}
             temp["id"] = id

postalservice/mercariservice.py

@@ -37,12 +37,12 @@ async def fetch_data_async(self, params: dict) -> httpx.Response:
             raise TypeError('params must be a dict')
 
         keyword = params.get('keyword')
-
         size = params.get('size')
+        item_count = params.get('item_count')
+
         if size is not None: mapped_size = SIZE_MAP.get(size)
         else: mapped_size = None
 
-        item_count = params.get('item_count')
         url = "https://api.mercari.jp/v2/entities:search"
         searchSessionId = ''.join(random.choice(CHARACTERS) for i in range(32))
         payload = {

tests/baseservicetestclass.py

@@ -20,16 +20,22 @@ def setUpClass(cls):
         cls.logger = logger
 
     def test_fetch_code_200(self):
-
         sparams = SearchParams("comme des garcons")
         res = self.service.fetch_data(sparams.get_dict())
         self.logger.info("Fetched data: %s", res)
 
         # Assert that the status code is 200
         self.assertEqual(res.status_code, 200)
 
-    def test_parse_results(self):
+    def test_fetch_code_200_async(self):
+        sparams = SearchParams("comme des garcons")
+        res = asyncio.run(self.service.fetch_data_async(sparams.get_dict()))
+        self.logger.info("Fetched data: %s", res)
+
+        # Assert that the status code is 200
+        self.assertEqual(res.status_code, 200)
 
+    def test_parse_results(self):
         sparams = SearchParams("comme des garcons")
         res: httpx.Response = self.service.fetch_data(sparams.get_dict())
         print(res)
@@ -40,8 +46,31 @@ def test_parse_results(self):
         # Assert that the count is greater than 0
         self.assertTrue(searchresults.count() > 0)
 
-    def test_search_by_size(self):
+    def test_parse_results_async(self):
+        sparams = SearchParams("comme des garcons")
+        res = asyncio.run(self.service.fetch_data_async(sparams.get_dict()))
+        items = asyncio.run(self.service.parse_response_async(res))
+        searchresults = SearchResults(items)
+        self.logger.info(searchresults)
 
+        # Assert that the count is greater than 0
+        self.assertTrue(searchresults.count() > 0)
+
+    def test_get_5_results(self):
+        sparams = SearchParams("comme des garcons", item_count=5)
+        searchresults = self.service.get_search_results(sparams.get_dict())
+
+        # Assert that the count is 5
+        self.assertEqual(searchresults.count(), 5)
+
+    def test_get_5_results_async(self):
+        sparams = SearchParams("comme des garcons", item_count=5)
+        searchresults = asyncio.run(self.service.get_search_results_async(sparams.get_dict()))
+
+        # Assert that the count is 5
+        self.assertEqual(searchresults.count(), 5)
+
+    def test_search_by_size(self):
         size_to_search = "XL"
         sparams = SearchParams("comme des garcons", size=size_to_search)
         searchresults = self.service.get_search_results(sparams.get_dict())
@@ -52,7 +81,6 @@ def test_search_by_size(self):
             self.assertTrue(size_to_search in searchresults.get(i)["size"])
 
     def test_search_by_size_async(self):
-
         size_to_search = "XL"
         sparams = SearchParams("comme des garcons", size=size_to_search)
         searchresults = asyncio.run(self.service.get_search_results_async(sparams.get_dict()))