Initial patch with all code from CableLabs repository.

Change-Id: I70a2778718c5e7f21fd14e4ad28c9269d3761cc7
Signed-off-by: spisarski <s.pisarski@cablelabs.com>

1 files changed, 228 insertions, 0 deletions

diff --git a/snaps/openstack/tests/create_project_tests.py b/snaps/openstack/tests/create_project_tests.py
new file mode 100644
index 0000000..9d53467
--- /dev/null
+++ b/snaps/openstack/tests/create_project_tests.py
@@ -0,0 +1,228 @@
+# Copyright (c) 2016 Cable Television Laboratories, Inc. ("CableLabs")
+#                    and others.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at:
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import uuid
+import unittest
+
+from snaps.openstack.create_project import OpenStackProject, ProjectSettings
+from snaps.openstack.create_security_group import OpenStackSecurityGroup
+from snaps.openstack.create_security_group import SecurityGroupSettings
+from snaps.openstack.create_user import OpenStackUser
+from snaps.openstack.create_user import UserSettings
+from snaps.openstack.tests.os_source_file_test import OSComponentTestCase
+from snaps.openstack.utils import keystone_utils
+
+__author__ = 'spisarski'
+
+
+class ProjectSettingsUnitTests(unittest.TestCase):
+    """
+    Tests the construction of the ProjectSettings class
+    """
+
+    def test_no_params(self):
+        with self.assertRaises(Exception):
+            ProjectSettings()
+
+    def test_empty_config(self):
+        with self.assertRaises(Exception):
+            ProjectSettings(config=dict())
+
+    def test_name_only(self):
+        settings = ProjectSettings(name='foo')
+        self.assertEquals('foo', settings.name)
+        self.assertEquals('default', settings.domain)
+        self.assertIsNone(settings.description)
+        self.assertTrue(settings.enabled)
+
+    def test_config_with_name_only(self):
+        settings = ProjectSettings(config={'name': 'foo'})
+        self.assertEquals('foo', settings.name)
+        self.assertEquals('default', settings.domain)
+        self.assertIsNone(settings.description)
+        self.assertTrue(settings.enabled)
+
+    def test_all(self):
+        settings = ProjectSettings(name='foo', domain='bar', description='foobar', enabled=False)
+        self.assertEquals('foo', settings.name)
+        self.assertEquals('bar', settings.domain)
+        self.assertEquals('foobar', settings.description)
+        self.assertFalse(settings.enabled)
+
+    def test_config_all(self):
+        settings = ProjectSettings(config={'name': 'foo', 'domain': 'bar', 'description': 'foobar', 'enabled': False})
+        self.assertEquals('foo', settings.name)
+        self.assertEquals('bar', settings.domain)
+        self.assertEquals('foobar', settings.description)
+        self.assertFalse(settings.enabled)
+
+
+class CreateProjectSuccessTests(OSComponentTestCase):
+    """
+    Test for the CreateImage class defined in create_image.py
+    """
+
+    def setUp(self):
+        """
+        Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
+        within OpenStack
+        """
+        guid = str(uuid.uuid4())[:-19]
+        guid = self.__class__.__name__ + '-' + guid
+        self.project_settings = ProjectSettings(name=guid + '-name')
+
+        self.keystone = keystone_utils.keystone_client(self.os_creds)
+
+        # Initialize for cleanup
+        self.project_creator = None
+
+    def tearDown(self):
+        """
+        Cleans the image and downloaded image file
+        """
+        if self.project_creator:
+            self.project_creator.clean()
+
+    def test_create_project(self):
+        """
+        Tests the creation of an OpenStack project.
+        """
+        self.project_creator = OpenStackProject(self.os_creds, self.project_settings)
+        created_project = self.project_creator.create()
+        self.assertIsNotNone(created_project)
+
+        retrieved_project = keystone_utils.get_project(keystone=self.keystone, project_name=self.project_settings.name)
+        self.assertIsNotNone(retrieved_project)
+        self.assertEquals(created_project, retrieved_project)
+
+    def test_create_project_2x(self):
+        """
+        Tests the creation of an OpenStack project twice to ensure it only creates one.
+        """
+        self.project_creator = OpenStackProject(self.os_creds, self.project_settings)
+        created_project = self.project_creator.create()
+        self.assertIsNotNone(created_project)
+
+        retrieved_project = keystone_utils.get_project(keystone=self.keystone, project_name=self.project_settings.name)
+        self.assertIsNotNone(retrieved_project)
+        self.assertEquals(created_project, retrieved_project)
+
+        project2 = OpenStackProject(self.os_creds, self.project_settings).create()
+        self.assertEquals(retrieved_project, project2)
+
+    def test_create_delete_project(self):
+        """
+        Tests the creation of an OpenStack project, it's deletion, then cleanup.
+        """
+        # Create Image
+        self.project_creator = OpenStackProject(self.os_creds, self.project_settings)
+        created_project = self.project_creator.create()
+        self.assertIsNotNone(created_project)
+
+        keystone_utils.delete_project(self.keystone, created_project)
+
+        self.project_creator.clean()
+
+        self.assertIsNone(self.project_creator.get_project())
+
+        # TODO - Expand tests
+
+
+class CreateProjectUserTests(OSComponentTestCase):
+    """
+    Test for the CreateImage class defined in create_image.py
+    """
+
+    def setUp(self):
+        """
+        Instantiates the CreateImage object that is responsible for downloading and creating an OS image file
+        within OpenStack
+        """
+        guid = str(uuid.uuid4())[:-19]
+        self.guid = self.__class__.__name__ + '-' + guid
+        self.project_settings = ProjectSettings(name=self.guid + '-name')
+
+        self.keystone = keystone_utils.keystone_client(self.os_creds)
+
+        # Initialize for cleanup
+        self.project_creator = None
+        self.user_creators = list()
+
+        self.sec_grp_creators = list()
+
+    def tearDown(self):
+        """
+        Cleans the image and downloaded image file
+        """
+        for sec_grp_creator in self.sec_grp_creators:
+            sec_grp_creator.clean()
+
+        for user_creator in self.user_creators:
+            user_creator.clean()
+
+        if self.project_creator:
+            self.project_creator.clean()
+
+    def test_create_project_sec_grp_one_user(self):
+        """
+        Tests the creation of an OpenStack object to a project with a new users and to create a security group
+        """
+        self.project_creator = OpenStackProject(self.os_creds, self.project_settings)
+        created_project = self.project_creator.create()
+        self.assertIsNotNone(created_project)
+
+        user_creator = OpenStackUser(self.os_creds, UserSettings(name=self.guid + '-user', password=self.guid))
+        self.project_creator.assoc_user(user_creator.create())
+        self.user_creators.append(user_creator)
+
+        sec_grp_os_creds = user_creator.get_os_creds(self.project_creator.get_project().name)
+        sec_grp_creator = OpenStackSecurityGroup(
+            sec_grp_os_creds, SecurityGroupSettings(name=self.guid + '-name', description='hello group'))
+        sec_grp = sec_grp_creator.create()
+        self.assertIsNotNone(sec_grp)
+        self.sec_grp_creators.append(sec_grp_creator)
+
+        if self.keystone.version == keystone_utils.V2_VERSION:
+            self.assertEquals(self.project_creator.get_project().id, sec_grp['security_group']['tenant_id'])
+        else:
+            self.assertEquals(self.project_creator.get_project().id, sec_grp['security_group']['project_id'])
+
+    def test_create_project_sec_grp_two_users(self):
+        """
+        Tests the creation of an OpenStack object to a project with two new users and use each user to create a
+        security group
+        """
+        self.project_creator = OpenStackProject(self.os_creds, self.project_settings)
+        created_project = self.project_creator.create()
+        self.assertIsNotNone(created_project)
+
+        user_creator_1 = OpenStackUser(self.os_creds, UserSettings(name=self.guid + '-user1', password=self.guid))
+        self.project_creator.assoc_user(user_creator_1.create())
+        self.user_creators.append(user_creator_1)
+
+        user_creator_2 = OpenStackUser(self.os_creds, UserSettings(name=self.guid + '-user2', password=self.guid))
+        self.project_creator.assoc_user(user_creator_2.create())
+        self.user_creators.append(user_creator_2)
+
+        ctr = 0
+        for user_creator in self.user_creators:
+            ctr += 1
+            sec_grp_os_creds = user_creator.get_os_creds(self.project_creator.get_project().name)
+
+            sec_grp_creator = OpenStackSecurityGroup(
+                sec_grp_os_creds, SecurityGroupSettings(name=self.guid + '-name', description='hello group'))
+            sec_grp = sec_grp_creator.create()
+            self.assertIsNotNone(sec_grp)
+            self.sec_grp_creators.append(sec_grp_creator)
+            self.assertEquals(self.project_creator.get_project().id, sec_grp['security_group']['tenant_id'])