import random
import unittest
import pprint
import os
from serpapi import GoogleSearch

class TestLocationApi(unittest.TestCase):

    def setUp(self):
        GoogleSearch.SERP_API_KEY = os.getenv("API_KEY","demo")

    def test_get_location(self):
        search = GoogleSearch({"q": None, "async": True})
        location_list = search.get_location("Austin", 3)
        pp = pprint.PrettyPrinter(indent=2)
        pp.pprint(location_list)
        self.assertIsNotNone(location_list[0].get("id"))

if __name__ == '__main__':
    unittest.main()