added assertion to get_html() test

gutoarraes · gutoarraes · commit 299d53baa988 · 2024-06-14T20:20:25.000-04:00
diff --git a/serpapi/serp_api_client.py b/serpapi/serp_api_client.py
@@ -10,8 +10,8 @@ class SerpApiClient(object):
     ```python
     from serpapi import GoogleSearch
     search = SerpApiClient({
-        "q": "Coffee",
-        "location": "Austin,Texas",
+        "q": "Coffee", 
+        "location": "Austin,Texas", 
         "engine": "google",
         "api_key": "<your private key>"
         })
@@ -61,15 +61,11 @@ def get_results(self, path='/search'):
         """
         return self.get_response(path).text
 
-    def get_html(self, path='/search'):
+    def get_html(self, path='/search.html'):
         """Returns:
             Raw HTML search result from Google
         """
-        response = self.get_dictionary() # get search URL with query parameters
-        response = response['search_metadata']['raw_html_file']
-        response = requests.get(response).text
-        # print(response)
-        return response
+        return self.get_response(path).text
 
     def get_json(self):
         """Returns:
@@ -99,7 +95,7 @@ def get_dict(self):
         return self.get_dictionary()
 
     def get_object(self):
-        """Returns:
+        """Returns: 
             Dynamically created python object wrapping the result data structure
         """
         # iterative over response hash
@@ -134,7 +130,7 @@ def make_pyobj(self, name, node):
     def get_search_archive(self, search_id, format = 'json'):
         """Retrieve search result from the Search Archive API
         Parameters:
-            search_id (int): unique identifier for the search provided by metadata.id
+            search_id (int): unique identifier for the search provided by metadata.id 
             format (string): search format: json or html [optional]
         Returns:
             dict|string: search result from the archive
diff --git a/tests/test_google_search.py b/tests/test_google_search.py
@@ -60,6 +60,7 @@ def test_get_html(self):
 				search = GoogleSearch({"q": "Coffee", "location": "Austin,Texas"})
 				data = search.get_html()
 				self.assertGreater(len(data), 10)
+				self.assertIn("<html>", data)
 
 		@unittest.skipIf((os.getenv("API_KEY") == None), "no api_key provided")
 		def test_get_response(self):
