Merge pull request #18 from flightaware/BCK-5495-fh-useragent

flightaware · Feb 10, 2021 · 8804353 · 8804353
2 parents 3fddea1 + 9fc733b
commit 8804353
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/connector/main.py b/connector/main.py
@@ -79,7 +79,7 @@ async def open_connection(
 
 def build_init_cmd(time_mode: str) -> str:
     """Builds the init command based on the environment variables provided in docker-compose"""
-    initiation_command = f"{time_mode} username {USERNAME} password {APIKEY}"
+    initiation_command = f"{time_mode} username {USERNAME} password {APIKEY} useragent firestarter"
     if COMPRESSION != "":
         initiation_command += f" compression {COMPRESSION}"
     if KEEPALIVE != "":

diff --git a/connector/test/test_connector.py b/connector/test/test_connector.py
@@ -75,8 +75,8 @@ def reconnect_after_error(
             self.assertEqual(
                 self.init_cmds,
                 [
-                    b"live username testuser password testapikey keepalive 60\n",
-                    b"live username testuser password testapikey keepalive 60\n",
+                    b"live username testuser password testapikey useragent firestarter keepalive 60\n",
+                    b"live username testuser password testapikey useragent firestarter keepalive 60\n",
                 ],
             )
             # verify expect output to kafka
@@ -86,8 +86,8 @@ def reconnect_after_error(
             self.assertEqual(
                 self.init_cmds,
                 [
-                    b"live username testuser password testapikey keepalive 60\n",
-                    b"pitr 1584126630 username testuser password testapikey keepalive 60\n",
+                    b"live username testuser password testapikey useragent firestarter keepalive 60\n",
+                    b"pitr 1584126630 username testuser password testapikey useragent firestarter keepalive 60\n",
                 ],
             )
             # verify expect output to kafka