def test_get_config_file_from_arguments(self):
js = SnapAdmin()
self.set_hardcoded_value_for_device(js)
js.args.testfiles = ["main.yml"]
js.get_config_file()
local_dict = {'hosts': [{'device': '1.1.1.1', 'username': '******', 'passwd': '123'}], 'tests': ['main.yml']}
self.assertEqual(js.main_file, local_dict)
def test_get_config_file_from_file_exact_path(self):
# config file is extracted when exact path is passed
js = SnapAdmin()
js.args.file = os.path.join(os.path.dirname(__file__),
'configs', 'main.yml')
js.get_config_file()
config_file = open(js.args.file, 'r')
config_file = yaml.load(config_file, Loader=yaml.FullLoader)
self.assertEqual(js.main_file, config_file)
def test_get_config_file_from_file_with_default_path(self, mock_path):
# config file to be appended with default path when file not present in the path.
mock_path.return_value = os.path.join(os.path.dirname(__file__), 'configs')
js = SnapAdmin()
js.args.file = "main.yml"
js.get_config_file()
conf_file = os.path.join(os.path.dirname(__file__),
'configs', 'main.yml')
config_file = open(conf_file, 'r')
config_file = yaml.load(config_file, Loader=yaml.FullLoader)
self.assertEqual(js.main_file, config_file)
def test_get_config_file_no_file(self, mock_exit):
js = SnapAdmin()
js.main_file = "hosts"
js.args.file = "file_not_found.yml"
js.get_config_file()
mock_exit.assert_called_with(1)