def test_supports_default(self):
cf = helper.build_config("""[defaults]
first=foo""")
self.assertEqual("defaults", cf.default_section)
cf = helper.build_config("""[defaults]
first=foo
[default]
second=bar""")
self.assertEqual("defaults", cf.default_section)
def test_is_initialised(self):
content = StringIO()
cf = ConfigurationFile(content)
self.assertFalse(cf.is_initialised)
cf = helper.build_config("""[section]
first=foo""")
self.assertTrue(cf.is_initialised)
cf = helper.build_config("""[defaults]
first=foo""")
self.assertTrue(cf.is_initialised)
def test_can_save_password_username_xor_2(self):
cfg = helper.build_config("""[defaults]
save_password = true
[profile_test]
save_password = false""")
self.assertFalse(cfg.section("profile_test").can_save_password)
def test_has_defaults(self):
content = StringIO()
cf = ConfigurationFile(content)
self.assertFalse(cf.has_defaults)
cf = helper.build_config("""[defaults]
first=foo""")
self.assertTrue(cf.has_defaults)
def test_load_password_promptuser(self):
cfg = helper.build_config("""[test-section]
username = mock_user
""")
sec = cfg.section("test-section")
creds = UserCredentials(sec)
creds._load_password_from_keychain = MagicMock()
creds._prompt_user_password = MagicMock()
creds._save_password_to_keychain = MagicMock()
creds.load_password()
creds._load_password_from_keychain.assert_not_called()
creds._save_password_to_keychain.assert_not_called()
creds._prompt_user_password.assert_called_once_with()
def test_save_password_success(self):
cfg = helper.build_config("""[test-section]
username = mock_user
save_password = true
""")
sec = cfg.section("test-section")
creds = UserCredentials(sec)
def create_password():
creds.password = "******"
creds._load_password_from_keychain = MagicMock()
creds._prompt_user_password = MagicMock(side_effect=create_password)
creds._save_password_to_keychain = MagicMock()
creds.load_password()
creds._load_password_from_keychain.assert_called_once_with()
creds._prompt_user_password.assert_called_once_with()
creds._save_password_to_keychain.assert_called_once_with()
def test_section_get(self):
cfg = helper.build_config("""[profile_test]
save_password = true""")
self.assertTrue(cfg.section("profile_test").get("save_password"))
def test_can_save_password_username_defaults_true(self):
cfg = helper.build_config("""[defaults]
save_password = true""")
self.assertTrue(cfg.section("defaults").can_save_password)
def test_can_save_password_username_inherit_false(self):
cfg = helper.build_config("""[defaults]
save_password = true
[profile_test]""")
self.assertTrue(cfg.section("profile_test").can_save_password)
def test_section_get_fallback(self):
cfg = helper.build_config("""[profile_test]
""")
self.assertFalse(cfg.section("profile_test").get("save_password"))
