File: test_container.py

package info (click to toggle)
custodia 0.6.0-5.2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 880 kB
  • sloc: python: 6,074; makefile: 317; sh: 62
file content (155 lines) | stat: -rw-r--r-- 6,178 bytes parent folder | download | duplicates (2)
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