def domain_metrics_per_tag(request, tag, **kwargs):
client = UC(request.session.get('username'),
request.session.get('password'),
request.session.get('api_url'))
domains = client.domains(tags=[tag]).json()
metrics = [kwargs['model'](domain=d['id']) for d in domains]
return stats_render(request, metrics, **kwargs)
class ClientTestCase(unittest.TestCase):
"""
https://github.com/unbit/uwsgi.it/blob/master/CustomerQuickstart.md
"""
def setUp(self):
username = os.environ['UWSGI_IT_USERNAME']
password = os.environ['UWSGI_IT_PASSWORD']
url = os.environ['UWSGI_IT_URL']
self.client = UClient(username, password, url)
self.container = os.environ['UWSGI_IT_CONTAINER']
self.loopbox = os.environ['UWSGI_IT_LOOPBOX']
self.alarm = os.environ['UWSGI_IT_ALARM']
"""
News: curl https://kratos:[email protected]/api/news/
"""
def test_news_get(self):
r = self.client.news()
news = r.json()
self.assertEqual(r.uerror, False)
def test_news_get_without_credentials(self):
client = UClient(None, None, os.environ['UWSGI_IT_URL'])
r = client.news()
news = r.json()
self.assertEqual(r.uerror, False)
"""
Account Info: curl https://kratos:[email protected]/api/me/
"""
def test_me_get(self):
r = self.client.me()
me = r.json()
self.assertEqual(r.uerror, False)
def test_me_post_company(self):
company = "DogeCom"
r = self.client.update_me({'company': company})
self.assertEqual(r.uerror, False)
r = self.client.me()
me = r.json()
self.assertEqual(me['company'], company)
"""
Distributions: curl https://kratos:[email protected]/api/distros/
"""
def test_distros_get(self):
r = self.client.distros()
distros = r.json()
self.assertEqual(r.uerror, False)
"""
Containers: curl https://kratos:[email protected]/api/me/containers/
"""
def test_containers_get(self):
r = self.client.containers()
containers = r.json()
self.assertEqual(r.uerror, False)
def test_containers_get_single(self):
r = self.client.container(self.container)
container = r.json()
self.assertEqual(r.uerror, False)
def test_containers_set_distro(self):
distros = self.client.distros().json()
distro = random.choice([d['id'] for d in distros])
r = self.client.container_set_distro(self.container, distro)
self.assertEqual(r.uerror, False)
r = self.client.container(self.container)
container = r.json()
self.assertEqual(container['distro'], distro)
def test_containers_set_keys(self):
keys = ["ssh-rsa miao"]
r = self.client.container_set_keys(self.container, keys)
self.assertEqual(r.uerror, False)
r = self.client.container(self.container)
container = r.json()
self.assertEqual(container['ssh_keys'], keys)
"""
Domains: curl https://kratos:[email protected]/api/domains/
"""
def test_domains_get(self):
r = self.client.domains()
domains = r.json()
self.assertEqual(r.uerror, False)
def tests_domains_get_single(self):
r = self.client.domain(100)
self.assertEqual(r.status_code, 404)
def test_domains_add_delete(self):
domain = str(uuid.uuid4())
# add
r = self.client.add_domain(domain)
# lame but fails without settings the dns
self.assertEqual(r.uerror, True)
self.assertEqual("FORBIDDEN" in r.umessage, True)
# delete
# lame but delete fails if domain does not exist
self.client.delete_domain(domain)
self.assertEqual(r.uerror, True)
self.assertEqual("FORBIDDEN" in r.umessage, True)
r = self.client.domains()
domains = r.json()
found = [d for d in domains if d['name'] == domain]
self.assertEqual(found, [])
"""
Tags: curl https://kratos:[email protected]/api/tags/
"""
def test_tag_creation_delete(self):
r = self.client.create_tag("mytag")
tag_id = r.json()["id"]
r = self.client.delete_tag(tag_id)
r = self.client.list_tags()
tags = r.json()
found_tag = [t for t in tags if t['id'] == tag_id]
self.assertEqual(found_tag, [])
def test_tag_domain_filter(self):
r = self.client.domains(tags=["ciao"])
domains = r.json()
self.assertEqual(r.uerror, False)
def test_tag_container_filter(self):
r = self.client.containers(tags=["ciao"])
containers = r.json()
self.assertEqual(r.uerror, False)
"""
Reboot: curl -X POST -d '{"reboot":1}' https://kratos:[email protected]/api/containers/30009
"""
def test_reboot(self):
r = self.client.reboot_container(self.container)
self.assertEqual(r.uerror, False)
"""
Metrics: curl https://kratos:[email protected]/api/metrics/container.io.read/
curl https://kratos:[email protected]/api/metrics/domain.net.rx/
"""
def test_container_metric(self):
r = self.client.container_metric(self.container, "io.read")
self.assertEqual(r.uerror, False)
def test_domain_metric(self):
r = self.client.domains()
domain = r.json()[0]['id']
r = self.client.domain_metric(domain, "net.rx")
self.assertEqual(r.uerror, False)
"""
Loopboxes: curl https://kratos:[email protected]/api/loopboxes/
"""
def test_loopboxes(self):
r = self.client.loopboxes()
loopboxes = r.json()
self.assertEqual(r.uerror, False)
def test_loopboxes_creation_delete(self):
r = self.client.create_loopbox(self.container, 'myloopbox', 'myloop')
created_message = r.json()['message']
self.assertEqual(created_message, 'Created')
r = self.client.loopboxes()
loopboxes = r.json()
loopbox_id = [x['id'] for x in loopboxes if x['container'] == int(self.container) and x['filename'] == 'myloopbox' and x['mountpoint'] == 'myloop'][0]
r = self.client.delete_loopbox(loopbox_id)
deleted_message = r.json()['message']
self.assertEqual(deleted_message, 'Ok')
def test_loopbox(self):
r = self.client.loopbox(self.loopbox)
loopbox = r.json()
self.assertEqual(r.uerror, False)
"""
Alarms: curl https://kratos:[email protected]/api/alarms/
"""
def test_alarms_get(self):
r = self.client.alarms()
alarms = r.json()
self.assertEqual(r.uerror, False)
def test_alarms_creation_delete(self):
r = self.client.create_alarm(self.container, 'myalarm', {'class': 'my-uwsgiit-client-testclass'})
created_message = r.json()['message']
self.assertEqual(created_message, 'Created')
r = self.client.alarms({'class': 'my-uwsgiit-client-testclass'})
alarms = r.json()
self.assertEqual(len(alarms), 1)
alarm_id = alarms[0]['id']
r = self.client.delete_alarm(alarm_id)
deleted_message = r.json()['message']
self.assertEqual(deleted_message, 'Ok')
def test_alarm(self):
r = self.client.alarm(self.alarm)
alarm = r.json()
self.assertEqual(r.uerror, False)
def test_create_alarm_key(self):
r = self.client.create_alarm_key(self.container)
key = r.json()
self.assertEqual(r.uerror, False)
class ClientTestCase(TestCase):
"""
https://github.com/unbit/uwsgi.it/blob/master/CustomerQuickstart.md
"""
def setUp(self):
username = os.environ['UWSGI_IT_USERNAME']
password = os.environ['UWSGI_IT_PASSWORD']
url = os.environ['UWSGI_IT_URL']
self.client = UClient(username, password, url)
self.container = os.environ['UWSGI_IT_CONTAINER']
"""
Account Info: curl https://kratos:[email protected]/api/me/
"""
def test_me_get(self):
r = self.client.me()
me = r.json()
self.assertEqual(r.uerror, False)
def test_me_post_company(self):
company = "DogeCom"
r = self.client.update_me({'company': company })
self.assertEqual(r.uerror, False)
r = self.client.me()
me = r.json()
self.assertEqual(me['company'], company)
"""
Distributions: curl https://kratos:[email protected]/api/distros/
"""
def test_distros_get(self):
r = self.client.distros()
distros = r.json()
self.assertEqual(r.uerror, False)
"""
Containers: curl https://kratos:[email protected]/api/me/containers/
"""
def test_containers_get(self):
r = self.client.containers()
containers = r.json()
self.assertEqual(r.uerror, False)
def test_containers_get_single(self):
r = self.client.container(self.container)
container = r.json()
self.assertEqual(r.uerror, False)
def test_containers_set_distro(self):
distros = self.client.distros().json()
distro = random.choice([d['id'] for d in distros])
r = self.client.container_set_distro(self.container, distro)
self.assertEqual(r.uerror, False)
r = self.client.container(self.container)
container = r.json()
self.assertEqual(container['distro'], distro)
def test_containers_set_keys(self):
keys = ["ssh-rsa miao"]
r = self.client.container_set_keys(self.container, keys)
self.assertEqual(r.uerror, False)
r = self.client.container(self.container)
container = r.json()
self.assertEqual(container['ssh_keys'], keys)
"""
Domains: curl https://kratos:[email protected]/api/domains/
"""
def test_domains_get(self):
r = self.client.domains()
domains = r.json()
self.assertEqual(r.uerror, False)
def tests_domains_get_single(self):
r = self.client.domain(100)
self.assertEqual(r.status_code, 404)
def test_domains_add_delete(self):
domain = str(uuid.uuid4())
# add
r = self.client.add_domain(domain)
# lame but fails without settings the dns
self.assertEqual(r.uerror, True)
self.assertEqual("FORBIDDEN" in r.umessage, True)
# delete
# lame but delete fails if domain does not exist
self.client.delete_domain(domain)
self.assertEqual(r.uerror, True)
self.assertEqual("FORBIDDEN" in r.umessage, True)
r = self.client.domains()
domains = r.json()
found = [d for d in domains if d['name'] == domain]
self.assertEqual(found, [])
"""
Tags: curl https://kratos:[email protected]/api/tags/
"""
def test_tag_creation_delete(self):
r = self.client.create_tag("mytag")
tag_id = r.json()["id"]
r = self.client.delete_tag(tag_id)
r = self.client.list_tags()
tags = r.json()
found_tag = [t for t in tags if t['id'] == tag_id]
self.assertEqual(found_tag, [])
def test_tag_domain_filter(self):
r = self.client.domains(tags=["ciao"])
domains = r.json()
self.assertEqual(r.uerror, False)
def test_tag_container_filter(self):
r = self.client.containers(tags=["ciao"])
containers = r.json()
self.assertEqual(r.uerror, False)
"""
Reboot: curl -X POST -d '{"reboot":1}' https://kratos:[email protected]/api/containers/30009
"""
def test_reboot(self):
r = self.client.reboot_container(self.container)
self.assertEqual(r.uerror, False)