save changes required for work; TODO locate all places that need to b…

…e updated to make test working

airbyte-integrations/bases/source-acceptance-test/source_acceptance_test/tests/__init__.py

@@ -1,5 +1,5 @@
-from .test_core import TestBasicRead, TestConnection, TestDiscovery, TestSpec
+from .test_core import TestBasicRead, TestConnection, TestDiscovery, TestSpec, TestEnvVarExists
 from .test_full_refresh import TestFullRefresh
 from .test_incremental import TestIncremental
 
-__all__ = ["TestSpec", "TestBasicRead", "TestConnection", "TestDiscovery", "TestFullRefresh", "TestIncremental"]
+__all__ = ["TestSpec", "TestBasicRead", "TestConnection", "TestDiscovery", "TestFullRefresh", "TestIncremental", "TestEnvVarExists"]

airbyte-integrations/bases/source-acceptance-test/source_acceptance_test/tests/test_core.py

@@ -35,6 +35,11 @@
 from source_acceptance_test.utils import ConnectorRunner, serialize
 
 
+class TestEnvVarExists(BaseTest):
+    def test_env_var_exists(self, connector_config, docker_runner: ConnectorRunner):
+        result = docker_runner.has_env_var()
+        assert result
+
 @pytest.mark.default_timeout(10)
 class TestSpec(BaseTest):
     def test_spec(self, connector_spec: ConnectorSpecification, docker_runner: ConnectorRunner):

airbyte-integrations/bases/source-acceptance-test/source_acceptance_test/utils/connector_runner.py

@@ -130,3 +130,9 @@ def run(self, cmd, config=None, state=None, catalog=None, **kwargs) -> Iterable[
                 yield AirbyteMessage.parse_raw(line)
             except ValidationError as exc:
                 logging.warning("Unable to parse connector's output %s", exc)
+
+    def has_env_var(self, lookup_env_var: str = "AIRBYTE_ENTRYPOINT"):
+        self._env_vars = self._image.attrs['Config']['Env']
+        self._env_airbyte_entrypoint = [i for i in self._env_vars if i.startswith(lookup_env_var)]
+        ret = self._env_airbyte_entrypoint and self._env_airbyte_entrypoint[0].split(f"{lookup_env_var}=", 1)[1]
+        return bool(ret)