Merge "Add "resources" parameter in Kubernetes context"

1 files changed, 18 insertions, 0 deletions

diff --git a/yardstick/tests/unit/orchestrator/test_kubernetes.py b/yardstick/tests/unit/orchestrator/test_kubernetes.py
index 5eba1a02a..2451518a4 100644
--- a/yardstick/tests/unit/orchestrator/test_kubernetes.py
+++ b/yardstick/tests/unit/orchestrator/test_kubernetes.py
@@ -302,6 +302,24 @@ class ContainerObjectTestCase(base.BaseUnitTestCase):
                              'value': 'fake_var_value'}]}
         self.assertEqual(expected, container_obj.get_container_item())
 
+    def test_get_container_item_with_resources(self):
+        volume_mount = {'name': 'fake_name',
+                        'mountPath': 'fake_path'}
+        args = ['arg1', 'arg2']
+        resources = {'requests': {'key1': 'val1'},
+                     'limits': {'key2': 'val2'},
+                     'other_key': {'key3': 'val3'}}
+        container_obj = kubernetes.ContainerObject(
+            'cname', ssh_key='fake_sshkey', volumeMount=[volume_mount],
+            args=args, resources=resources)
+        expected = {'args': args,
+                    'command': [kubernetes.ContainerObject.COMMAND_DEFAULT],
+                    'image': kubernetes.ContainerObject.IMAGE_DEFAULT,
+                    'name': 'cname-container',
+                    'volumeMounts': container_obj._create_volume_mounts(),
+                    'resources': {'requests': {'key1': 'val1'},
+                                  'limits': {'key2': 'val2'}}}
+        self.assertEqual(expected, container_obj.get_container_item())
 
 class CustomResourceDefinitionObjectTestCase(base.BaseUnitTestCase):