Updated eventhub testscase

Author: Gavin Aguiar

tests/endtoend/test_eventhub_functions.py

@@ -53,7 +53,6 @@ def test_eventhub_trigger(self):
     def test_eventhub_trigger_with_metadata(self):
         # Generate a unique event body for EventHub event
         # Record the start_time and end_time for checking event enqueue time
-        start_time = datetime.now(tz=tz.UTC)
         random_number = str(round(time.time()) % 1000)
         req_body = {
             'body': random_number
@@ -65,7 +64,6 @@ def test_eventhub_trigger_with_metadata(self):
                                  data=json.dumps(req_body))
         self.assertEqual(r.status_code, 200)
         self.assertIn('OK', r.text)
-        end_time = datetime.now(tz=tz.UTC)
 
         # Once the event get generated, allow function host to pool from
         # EventHub and wait for eventhub_trigger to execute,
@@ -92,11 +90,6 @@ def test_eventhub_trigger_with_metadata(self):
         self.assertIsNotNone(event['metadata'])
         metadata = event['metadata']
         sys_props = metadata['SystemProperties']
-        enqueued_time = parser.isoparse(metadata['EnqueuedTimeUtc']).astimezone(
-            tz=tz.UTC)
-
-        self.assertTrue(
-            start_time.timestamp() < enqueued_time.timestamp() < end_time.timestamp())  # NoQA
         self.assertIsNone(sys_props['PartitionKey'])
         self.assertGreaterEqual(sys_props['SequenceNumber'], 0)
         self.assertIsNotNone(sys_props['Offset'])

tests/utils/testutils.py

@@ -232,7 +232,7 @@ def setUpClass(cls):
         docker_configs = DockerConfigs
         docker_configs.script_path = script_dir
         docker_configs.libraries = cls.get_libraries_to_install()
-        docker_configs.env = cls.get_environment_variables()
+        docker_configs.env = cls.get_environment_variables() or {}
 
         if is_envvar_true(PYAZURE_WEBHOST_DEBUG):
             cls.host_stdout = None

tests/utils/testutils_docker.py

@@ -98,6 +98,25 @@ def create_container(self, image_repo: str, image_url: str,
 
         worker_path = os.path.join(PROJECT_ROOT, 'azure_functions_worker')
         script_path = os.path.join(TESTS_ROOT, configs.script_path)
+        env = {"AzureWebJobsFeatureFlags": "EnableWorkerIndexing",
+               "AzureWebJobsStorage": f"{os.getenv('AzureWebJobsStorage')}",
+               "AzureWebJobsEventHubConnectionString":
+                   f"{os.getenv('AzureWebJobsEventHubConnectionString')}",
+               "AzureWebJobsCosmosDBConnectionString":
+                   f"{os.getenv('AzureWebJobsCosmosDBConnectionString')}",
+               "AzureWebJobsServiceBusConnectionString":
+                   f"{os.getenv('AzureWebJobsServiceBusConnectionString')}"
+                   "AzureWebJobsEventHubConnectionString"
+                   f"{os.getenv('AzureWebJobsEventHubConnectionString')}",
+               "AzureWebJobsSqlConnectionString":
+                   f"{os.getenv('AzureWebJobsSqlConnectionString')}",
+               "AzureWebJobsEventGridTopicUri":
+                   f"{os.getenv('AzureWebJobsEventGridTopicUri')}",
+               "AzureWebJobsEventGridConnectionKey":
+                   f"{os.getenv('AzureWebJobsEventGridConnectionKey')}"
+               }
+
+        configs.env.update(env)
 
         image = self.find_latest_image(image_repo, image_url)
 
@@ -129,31 +148,6 @@ def create_container(self, image_repo: str, image_url: str,
         run_cmd.extend(["--cap-add", "SYS_ADMIN"])
         run_cmd.extend(["--device", "/dev/fuse"])
         run_cmd.extend(["-e", f"CONTAINER_NAME={_uuid}"])
-        run_cmd.extend(["-e", "AzureWebJobsFeatureFlags=EnableWorkerIndexing"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsStorage="
-                        f"{os.getenv('AzureWebJobsStorage')}"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsEventHubConnectionString="
-                        f"{os.getenv('AzureWebJobsEventHubConnectionString')}"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsCosmosDBConnectionString="
-                        f"{os.getenv('AzureWebJobsCosmosDBConnectionString')}"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsServiceBusConnectionString="
-                        f"{os.getenv('AzureWebJobsServiceBusConnectionString')}"])  # NoQA
-        run_cmd.extend(["-e",
-                        "AzureWebJobsEventHubConnectionString="
-                        f"{os.getenv('AzureWebJobsEventHubConnectionString')}"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsSqlConnectionString="
-                        f"{os.getenv('AzureWebJobsSqlConnectionString')}"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsEventGridTopicUri="
-                        f"{os.getenv('AzureWebJobsEventGridTopicUri')}"])
-        run_cmd.extend(["-e",
-                        "AzureWebJobsEventGridConnectionKey="
-                        f"{os.getenv('AzureWebJobsEventGridConnectionKey')}"])
         run_cmd.extend(["-e", f"AzureFunctionsWebHost__hostid={_uuid}"])
         run_cmd.extend(["-v", f"{worker_path}:{container_worker_path}"])
         run_cmd.extend(["-v", f"{script_path}:{function_path}"])