Merge pull request #1145 from xliiv/release/2.1.0

added get_environments method to ServiceCatalog

src/ralph/discovery/models_device.py

@@ -311,6 +311,13 @@ class Meta:
     def __unicode__(self):
         return self.name
 
+    def get_environments(self):
+        env_ids_from_service = models_ci.CIRelation.objects.filter(
+            parent=self.id,
+        ).values('child__id')
+        envs = DeviceEnvironment.objects.filter(id__in=env_ids_from_service)
+        return envs
+
 
 class Device(
     LastSeen,

src/ralph/discovery/tests/test_models.py

@@ -10,6 +10,11 @@
 from django.test import TestCase
 import mock
 
+from ralph.cmdb.tests.utils import (
+    CIRelationFactory,
+    DeviceEnvironmentFactory,
+    ServiceCatalogFactory,
+)
 from ralph.discovery.models import DeviceType, Device, UptimeSupport
 from ralph.discovery.models_history import HistoryChange
 
@@ -82,3 +87,13 @@ class Model(UptimeSupport):
         self.assertEqual(m.uptime, None)
         m.uptime = 132
         self.assertEqual(m.uptime, datetime.timedelta(seconds=132))
+
+
+class ServiceEnvironments(TestCase):
+
+    def test_getting_environments(self):
+        service = ServiceCatalogFactory()
+        env = DeviceEnvironmentFactory()
+        self.assertEqual(len(service.get_environments()), 0)
+        CIRelationFactory(parent=service, child=env)
+        self.assertEqual(len(service.get_environments()), 1)