fix(env/integration): Add entity_selector field used by dynatrace (#124)

Author: gagantrivedi

flag_engine/environments/integrations/models.py

@@ -5,3 +5,4 @@
 class IntegrationModel:
     api_key: str = None
     base_url: str = None
+    entity_selector: str = None

flag_engine/environments/integrations/schemas.py

@@ -7,6 +7,7 @@
 class IntegrationSchema(LoadToModelSchema):
     api_key = fields.Str(required=False, allow_none=True)
     base_url = fields.Str(required=False, allow_none=True)
+    entity_selector = fields.Str(required=False, allow_none=True)
 
     class Meta:
         model_class = IntegrationModel

flag_engine/environments/models.py

@@ -44,13 +44,15 @@ class EnvironmentModel:
     segment_config: IntegrationModel = None
     mixpanel_config: IntegrationModel = None
     heap_config: IntegrationModel = None
+    dynatrace_config: IntegrationModel = None
     webhooks: typing.List[WebhookModel] = field(default_factory=list)
 
     _INTEGRATION_ATTS = [
         "amplitude_config",
         "segment_config",
         "mixpanel_config",
         "heap_config",
+        "dynatrace_config",
         "webhooks",
     ]
 
@@ -76,5 +78,6 @@ def integrations_data(self) -> dict:
                 integrations_data[integration_attr] = {
                     "base_url": integration_config.base_url,
                     "api_key": integration_config.api_key,
+                    "entity_selector": integration_config.entity_selector,
                 }
         return integrations_data

flag_engine/environments/schemas.py

@@ -45,6 +45,7 @@ class BaseEnvironmentSchema(Schema):
     heap_config = fields.Nested(IntegrationSchema, required=False, allow_none=True)
     mixpanel_config = fields.Nested(IntegrationSchema, required=False, allow_none=True)
     amplitude_config = fields.Nested(IntegrationSchema, required=False, allow_none=True)
+    dynatrace_config = fields.Nested(IntegrationSchema, required=False, allow_none=True)
 
 
 class EnvironmentSchema(LoadToModelMixin, BaseEnvironmentSchema):

tests/unit/environments/test_environments_models.py

@@ -70,18 +70,36 @@ def test_environment_integrations_data_returns_correct_data_when_multiple_integr
 ):
     # Given
     example_key = "some-key"
-    segment_url = "https://api.segment.com"
-
+    base_url = "https://some-integration-url"
+    entity_selector = "some-entity-selector"
+    environment.dynatrace_config = IntegrationModel(
+        api_key=example_key,
+        base_url=base_url,
+        entity_selector=entity_selector,
+    )
     environment.mixpanel_config = IntegrationModel(api_key=example_key)
     environment.segment_config = IntegrationModel(
-        api_key=example_key, base_url=segment_url
+        api_key=example_key, base_url=base_url
     )
 
     # When
     integrations_data = environment.integrations_data
 
     # Then
     assert integrations_data == {
-        "mixpanel_config": {"api_key": example_key, "base_url": None},
-        "segment_config": {"api_key": example_key, "base_url": segment_url},
+        "mixpanel_config": {
+            "api_key": example_key,
+            "base_url": None,
+            "entity_selector": None,
+        },
+        "segment_config": {
+            "api_key": example_key,
+            "base_url": base_url,
+            "entity_selector": None,
+        },
+        "dynatrace_config": {
+            "api_key": example_key,
+            "base_url": base_url,
+            "entity_selector": entity_selector,
+        },
     }