コード例 #1
0
ファイル: configlib_test.py プロジェクト: willoughbyrm/magma
    def test_configure_set(self, mock_check_call, mock_get_input):
        mock_vals = {
            'nms_db_password': '******',
        }
        mock_get_input.side_effect = [
            mock_vals['nms_db_password'],
        ]

        # configure platform
        mgr = configlib.ConfigManager(self.constants)
        mgr.configure('platform')

        # check if we only invoked input for nms_db_password
        self.assertEqual(mock_get_input.call_count, 1)
        fn = "%s/platform.tfvars.json" % self.constants['config_dir']
        cfg = configlib.get_json(fn)
        self.assertEqual(len(cfg.keys()), 1)
        self.assertEqual(cfg['nms_db_password'], "foo")

        mgr.set('platform', 'deploy_elastic', 'true')
        cfg = configlib.get_json(fn)
        self.assertEqual(len(cfg.keys()), 2)
        self.assertEqual(cfg['deploy_elastic'], "true")

        # finally verify if all configs required by tf is present
        cfg = configlib.get_json(self.constants['auto_tf'])
        self.assertEqual(len(cfg.keys()), 2)
        self.assertEqual(cfg['nms_db_password'], "foo")
        self.assertEqual(cfg['deploy_elastic'], "true")
コード例 #2
0
ファイル: configlib_test.py プロジェクト: willoughbyrm/magma
    def test_configure_sanity(self, mock_check_call, mock_get_input):
        mock_vals = {
            'aws_access_key_id': 'foo',
            'aws_secret_access_key': 'bar',
            'secretsmanager_orc8r_secret': 'jar',
        }
        mock_get_input.side_effect = [
            mock_vals['aws_access_key_id'],
            mock_vals['aws_secret_access_key'],
            mock_vals['secretsmanager_orc8r_secret'],
        ]
        # verify if components tfvars json is created
        mgr = configlib.ConfigManager(self.constants)
        mgr.configure('infra')

        # verify if configs are set in infra tfvars json
        fn = "%s/infra.tfvars.json" % self.constants['config_dir']
        cfg = configlib.get_json(fn)
        self.assertEqual(len(cfg.keys()), 3)
        self.assertEqual(cfg['aws_access_key_id'], "foo")
        self.assertEqual(cfg['aws_secret_access_key'], "bar")
        self.assertEqual(cfg['secretsmanager_orc8r_secret'], "jar")

        # check if aws configs are set
        aws_config_cmd = ['aws', 'configure', 'set']
        mock_check_call.assert_any_call(
            aws_config_cmd + ['aws_access_key_id', 'foo'])
        mock_check_call.assert_any_call(
            aws_config_cmd + ['aws_secret_access_key', 'bar'])

        # verify that platform tfvars json file isn't present
        fn = "%s/platform.tfvars.json" % self.constants['config_dir']
        self.assertEqual(os.path.isfile(fn), False)

        # reset mocks
        mock_get_input.reset_mock()
        mock_check_call.reset_mock()

        mock_vals = {
            'nms_db_password': '******',
        }
        mock_get_input.side_effect = [
            mock_vals['nms_db_password'],
        ]

        # configure platform
        mgr.configure('platform')

        # verify that no aws call was invoked
        self.assertEqual(mock_check_call.call_count, 0)

        # check if we only invoked input for nms_db_password
        self.assertEqual(mock_get_input.call_count, 1)
        fn = "%s/platform.tfvars.json" % self.constants['config_dir']
        cfg = configlib.get_json(fn)
        self.assertEqual(len(cfg.keys()), 1)
        self.assertEqual(cfg['nms_db_password'], "foo")


        # verify that service tfvars json file isn't present
        fn = "%s/service.tfvars.json" % self.constants['config_dir']
        self.assertEqual(os.path.isfile(fn), False)

        # reset mocks
        mock_get_input.reset_mock()
        mock_check_call.reset_mock()

        # configure service
        mgr.configure('service')

        # verify that no input or aws call was invoked
        self.assertEqual(mock_check_call.call_count, 0)
        self.assertEqual(mock_get_input.call_count, 0)

        fn = "%s/service.tfvars.json" % self.constants['config_dir']
        cfg = configlib.get_json(fn)
        # verify that default value was set
        self.assertEqual(len(cfg.keys()), 1)
        self.assertEqual(cfg['lte_orc8r_chart_version'], "0.2.4")

        # finally verify if all configs required by tf is present
        cfg = configlib.get_json(self.constants['auto_tf'])
        self.assertEqual(len(cfg.keys()), 3)
        self.assertEqual(cfg['secretsmanager_orc8r_secret'], "jar")
        self.assertEqual(cfg['nms_db_password'], "foo")
        self.assertEqual(cfg['lte_orc8r_chart_version'], "0.2.4")

        # verify if jinja template has been rendered accordingly
        with open(self.constants['main_tf']) as f:
            jinja_cfg = dict(ln.split('=') for ln in f.readlines() if ln.strip())

        # all infra terraform keys should be present in the jinja template
        self.assertEqual(
            set(jinja_cfg.keys()),
            set(['secretsmanager_orc8r_secret']))

        # variable tf is of the form variable "var_name" {}
        # get the middle element and remove the quotes
        with open(self.constants['vars_tf']) as f:
            jinja_cfg = [ln.split()[1][1:-1] for ln in f.readlines() if ln.strip()]

        # all infra terraform keys should be present in the jinja template
        self.assertEqual(set(jinja_cfg), set(mgr.tf_vars))