Testcase for geocoder user agent

src/scrapers/tools/geocoder.py

@@ -7,14 +7,13 @@
 
 
 class Geocoder:
-
     def __init__(self):
         self.geolocator = Nominatim(user_agent="codefornl-covid19")
 
     def geocode(self):
         db = Db()
-        locationset = db.session.query(InitiativeImport)\
-            .filter(InitiativeImport.location.isnot(None))\
+        locationset = db.session.query(InitiativeImport) \
+            .filter(InitiativeImport.location.isnot(None)) \
             .with_for_update().all()
 
         # Regex voor postcode geschreven ls `9999XX`
@@ -56,10 +55,4 @@ def geocode(self):
                 print("SUCCESS: " + match.address)
             db.session.add(item)
             db.session.commit()
-            time.sleep(1) # Sleep so we don't overstretch the nominatim api
-
-
-
-
-
-
+            time.sleep(1)  # Sleep so we don't overstretch the nominatim api

src/tests/test_scraper_base.py

@@ -7,6 +7,7 @@
 from models.initiatives import InitiativeImportState
 from platformen import Scraper
 from platformen.scraper import ScrapeException
+from tools import Geocoder
 
 
 class TestDatabase(TestCase):
@@ -21,6 +22,14 @@ def test_postgres_from_env(self):
             assert test_db.get_db_url().startswith("postgres")
 
 
+class TestGeocoder(TestCase):
+    def setUp(self):
+        self.geocoder = Geocoder()
+
+    def test_user_agent_set(self):
+        assert self.geocoder.geolocator.headers.get('User-Agent') is not None
+
+
 class TestScraper(TestCase):
 
     def setUp(self):