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
|
#
# -------------------------------------------------------------------------
# Copyright (c) 2018 Intel Corporation Intellectual Property
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file 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.
#
# -------------------------------------------------------------------------
#
import unittest
from uuid import uuid4
import requests_mock
import osdf.adapters.aaf.sms as SMS
from osdf.config.base import osdf_config
class TestSMS(unittest.TestCase):
def setUp(self):
self.config = osdf_config.deployment
self.base_domain_url = '{}/v1/sms/domain'
self.domain_url = '{}/v1/sms/domain/{}'
self.secret_url = self.domain_url + '/secret'
@requests_mock.mock()
def test_sms(self, mock_sms):
''' NOTE: preload_secret during the deployment using a preload script.
For test purposes we need to do preload ourselves'''
sms_url = self.config["aaf_sms_url"]
# JSON Data responses
secretnames = {'secretnames': ['s1', 's2', 's3', 's4']}
secretvalues = {'values': {'Password': '', 'UserName': ''}}
expecect_secret_dict = dict()
for secret in secretnames['secretnames']:
expecect_secret_dict[secret] = secretvalues['values']
# Part 1 : Preload Secrets ONLY FOR TEST
# Mock requests for preload_secret
cd_url = self.base_domain_url.format(sms_url)
domain_uuid1 = str(uuid4())
domain_name = self.config['secret_domain']
s_url = self.secret_url.format(sms_url, domain_name)
mock_sms.post(cd_url, status_code=200, json={'uuid': domain_uuid1})
mock_sms.post(s_url, status_code=200)
# Initialize Secrets from SMS
SMS.preload_secrets()
# Part 2: Retrieve Secret Test
# Mock requests for retrieve_secrets
d_url = self.domain_url.format(sms_url, domain_name)
s_url = self.secret_url.format(sms_url, domain_name)
# Retrieve Secrets from SMS and load to secret cache
# Use the secret_cache instead of config files
mock_sms.get(s_url, status_code=200, json=secretnames)
for secret in secretnames['secretnames']:
mock_sms.get('{}/{}'.format(s_url, secret),
status_code=200, json=secretvalues)
secret_cache = SMS.retrieve_secrets()
self.assertDictEqual(expecect_secret_dict, secret_cache,
'Failed to retrieve secrets')
# Part 3: Clean up Delete secrets and domain
# Mock requests for delete_secrets
mock_sms.delete(d_url, status_code=200)
self.assertTrue(SMS.delete_secrets())
if __name__ == "__main__":
unittest.main()
|
