newrelic piwd test

Author: Victoria Hall

tests/unittests/test_utilities_dependency.py

@@ -642,6 +642,116 @@ def test_remove_module_cache_with_namespace_remain(self):
         DependencyManager._remove_module_cache(self._customer_deps_path)
         self.assertIsNotNone(common_module)
 
+    @unittest.skipIf(sys.version_info.minor > 7,
+                     "The worker brings different protobuf versions"
+                     "between 3.7 and 3.8+.")
+    def test_newrelic_protobuf_import_scenario_worker_deps_37(self):
+        # https://github.com/Azure/azure-functions-python-worker/issues/1339
+        # newrelic checks if protobuf has been imported and based on the
+        # version it finds, imports a specific pb2 file.
+
+        # PIWD = 0. protobuf is brought through the worker's deps.
+        os.environ['PYTHON_ISOLATE_WORKER_DEPENDENCIES'] = 'false'
+
+        # Setup paths
+        DependencyManager.worker_deps_path = self._worker_deps_path
+        DependencyManager.cx_deps_path = self._customer_deps_path
+        DependencyManager.cx_working_dir = self._customer_func_path
+
+        DependencyManager.prioritize_customer_dependencies()
+
+        # protobuf v3 is found
+        from google.protobuf import __version__
+
+        protobuf_version = tuple(int(v) for v in __version__.split("."))
+        self.assertIsNotNone(protobuf_version)
+        self.assertEqual(protobuf_version[0], 3)
+
+    @unittest.skipIf(sys.version_info.minor <= 7,
+                     "The worker brings different protobuf versions"
+                     "between 3.7 and 3.8+.")
+    def test_newrelic_protobuf_import_scenario_worker_deps(self):
+        # https://github.com/Azure/azure-functions-python-worker/issues/1339
+        # newrelic checks if protobuf has been imported and based on the
+        # version it finds, imports a specific pb2 file.
+
+        # PIWD = 0. protobuf is brought through the worker's deps.
+        os.environ['PYTHON_ISOLATE_WORKER_DEPENDENCIES'] = 'false'
+
+        # Setup paths
+        DependencyManager.worker_deps_path = self._worker_deps_path
+        DependencyManager.cx_deps_path = self._customer_deps_path
+        DependencyManager.cx_working_dir = self._customer_func_path
+
+        DependencyManager.prioritize_customer_dependencies()
+
+        # protobuf v4 is found
+        from google.protobuf import __version__
+
+        protobuf_version = tuple(int(v) for v in __version__.split("."))
+        self.assertIsNotNone(protobuf_version)
+        self.assertEqual(protobuf_version[0], 4)
+
+    @unittest.skipIf(sys.version_info.minor > 7,
+                     "The worker brings different protobuf versions"
+                     "between 3.7 and 3.8+.")
+    def test_newrelic_protobuf_import_scenario_user_deps_37(self):
+        # https://github.com/Azure/azure-functions-python-worker/issues/1339
+        # newrelic checks if protobuf has been imported and based on the
+        # version it finds, imports a specific pb2 file.
+
+        # PIWD = 1. protobuf is brought through the user's deps.
+        os.environ['PYTHON_ISOLATE_WORKER_DEPENDENCIES'] = 'true'
+
+        # Setup paths
+        DependencyManager.worker_deps_path = self._worker_deps_path
+        DependencyManager.cx_deps_path = self._customer_deps_path
+        DependencyManager.cx_working_dir = self._customer_func_path
+
+        DependencyManager.prioritize_customer_dependencies()
+
+        # protobuf is found from worker deps, but newrelic won't find it
+        from google.protobuf import __version__
+
+        protobuf_version = tuple(int(v) for v in __version__.split("."))
+        self.assertIsNotNone(protobuf_version)
+
+        # newrelic tries to import protobuf v3
+        self.assertEqual(protobuf_version[0], 3)
+
+        # newrelic tries to import protobuf v4
+        self.assertNotEqual(protobuf_version[0], 4)
+
+    @unittest.skipIf(sys.version_info.minor <= 7,
+                     "The worker brings different protobuf versions"
+                     "between 3.7 and 3.8+.")
+    def test_newrelic_protobuf_import_scenario_user_deps(self):
+        # https://github.com/Azure/azure-functions-python-worker/issues/1339
+        # newrelic checks if protobuf has been imported and based on the
+        # version it finds, imports a specific pb2 file.
+
+        # PIWD = 1. protobuf is brought through the user's deps.
+        os.environ['PYTHON_ISOLATE_WORKER_DEPENDENCIES'] = 'true'
+
+        # Setup paths
+        DependencyManager.worker_deps_path = self._worker_deps_path
+        DependencyManager.cx_deps_path = self._customer_deps_path
+        DependencyManager.cx_working_dir = self._customer_func_path
+
+        DependencyManager.prioritize_customer_dependencies()
+
+        # protobuf is found from worker deps, but newrelic won't find it
+        from google.protobuf import __version__
+
+        protobuf_version = tuple(int(v) for v in __version__.split("."))
+        self.assertIsNotNone(protobuf_version)
+
+        # newrelic tries to import protobuf v4
+        self.assertEqual(protobuf_version[0], 4)
+
+        # newrelic tries to import protobuf v3
+        self.assertNotEqual(protobuf_version[0], 3)
+
     def _initialize_scenario(self):
         # Setup app settings
         os.environ['PYTHON_ISOLATE_WORKER_DEPENDENCIES'] = 'true'