mirror of
https://github.com/aptly-dev/aptly.git
synced 2026-01-12 03:21:33 +00:00
81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
from lib import BaseTest
|
|
import uuid
|
|
import os
|
|
|
|
try:
|
|
import swiftclient
|
|
|
|
if 'OS_USERNAME' in os.environ and 'OS_PASSWORD' in os.environ:
|
|
auth_username = os.environ.get('OS_USERNAME')
|
|
auth_password = os.environ.get('OS_PASSWORD')
|
|
# Using auth version 2 /v2.0/
|
|
auth_url = os.environ.get('OS_AUTH_URL')
|
|
auth_tenant = os.environ.get('OS_TENANT_NAME')
|
|
|
|
account_username = "%s:%s" % (auth_tenant, auth_username)
|
|
swift_conn = swiftclient.Connection(auth_url, account_username,
|
|
auth_password, auth_version=2)
|
|
else:
|
|
swift_conn = None
|
|
except ImportError:
|
|
swift_conn = None
|
|
|
|
|
|
class SwiftTest(BaseTest):
|
|
"""
|
|
BaseTest + support for Swift
|
|
"""
|
|
|
|
def fixture_available(self):
|
|
return super(SwiftTest, self).fixture_available() and swift_conn is not None
|
|
|
|
def prepare(self):
|
|
self.container_name = "aptly-sys-test-" + str(uuid.uuid4())
|
|
swift_conn.put_container(self.container_name)
|
|
|
|
self.configOverride = {"SwiftPublishEndpoints": {
|
|
"test1": {
|
|
"container": self.container_name,
|
|
}
|
|
}}
|
|
|
|
super(SwiftTest, self).prepare()
|
|
|
|
def shutdown(self):
|
|
if hasattr(self, "container_name"):
|
|
for obj in swift_conn.get_container(self.container_name,
|
|
full_listing=True)[1]:
|
|
swift_conn.delete_object(self.container_name, obj.get("name"))
|
|
|
|
swift_conn.delete_container(self.container_name)
|
|
super(SwiftTest, self).shutdown()
|
|
|
|
def check_path(self, path):
|
|
if not hasattr(self, "container_contents"):
|
|
self.container_contents = [obj.get('name') for obj in
|
|
swift_conn.get_container(self.container_name)[1]]
|
|
|
|
if path in self.container_contents:
|
|
return True
|
|
|
|
if not path.endswith("/"):
|
|
path = path + "/"
|
|
|
|
for item in self.container_contents:
|
|
if item.startswith(path):
|
|
return True
|
|
|
|
return False
|
|
|
|
def check_exists(self, path):
|
|
if not self.check_path(path):
|
|
raise Exception("path %s doesn't exist" % (path, ))
|
|
|
|
def check_not_exists(self, path):
|
|
if self.check_path(path):
|
|
raise Exception("path %s exists" % (path, ))
|
|
|
|
def read_file(self, path):
|
|
hdrs, body = swift_conn.get_object(self.container_name, path)
|
|
return body
|