chore(entity): change logic of update and delete

test/entities/test_objects_to_yaml.py

@@ -8,27 +8,27 @@
 pytestmark = pytest.mark.django_db
 
 
-@pytest.fixture
-def generate_base_component(component_base_factory, link_factory, group_factory):
+def test_generate_base_component(group_factory, component_base_factory, link_factory):
+    group = group_factory(
+        id=1, product_owner="john", project_owner="doe"
+    )
     component = component_base_factory(
         id=1,
         name="base",
+        label="Base",
         type="database",
         description="This is my fancy component",
         kind="component",
         owner="platform",
         service=None,
         library=None,
+        group=group,
         source__id=1,
         source__remote_id=1,
         source__owner="jasckson",
         source__name="thiwer",
         source__url="https://gitlab.com/thiwer/thiwer",
     )
-    component.save()
-
-    group = group_factory(id=1, product_owner="john", project_owner="doe", entity=component)
-    group.save()
 
     link_factory(
         id=1,
@@ -44,13 +44,12 @@ def generate_base_component(component_base_factory, link_factory, group_factory)
         entity=component,
     )
 
-
-def test_generate_base_component(generate_base_component):
     expected = """
 - apiVersion: v1alpha1
   kind_: component
   metadata:
     name: base
+    label: Base
     owner: platform
     group:
       product_owner: john
@@ -64,11 +63,92 @@ def test_generate_base_component(generate_base_component):
       icon: poop
     - name: Sentry
       url: https://sentry.skypicker.com
-      icon: null
   spec:
     type_: database
 """
     repository = Repository.objects.get(pk=1)
     content = generate(repository)
     assert validate(content)
     assert expected.strip() == content.strip()
+
+
+def test_generate_component_service(component_base_factory, service_factory, group_factory,
+                                    link_factory, repository_factory, environment_factory):
+    repository = repository_factory(
+        id=22,
+        remote_id=22,
+        name="test_proj1",
+        owner="john_doe1",
+        url="https://github.com/john_doe1/test_proj1",
+        provider="github",
+    )
+
+    service = service_factory(
+        owner="platform",
+        name="my-service",
+        lifecycle="production",
+        impact="employees",
+        repository=repository,
+        slack_channel="#platform-software",
+        docs_url="https://docs.com"
+    )
+    environment_factory(
+        name="production",
+        service=service,
+        health_check_url="https://health.com",
+        dashboard_url="https://dashboard.datadog.com",
+        service_urls=["https://service.com"]
+    )
+    group = group_factory(
+        id=1, product_owner="john", project_owner="doe"
+    )
+    component_service = component_base_factory(
+        id=1,
+        name="service",
+        label="Service",
+        type="service",
+        description="This is my fancy service",
+        kind="component",
+        owner="platform",
+        service=service,
+        library=None,
+        group=group,
+        source=repository
+    )
+    link_factory(
+        id=1,
+        name="Datadog",
+        url="https://dashboard.datadog.com",
+        icon="poop",
+        entity=component_service,
+    )
+    expected = """
+- apiVersion: v1alpha1
+  kind_: component
+  metadata:
+    name: service
+    label: Service
+    owner: platform
+    group:
+      product_owner: john
+      project_owner: doe
+      maintainers: []
+    description: This is my fancy service
+    tags: []
+    links:
+    - name: Datadog
+      url: https://dashboard.datadog.com
+      icon: poop
+  spec:
+    type_: service
+    environments:
+    - name: production
+      dashboard_url: https://dashboard.datadog.com
+      health_check_url: https://health.com
+      service_urls:
+      - https://service.com
+"""
+    repository = Repository.objects.get(pk=22)
+    content = generate(repository)
+    assert validate(content)
+    assert expected.strip() == content.strip()