371 lines
12 KiB
Python
371 lines
12 KiB
Python
import os
|
|
from pathlib import Path
|
|
from rougail import Rougail, RougailConfig
|
|
from rougail.error import DictConsistencyError
|
|
#########################
|
|
from rougail.user_data_bitwarden import RougailUserDataBitwarden as RougailUserData
|
|
from json import load, dump
|
|
#########################
|
|
from pytest import raises
|
|
from unittest import mock
|
|
|
|
|
|
from rougail_tests.utils import config_to_dict
|
|
|
|
|
|
test_dir = Path(__file__).parent / 'structures'
|
|
|
|
|
|
def _test_dictionaries(test_dir, command, env=False):
|
|
rougailconfig = RougailConfig.copy()
|
|
rougailconfig['main_dictionaries'] = [str(test_dir)]
|
|
# rougailconfig['tiramisu_cache'] = "cache.py"
|
|
rougailconfig['step.user_data'] = ['bitwarden']
|
|
rougailconfig['bitwarden.command'] = command
|
|
rougail = Rougail(rougailconfig)
|
|
config = rougail.run()
|
|
# loads variables in the tiramisu config
|
|
errors = RougailUserData(config, rougailconfig=rougailconfig).run()
|
|
#expected output
|
|
config_dict = dict(config_to_dict(config.forcepermissive.value.get()))
|
|
if not env:
|
|
base_filename = 'bitwarden.json'
|
|
else:
|
|
base_filename = 'bitwarden_env.json'
|
|
ok_file = Path('tests') / 'results' / test_dir.name / 'makedict' / (base_filename + '.' + command)
|
|
if not ok_file.is_file():
|
|
ok_file = Path('tests') / 'results' / test_dir.name / 'makedict' / base_filename
|
|
if not ok_file.is_file():
|
|
ok_file.parent.mkdir(parents=True, exist_ok=True)
|
|
with open(ok_file, 'a') as json_file:
|
|
dump(config_dict, json_file, indent=4)
|
|
with open(ok_file) as json_file:
|
|
expected = load(json_file)
|
|
errors_file = Path('tests') / 'results' / test_dir.name / 'errors' / (base_filename + '.' + command)
|
|
if not errors_file.is_file():
|
|
errors_file = Path('tests') / 'results' / test_dir.name / 'errors' / base_filename
|
|
if not errors_file.is_file():
|
|
errors_file.parent.mkdir(parents=True, exist_ok=True)
|
|
with open(errors_file, 'a') as json_file:
|
|
dump(errors, json_file, indent=4)
|
|
with open(errors_file) as json_file:
|
|
expected_errors = load(json_file)
|
|
# expected_errors = {
|
|
# 'errors': [],
|
|
# 'warnings': [],
|
|
# }
|
|
assert expected_errors == errors
|
|
#
|
|
assert expected == config_dict
|
|
|
|
|
|
def test_dictionaries_1_secret_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '1_secret', 'rbw')
|
|
|
|
|
|
def test_dictionaries_1_secret_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '1_secret', 'bw')
|
|
|
|
|
|
def test_dictionaries_1_secret_unknown_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '1_secret_unknown', 'rbw')
|
|
|
|
|
|
def test_dictionaries_1_secret_unknown_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '1_secret_unknown', 'bw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret', 'rbw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret', 'bw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_upper_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret_upper', 'rbw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_upper_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret_upper', 'bw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_invalid_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret_invalid', 'rbw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_invalid_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret_invalid', 'bw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_hidden_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret_hidden', 'rbw')
|
|
|
|
|
|
def test_dictionaries_2_username_secret_hidden_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '2_username_secret_hidden', 'bw')
|
|
|
|
|
|
def test_dictionaries_3_leadership_secret_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '3_leadership_secret', 'rbw')
|
|
|
|
|
|
def test_dictionaries_3_leadership_secret_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '3_leadership_secret', 'bw')
|
|
|
|
|
|
def test_dictionaries_3_leadership_secret_several_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '3_leadership_secret_several', 'rbw')
|
|
|
|
|
|
def test_dictionaries_3_leadership_secret_several_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '3_leadership_secret_several', 'bw')
|
|
|
|
|
|
def test_dictionaries_4_several_secrets_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '4_several_secrets', 'rbw')
|
|
|
|
|
|
def test_dictionaries_4_several_secrets_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '4_several_secrets', 'bw')
|
|
|
|
|
|
def test_dictionaries_4_several_secrets_upper_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '4_several_secrets_upper', 'rbw')
|
|
|
|
|
|
def test_dictionaries_4_several_secrets_upper_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '4_several_secrets_upper', 'bw')
|
|
|
|
|
|
def test_dictionaries_5_secret_calc_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '5_secret_calc', 'rbw')
|
|
|
|
|
|
def test_dictionaries_5_secret_calc_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '5_secret_calc', 'bw')
|
|
|
|
|
|
def test_dictionaries_6_no_username_rbw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '6_no_username', 'rbw')
|
|
|
|
|
|
def test_dictionaries_6_no_username_bw():
|
|
"tests the output"
|
|
_test_dictionaries(test_dir / '6_no_username', 'bw')
|
|
|
|
|
|
def test_dictionaries_8_multi_variable_rbw():
|
|
"tests the output"
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '8_multi_variable', 'rbw')
|
|
assert err.value.errno == 57
|
|
|
|
|
|
def test_dictionaries_8_multi_variable_bw():
|
|
"tests the output"
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '8_multi_variable', 'bw')
|
|
assert err.value.errno == 57
|
|
|
|
|
|
def test_dictionaries_9_unknown_type_rbw():
|
|
"tests the output"
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '9_unknown_type', 'rbw')
|
|
assert err.value.errno == 56
|
|
|
|
|
|
def test_dictionaries_9_unknown_type_bw():
|
|
"tests the output"
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '9_unknown_type', 'bw')
|
|
assert err.value.errno == 56
|
|
|
|
|
|
def test_dictionaries_env_1_secret_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '1_secret', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_1_secret_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '1_secret', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_1_secret_unknown_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '1_secret_unknown', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_1_secret_unknown_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '1_secret_unknown', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_upper_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret_upper', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_upper_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret_upper', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_invalid_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret_invalid', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_invalid_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret_invalid', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_hidden_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret_hidden', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_2_username_secret_hidden_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '2_username_secret_hidden', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_3_leadership_secret_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '3_leadership_secret', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_3_leadership_secret_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '3_leadership_secret', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_3_leadership_secret_several_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '3_leadership_secret_several', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_3_leadership_secret_several_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '3_leadership_secret_several', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_4_several_secrets_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '4_several_secrets', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_4_several_secrets_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '4_several_secrets', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_4_several_secrets_upper_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '4_several_secrets_upper', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_4_several_secrets_upper_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '4_several_secrets_upper', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_5_secret_calc_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '5_secret_calc', 'rbw', True)
|
|
|
|
|
|
def test_dictionaries_env_5_secret_calc_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
_test_dictionaries(test_dir / '5_secret_calc', 'bw', True)
|
|
|
|
|
|
def test_dictionaries_env_8_multi_variable_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '8_multi_variable', 'rbw', True)
|
|
assert err.value.errno == 57
|
|
|
|
|
|
def test_dictionaries_env_8_multi_variable_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '8_multi_variable', 'bw', True)
|
|
assert err.value.errno == 57
|
|
|
|
|
|
def test_dictionaries_env_9_unknown_type_rbw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '9_unknown_type', 'rbw', True)
|
|
assert err.value.errno == 56
|
|
|
|
|
|
def test_dictionaries_env_9_unknown_type_bw():
|
|
"tests the output"
|
|
with mock.patch.dict(os.environ, {'ROUGAIL_BITWARDEN_MOCK_ENABLE': '1'}):
|
|
with raises(DictConsistencyError) as err:
|
|
_test_dictionaries(test_dir / '9_unknown_type', 'bw', True)
|
|
assert err.value.errno == 56
|