Initial seed

Initial upload of django test framework

Change-Id: I643a7f4efc52cfafe4cc6d92e3178f36a0c1837c
Issue-Id: VVP-1
Signed-off-by: Edan Binshtok <eb578m@intl.att.com>

1 files changed, 161 insertions, 0 deletions

diff --git a/services/api/api_rados.py b/services/api/api_rados.py
new file mode 100644
index 0000000..61cfa5c
--- /dev/null
+++ b/services/api/api_rados.py
@@ -0,0 +1,161 @@
+ 
+# ============LICENSE_START========================================== 
+# org.onap.vvp/test-engine
+# ===================================================================
+# Copyright © 2017 AT&T Intellectual Property. All rights reserved.
+# ===================================================================
+#
+# Unless otherwise specified, all software contained herein is licensed
+# under the Apache License, Version 2.0 (the “License”);
+# you may not use this software 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.
+#
+#
+#
+# Unless otherwise specified, all documentation contained herein is licensed
+# under the Creative Commons License, Attribution 4.0 Intl. (the “License”);
+# you may not use this documentation except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#             https://creativecommons.org/licenses/by/4.0/
+#
+# Unless required by applicable law or agreed to in writing, documentation
+# 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.
+#
+# ============LICENSE_END============================================
+#
+# ECOMP is a trademark and service mark of AT&T Intellectual Property.
+import logging
+import time
+
+from boto.s3.connection import S3Connection, OrdinaryCallingFormat
+from django.conf import settings
+
+from rados.rgwa_client_factory import RGWAClientFactory
+from services.constants import Constants
+from services.logging_service import LoggingServiceFactory
+from services.session import session
+
+
+logger = LoggingServiceFactory.get_logger()
+
+class APIRados:
+
+    @staticmethod
+    def get_bucket(name):
+        """Return the Bucket."""
+        boto_conn = RGWAClientFactory.standard()
+        try:
+            return boto_conn.lookup(name)
+        except Exception as e:
+            logger.error("Problem on get bucket", e)
+            raise e
+
+    @staticmethod
+    def get_bucketfor_specific_user(name, access_key_id, secret_access_key):
+        """Return the Bucket."""
+        boto_conn = APIRados.specific_client(access_key_id, secret_access_key)
+        try:
+            return boto_conn.lookup(name)
+        except Exception as e:
+            logger.error("Problem on get bucket for specific user", e)
+            raise e
+
+    @staticmethod
+    def get_bucket_grants(bucket_name):
+        """Return the Grants."""
+        counter = 1
+        bucket = APIRados.get_bucket(bucket_name)
+        while not bucket and counter <= Constants.RGWAConstants.BUCKET_RETRIES_NUMBER:
+            logger.error("Bucket not found. Retry #%s" % counter)
+            time.sleep(session.wait_until_time_pause_long)
+            bucket = APIRados.get_bucket(bucket_name)
+            counter += 1
+        if not bucket:
+            raise TimeoutError("Max retries exceeded, failing test...")
+        grants = bucket.list_grants()
+        print("***********grants=", grants)
+        return grants
+
+    @staticmethod
+    def is_bucket_ready(bucket_id):
+        counter = 1
+        bucket = APIRados.get_bucket(bucket_id)
+        while (bucket == None and counter <=
+               Constants.RGWAConstants.BUCKET_RETRIES_NUMBER):
+            time.sleep(session.wait_until_time_pause_long)
+            logger.debug(
+                "bucket are not ready yet, trying again (%s of 180)" % counter)
+            bucket = APIRados.get_bucket(bucket_id)
+            counter += 1
+            print("****_+__+bucket= ", str(bucket))
+        time.sleep(session.wait_until_time_pause_long)
+        if bucket == None:
+            raise TimeoutError("Max retries exceeded, failing test...")
+        elif bucket != None:
+            logger.debug("bucket are ready to continue!")
+            return True
+
+    @staticmethod
+    def users_of_bucket_ready_after_complete(bucket_id, user_name):
+        grants = APIRados.get_bucket_grants(bucket_id)
+        count = 0
+        counter = 1
+        while (count != 0 and counter <=
+               Constants.RGWAConstants.BUCKET_RETRIES_NUMBER):
+            grants = APIRados.get_bucket_grants(bucket_id)
+            time.sleep(session.wait_until_time_pause_long)
+            for g in grants:
+                if g.id == user_name:
+                    count = +1
+        time.sleep(session.wait_until_time_pause_long)
+        if count != 0:
+            raise Exception("Max retries exceeded, failing test...")
+            return False
+        elif count == 0:
+            logger.debug("users_of_bucket are ready to continue!")
+            return True
+
+    @staticmethod
+    def users_of_bucket_ready_after_created(bucket_id, user_name):
+        grants = APIRados.get_bucket_grants(bucket_id)
+        count = 0
+        counter = 1
+        while (count == 0 and counter <=
+               Constants.RGWAConstants.BUCKET_RETRIES_NUMBER):
+            grants = APIRados.get_bucket_grants(bucket_id)
+            time.sleep(session.wait_until_time_pause_long)
+            for g in grants:
+                if g.id == user_name:
+                    count = +1
+        time.sleep(session.wait_until_time_pause_long)
+        if count == 0:
+            raise Exception("Max retries exceeded, failing test...")
+            return False
+        elif count > 0:
+            logger.debug("users_of_bucket are ready to continue!")
+            return True
+
+    @staticmethod
+    def specific_client(access_key_id, secret_access_key):
+        boto_conn = S3Connection(
+            host=settings.AWS_S3_HOST,
+            port=settings.AWS_S3_PORT,
+            aws_access_key_id=access_key_id,
+            aws_secret_access_key=secret_access_key,
+            calling_format=OrdinaryCallingFormat(),
+            is_secure=True,)
+
+        boto_conn.num_retries = 0
+        return boto_conn