aboutsummaryrefslogtreecommitdiffstats
path: root/test/adapters/test_sms.py
blob: ae9346d468e220be1fee8bc0ad9eb63d5b351209 (plain)
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()