getitem and delitem implemented on resource group

orchestrator/model/resources.py

@@ -130,6 +130,12 @@ class ResourceGroup(Resource):
         except InvalidResourceException:
         except InvalidResourceException:
             del self.__resources[resource.id]
             del self.__resources[resource.id]
 
 
+    def __getitem__(self, resource_id):
+        return self.__resources[resource_id]
+
+    def __delitem__(self, resource_id):
+        self.remove(self[resource_id])
+
     @property
     @property
     def resources(self):
     def resources(self):
         return list(self.__resources.values())
         return list(self.__resources.values())

tests/test_resources.py

+from __future__ import print_function
+
 import copy
 import copy
 import json
 import json
 from unittest.case import TestCase
 from unittest.case import TestCase
@@ -208,7 +210,40 @@ class ResourceGroupTest(TestCase):
         # then
         # then
         self.assertEqual([], group.resources)
         self.assertEqual([], group.resources)
 
 
-    # @skip('works same way as resource addition')
+    def test_resource_group_del_item_by_id(self):
+        # given
+        expected_resource_id = 'some_resource'
+        expected_resource = Instance(id=expected_resource_id)
+        group = ResourceGroup(resources=[expected_resource])
+
+        # when
+        del group[expected_resource_id]
+
+        # then
+        self.assertNotIn(expected_resource, group.resources)
+
+    def test_resource_group_del_invalid_id(self):
+        # given
+        group = ResourceGroup()
+
+        # then
+        with self.assertRaises(KeyError):
+            del group['invalid_resource_id']
+
+    def test_resource_group_get_item_by_id(self):
+        # given
+        expected_resource_id = 'some_resource'
+        expected_resource = Instance(id=expected_resource_id)
+        group = ResourceGroup(resources=[expected_resource])
+
+        # when
+        result = group[expected_resource_id]
+
+        # then
+        self.assertEqual(expected_resource, result)
+        with self.assertRaises(KeyError):
+            print(group['invalid_resource_id'])
+
     def test_resource_group_addition(self):
     def test_resource_group_addition(self):
         # given
         # given
         group1 = ResourceGroup()
         group1 = ResourceGroup()