1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
|
# Copyright (C) 2017 Custodia Project Contributors - see LICENSE file
from __future__ import absolute_import
import json
from .base import CustodiaServerRunner
class TestContainer(CustodiaServerRunner):
def test_create_container(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
def test_create_container_again(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 200
def test_create_container_invalid_format(self, custodia_server):
invalid_container = 'secrets/bucket{}'.format(self.get_unique_number())
resp = custodia_server.post(invalid_container,
headers=self.request_headers)
assert resp.status_code == 405
def test_create_container_forbidden_key(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
resp = custodia_server.post(container, headers={})
assert resp.status_code == 403
def test_list_container(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
mykey = '{}mykey'.format(container)
yourkey = '{}yourkey'.format(container)
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
assert resp.text == '[]'
resp = custodia_server.put(mykey, json={"type": "simple",
"value": 'P@ssw0rd'},
headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
data = json.loads(resp.text)
assert 'mykey' in data
resp = custodia_server.put(yourkey, json={"type": "simple",
"value": 'AnotherP@ssw0rd'},
headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
data = json.loads(resp.text)
assert 'mykey' in data
assert 'yourkey' in data
def test_remove_container(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
assert resp.text == '[]'
resp = custodia_server.delete(container, headers=self.request_headers)
assert resp.status_code == 204
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 404
def test_remove_container_not_empty(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
mykey = '{}mykey'.format(container)
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.put(mykey, json={"type": "simple",
"value": 'P@ssw0rd'},
headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
data = json.loads(resp.text)
assert 'mykey' in data
resp = custodia_server.delete(container, headers=self.request_headers)
assert resp.status_code == 409
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
data = json.loads(resp.text)
assert 'mykey' in data
def test_remove_container_not_found(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
assert resp.text == '[]'
resp = custodia_server.delete(container, headers=self.request_headers)
assert resp.status_code == 204
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 404
resp = custodia_server.delete(container, headers=self.request_headers)
assert resp.status_code == 404
def test_remove_container_forbidden_key(self, custodia_server):
container = 'secrets/bucket{}/'.format(self.get_unique_number())
mykey = '{}mykey'.format(container)
resp = custodia_server.post(container, headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.put(mykey, json={"type": "simple",
"value": 'P@ssw0rd'},
headers=self.request_headers)
assert resp.status_code == 201
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
data = json.loads(resp.text)
assert 'mykey' in data
resp = custodia_server.delete(container, headers={})
assert resp.status_code == 403
resp = custodia_server.get(container, headers=self.request_headers)
assert resp.status_code == 200
data = json.loads(resp.text)
assert 'mykey' in data
|