def setUp(self):
super(BulkTransformerTest, self).setUp()
self.cryptor = generate_cryptor(**config_factory('1'))
self.crypto_controller = CryptoController(self.cryptor)
self.app_tempdir = tempfile.mkdtemp()
config = Config(Mock(**{'app.directory_path': self.app_tempdir}))
self.account_manager = AccountManager(config)
def setUp(self):
super(BulkTransformerTest, self).setUp()
self.cryptor = generate_cryptor(**config_factory("1"))
self.crypto_controller = CryptoController(self.cryptor)
self.app_tempdir = tempfile.mkdtemp()
config = Config(Mock(**{"app.directory_path": self.app_tempdir}))
self.account_manager = AccountManager(config)
def test_get_analytics_id_when_exists(self):
aid = 'id'
config_mock = Mock()
config_mock.get.return_value = aid
manager = AccountManager(config_mock)
self.assertEquals(aid, manager.analytics_id)
config_mock.get.assert_called_once_with('User', 'analytics_id')
def __init__(self, app, config=None):
"""Construct new Analytics instance."""
self.config = config
self.app = app
if self.config is None:
self.config = Config(self)
self.manager = AccountManager(self.config)
def test_get_analytics_id_when_not_exists(self, uuid):
aid = 'id'
uuid.uuid4.return_value = aid
config_mock = Mock()
config_mock.get.side_effect = configparser.NoSectionError('msg')
manager = AccountManager(config_mock)
self.assertEquals(aid, manager.analytics_id)
config_mock.set.assert_called_once_with('User', 'analytics_id', aid)
config_mock.write.assert_called_once()
def get_manager(self, config_side_effect):
config_mock = Mock(**{'get_safe.side_effect': config_side_effect})
return AccountManager(config_mock)
class BulkTransformerTest(StrategyCase):
maxDiff = None
get_strategy_class = RelatedGetStrategy
save_strategy_class = SyncSaveStrategy
def setUp(self):
super(BulkTransformerTest, self).setUp()
self.cryptor = generate_cryptor(**config_factory("1"))
self.crypto_controller = CryptoController(self.cryptor)
self.app_tempdir = tempfile.mkdtemp()
config = Config(Mock(**{"app.directory_path": self.app_tempdir}))
self.account_manager = AccountManager(config)
def tearDown(self):
super(BulkTransformerTest, self).tearDown()
self._clean_dir(Path(self.app_tempdir))
def test_create_transformer_sync_key(self):
self.transformer = BulkTransformer(
storage=self.storage, crypto_controller=self.crypto_controller, account_manager=self.account_manager
)
ssh_config = SshConfig()
identity = Identity(label="identity")
ssh_key = SshKey(label="ssh_key")
host = Host(label="host")
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced="")
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(
payload,
{
"host_set": [
{
"label": payload["host_set"][0]["label"],
"group": None,
"ssh_config": "sshconfig_set/{}".format(ssh_config.id),
"local_id": host.id,
"address": "",
"interaction_date": host.interaction_date,
}
],
"sshconfig_set": [
{
"font_size": None,
"keep_alive_packages": None,
"charset": None,
"identity": "identity_set/{}".format(identity.id),
"local_id": ssh_config.id,
"use_ssh_key": None,
"timeout": None,
"color_scheme": None,
"is_forward_ports": None,
"strict_host_key_check": None,
"port": None,
"startup_snippet": None,
"cursor_blink": None,
}
],
"snippet_set": [],
"last_synced": "",
"sshkeycrypt_set": [
{
"public_key": "",
"private_key": "",
"local_id": ssh_key.id,
"passphrase": "",
"label": payload["sshkeycrypt_set"][0]["label"],
}
],
"group_set": [],
"tag_set": [],
"taghost_set": [],
"pfrule_set": [],
"delete_sets": {},
"identity_set": [
{
"username": "",
"is_visible": False,
"ssh_key": "sshkeycrypt_set/{}".format(ssh_key.id),
"label": payload["identity_set"][0]["label"],
"local_id": identity.id,
"password": "",
}
],
},
)
eq_(host.label, self.cryptor.decrypt(payload["host_set"][0]["label"]))
eq_(ssh_key.label, self.cryptor.decrypt(payload["sshkeycrypt_set"][0]["label"]))
eq_(identity.label, self.cryptor.decrypt(payload["identity_set"][0]["label"]))
def test_create_transformer_nosync_key(self):
self.account_manager.set_settings({"synchronize_key": False, "agent_forwarding": True})
self.transformer = BulkTransformer(
storage=self.storage, crypto_controller=self.crypto_controller, account_manager=self.account_manager
)
ssh_config = SshConfig()
identity = Identity(label="identity")
ssh_key = SshKey(label="ssh_key")
host = Host(label="host")
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced="")
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(
payload,
{
"host_set": [
{
"label": payload["host_set"][0]["label"],
"group": None,
"ssh_config": "sshconfig_set/{}".format(ssh_config.id),
"local_id": host.id,
"address": "",
"interaction_date": host.interaction_date,
}
],
"sshconfig_set": [
{
"font_size": None,
"keep_alive_packages": None,
"charset": None,
"local_id": ssh_config.id,
"use_ssh_key": None,
"timeout": None,
"color_scheme": None,
"is_forward_ports": None,
"strict_host_key_check": None,
"port": None,
"startup_snippet": None,
"cursor_blink": None,
}
],
"snippet_set": [],
"last_synced": "",
"group_set": [],
"tag_set": [],
"taghost_set": [],
"pfrule_set": [],
"delete_sets": {},
},
)
eq_(host.label, self.cryptor.decrypt(payload["host_set"][0]["label"]))
def _clean_dir(self, dir_path):
[self._clean_dir(i) for i in dir_path.iterdir() if i.is_dir()]
[i.unlink() for i in dir_path.iterdir() if i.is_file()]
dir_path.rmdir()
class BulkTransformerTest(StrategyCase):
maxDiff = None
get_strategy_class = RelatedGetStrategy
save_strategy_class = SyncSaveStrategy
def setUp(self):
super(BulkTransformerTest, self).setUp()
self.cryptor = generate_cryptor(**config_factory('1'))
self.crypto_controller = CryptoController(self.cryptor)
self.app_tempdir = tempfile.mkdtemp()
config = Config(Mock(**{'app.directory_path': self.app_tempdir}))
self.account_manager = AccountManager(config)
def tearDown(self):
super(BulkTransformerTest, self).tearDown()
self._clean_dir(Path(self.app_tempdir))
def test_create_transformer_sync_key(self):
self.transformer = BulkTransformer(
storage=self.storage,
crypto_controller=self.crypto_controller,
account_manager=self.account_manager,
)
ssh_config = SshConfig()
identity = Identity(label='identity')
ssh_key = SshKey(label='ssh_key')
host = Host(label='host')
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced='')
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(
payload, {
'host_set':
[{
'label': payload['host_set'][0]['label'],
'group': None,
'ssh_config': 'sshconfig_set/{}'.format(ssh_config.id),
'local_id': host.id,
'address': '',
'interaction_date': host.interaction_date,
}],
'sshconfig_set':
[{
'font_size': None,
'keep_alive_packages': None,
'charset': None,
'identity': 'identity_set/{}'.format(identity.id),
'local_id': ssh_config.id,
'use_ssh_key': None,
'timeout': None,
'color_scheme': None,
'is_forward_ports': None,
'strict_host_key_check': None,
'port': None,
'startup_snippet': None,
'cursor_blink': None
}],
'snippet_set': [],
'last_synced':
'',
'sshkeycrypt_set': [
{
'public_key': '',
'private_key': '',
'local_id': ssh_key.id,
'passphrase': '',
'label': payload['sshkeycrypt_set'][0]['label']
}
],
'group_set': [],
'tag_set': [],
'taghost_set': [],
'pfrule_set': [],
'delete_sets': {},
'identity_set': [
{
'username': '',
'is_visible': False,
'ssh_key': 'sshkeycrypt_set/{}'.format(ssh_key.id),
'label': payload['identity_set'][0]['label'],
'local_id': identity.id,
'password': ''
}
]
})
eq_(host.label, self.cryptor.decrypt(payload['host_set'][0]['label']))
eq_(ssh_key.label,
self.cryptor.decrypt(payload['sshkeycrypt_set'][0]['label']))
eq_(identity.label,
self.cryptor.decrypt(payload['identity_set'][0]['label']))
def test_create_transformer_nosync_key(self):
self.account_manager.set_settings({
'synchronize_key': False,
'agent_forwarding': True
})
self.transformer = BulkTransformer(
storage=self.storage,
crypto_controller=self.crypto_controller,
account_manager=self.account_manager,
)
ssh_config = SshConfig()
identity = Identity(label='identity')
ssh_key = SshKey(label='ssh_key')
host = Host(label='host')
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced='')
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(
payload, {
'host_set':
[{
'label': payload['host_set'][0]['label'],
'group': None,
'ssh_config': 'sshconfig_set/{}'.format(ssh_config.id),
'local_id': host.id,
'address': '',
'interaction_date': host.interaction_date,
}],
'sshconfig_set': [{
'font_size': None,
'keep_alive_packages': None,
'charset': None,
'local_id': ssh_config.id,
'use_ssh_key': None,
'timeout': None,
'color_scheme': None,
'is_forward_ports': None,
'strict_host_key_check': None,
'port': None,
'startup_snippet': None,
'cursor_blink': None
}],
'snippet_set': [],
'last_synced':
'',
'group_set': [],
'tag_set': [],
'taghost_set': [],
'pfrule_set': [],
'delete_sets': {},
})
eq_(host.label, self.cryptor.decrypt(payload['host_set'][0]['label']))
def _clean_dir(self, dir_path):
[self._clean_dir(i) for i in dir_path.iterdir() if i.is_dir()]
[i.unlink() for i in dir_path.iterdir() if i.is_file()]
dir_path.rmdir()
class BulkTransformerTest(StrategyCase):
maxDiff = None
get_strategy_class = RelatedGetStrategy
save_strategy_class = SyncSaveStrategy
def setUp(self):
super(BulkTransformerTest, self).setUp()
self.cryptor = generate_cryptor(**config_factory('1'))
self.crypto_controller = CryptoController(self.cryptor)
self.app_tempdir = tempfile.mkdtemp()
config = Config(Mock(**{'app.directory_path': self.app_tempdir}))
self.account_manager = AccountManager(config)
def tearDown(self):
super(BulkTransformerTest, self).tearDown()
self._clean_dir(Path(self.app_tempdir))
def test_create_transformer_sync_key(self):
self.transformer = self.get_transformer()
ssh_config = SshConfig()
identity = Identity(label='identity')
ssh_key = SshKey(label='ssh_key')
host = Host(label='host')
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced='')
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(payload, {
'host_set': [
{
'label': payload['host_set'][0]['label'],
'group': None,
'ssh_config': 'sshconfig_set/{}'.format(ssh_config.id),
'local_id': host.id,
'address': None,
'interaction_date': host.interaction_date,
}
],
'sshconfig_set': [
{
'font_size': None,
'keep_alive_packages': None,
'charset': None,
'identity': 'identity_set/{}'.format(identity.id),
'local_id': ssh_config.id,
'use_ssh_key': None,
'timeout': None,
'color_scheme': None,
'is_forward_ports': None,
'strict_host_key_check': None,
'port': None,
'startup_snippet': None,
'cursor_blink': None
}
],
'snippet_set': [],
'last_synced': '',
'sshkeycrypt_set': [
{
'public_key': None,
'private_key': None,
'local_id': ssh_key.id,
'passphrase': None,
'label': payload['sshkeycrypt_set'][0]['label']
}
],
'group_set': [],
'tag_set': [],
'taghost_set': [],
'pfrule_set': [],
'delete_sets': {},
'identity_set': [
{
'username': None,
'is_visible': None,
'ssh_key': 'sshkeycrypt_set/{}'.format(ssh_key.id),
'label': payload['identity_set'][0]['label'],
'local_id': identity.id,
'password': None
}
]
})
eq_(host.label, self.cryptor.decrypt(
payload['host_set'][0]['label']
))
eq_(ssh_key.label, self.cryptor.decrypt(
payload['sshkeycrypt_set'][0]['label']
))
eq_(identity.label, self.cryptor.decrypt(
payload['identity_set'][0]['label']
))
def test_create_transformer_nosync_key(self):
self.account_manager.set_settings({
'synchronize_key': False, 'agent_forwarding': True
})
self.transformer = self.get_transformer()
ssh_config = SshConfig()
identity = Identity(label='identity')
ssh_key = SshKey(label='ssh_key')
host = Host(label='host')
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced='')
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(payload, {
'host_set': [
{
'label': payload['host_set'][0]['label'],
'group': None,
'ssh_config': 'sshconfig_set/{}'.format(ssh_config.id),
'local_id': host.id,
'address': None,
'interaction_date': host.interaction_date,
}
],
'sshconfig_set': [
{
'font_size': None,
'keep_alive_packages': None,
'charset': None,
'local_id': ssh_config.id,
'use_ssh_key': None,
'timeout': None,
'color_scheme': None,
'is_forward_ports': None,
'strict_host_key_check': None,
'port': None,
'startup_snippet': None,
'cursor_blink': None
}
],
'snippet_set': [],
'last_synced': '',
'group_set': [],
'tag_set': [],
'taghost_set': [],
'pfrule_set': [],
'delete_sets': {},
})
eq_(host.label, self.cryptor.decrypt(
payload['host_set'][0]['label']
))
def test_bad_encrytped_data_no_in_storage(self):
self.transformer = self.get_transformer()
last_synced_data = dict(
now='',
deleted_sets=self.empty_set(),
**self.empty_set()
)
last_synced_data['host_set'] = [
{
'id': id,
'label': i,
'interaction_date': '',
'address': None,
'group': None,
'ssh_config': None,
}
for id, i in enumerate(['not encrypted plain text',
'123' + self.cryptor.encrypt('1')])
]
payload = self.transformer.to_model(last_synced_data)
self.assertEqual(payload['host_set'], [])
self.assertEqual(
self.storage.strategies.deleter.get_delete_sets(),
{'host_set': [0, 1]}
)
def test_bad_encrytped_data_in_storage(self):
self.transformer = self.get_transformer()
last_synced_data = dict(
now='',
deleted_sets=self.empty_set(),
**self.empty_set()
)
first_host, second_host = [
self.storage.save(Host(
address='host', remote_instance=RemoteInstance(id=i)
)) for i in range(2)
]
last_synced_data['host_set'] = [
{
'id': i.remote_instance.id,
'label': label,
'interaction_date': '',
'address': None,
'group': None,
'ssh_config': None,
}
for i, label in zip(
[first_host, second_host],
['not encrypted plain text', '13' + self.cryptor.encrypt('1')]
)
]
payload = self.transformer.to_model(last_synced_data)
self.assertEqual(payload['host_set'], [])
self.assertEqual(
self.storage.strategies.deleter.get_delete_sets(),
{'host_set': [0, 1]}
)
self.assertEqual(self.storage.get_all(Host), [])
def _clean_dir(self, dir_path):
[self._clean_dir(i) for i in dir_path.iterdir() if i.is_dir()]
[i.unlink() for i in dir_path.iterdir() if i.is_file()]
dir_path.rmdir()
def get_transformer(self):
return BulkTransformer(
storage=self.storage,
crypto_controller=self.crypto_controller,
account_manager=self.account_manager,
)
def empty_set(self):
return dict(
snippet_set=[],
group_set=[],
tag_set=[],
taghost_set=[],
pfrule_set=[],
identity_set=[],
sshkeycrypt_set=[],
sshconfig_set=[],
host_set=[],
)
class BulkTransformerTest(StrategyCase):
maxDiff = None
get_strategy_class = RelatedGetStrategy
save_strategy_class = SyncSaveStrategy
def setUp(self):
super(BulkTransformerTest, self).setUp()
self.cryptor = generate_cryptor(**config_factory('1'))
self.crypto_controller = CryptoController(self.cryptor)
self.app_tempdir = tempfile.mkdtemp()
config = Config(Mock(**{'app.directory_path': self.app_tempdir}))
self.account_manager = AccountManager(config)
def tearDown(self):
super(BulkTransformerTest, self).tearDown()
self._clean_dir(Path(self.app_tempdir))
def test_create_transformer_sync_key(self):
self.transformer = self.get_transformer()
ssh_config = SshConfig()
identity = Identity(label='identity')
ssh_key = SshKey(label='ssh_key')
host = Host(label='host')
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced='')
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(
payload, {
'host_set':
[{
'label': payload['host_set'][0]['label'],
'group': None,
'ssh_config': 'sshconfig_set/{}'.format(ssh_config.id),
'local_id': host.id,
'address': None,
'interaction_date': host.interaction_date,
}],
'sshconfig_set':
[{
'font_size': None,
'keep_alive_packages': None,
'charset': None,
'identity': 'identity_set/{}'.format(identity.id),
'local_id': ssh_config.id,
'use_ssh_key': None,
'timeout': None,
'color_scheme': None,
'is_forward_ports': None,
'strict_host_key_check': None,
'port': None,
'startup_snippet': None,
'cursor_blink': None
}],
'snippet_set': [],
'last_synced':
'',
'sshkeycrypt_set': [
{
'public_key': None,
'private_key': None,
'local_id': ssh_key.id,
'label': payload['sshkeycrypt_set'][0]['label']
}
],
'group_set': [],
'tag_set': [],
'taghost_set': [],
'pfrule_set': [],
'delete_sets': {},
'identity_set': [
{
'username': None,
'is_visible': None,
'ssh_key': 'sshkeycrypt_set/{}'.format(ssh_key.id),
'label': payload['identity_set'][0]['label'],
'local_id': identity.id
}
]
})
eq_(host.label, self.cryptor.decrypt(payload['host_set'][0]['label']))
eq_(ssh_key.label,
self.cryptor.decrypt(payload['sshkeycrypt_set'][0]['label']))
eq_(identity.label,
self.cryptor.decrypt(payload['identity_set'][0]['label']))
def test_create_transformer_nosync_key(self):
self.account_manager.set_settings({
'synchronize_key': False,
'agent_forwarding': True
})
self.transformer = self.get_transformer()
ssh_config = SshConfig()
identity = Identity(label='identity')
ssh_key = SshKey(label='ssh_key')
host = Host(label='host')
identity.ssh_key = ssh_key.id = self.storage.save(ssh_key).id
ssh_config.identity = identity.id = self.storage.save(identity).id
host.ssh_config = ssh_config.id = self.storage.save(ssh_config).id
host = self.storage.save(host)
last_synced_data = dict(last_synced='')
payload = self.transformer.to_payload(last_synced_data)
self.assertEqual(
payload, {
'host_set':
[{
'label': payload['host_set'][0]['label'],
'group': None,
'ssh_config': 'sshconfig_set/{}'.format(ssh_config.id),
'local_id': host.id,
'address': None,
'interaction_date': host.interaction_date,
}],
'sshconfig_set': [{
'font_size': None,
'keep_alive_packages': None,
'charset': None,
'local_id': ssh_config.id,
'use_ssh_key': None,
'timeout': None,
'color_scheme': None,
'is_forward_ports': None,
'strict_host_key_check': None,
'port': None,
'startup_snippet': None,
'cursor_blink': None
}],
'snippet_set': [],
'last_synced':
'',
'group_set': [],
'tag_set': [],
'taghost_set': [],
'pfrule_set': [],
'delete_sets': {},
})
eq_(host.label, self.cryptor.decrypt(payload['host_set'][0]['label']))
def test_bad_encrytped_data_no_in_storage(self):
self.transformer = self.get_transformer()
last_synced_data = dict(now='',
deleted_sets=self.empty_set(),
**self.empty_set())
last_synced_data['host_set'] = [{
'id': id,
'label': i,
'interaction_date': '',
'address': None,
'group': None,
'ssh_config': None,
} for id, i in enumerate(
['not encrypted plain text', '123' + self.cryptor.encrypt('1')])]
payload = self.transformer.to_model(last_synced_data)
self.assertEqual(payload['host_set'], [])
self.assertEqual(self.storage.strategies.deleter.get_delete_sets(),
{'host_set': [0, 1]})
def test_bad_encrytped_data_in_storage(self):
self.transformer = self.get_transformer()
last_synced_data = dict(now='',
deleted_sets=self.empty_set(),
**self.empty_set())
first_host, second_host = [
self.storage.save(
Host(address='host', remote_instance=RemoteInstance(id=i)))
for i in range(2)
]
last_synced_data['host_set'] = [{
'id': i.remote_instance.id,
'label': label,
'interaction_date': '',
'address': None,
'group': None,
'ssh_config': None,
} for i, label in zip(
[first_host, second_host],
['not encrypted plain text', '13' + self.cryptor.encrypt('1')])]
payload = self.transformer.to_model(last_synced_data)
self.assertEqual(payload['host_set'], [])
self.assertEqual(self.storage.strategies.deleter.get_delete_sets(),
{'host_set': [0, 1]})
self.assertEqual(self.storage.get_all(Host), [])
def _clean_dir(self, dir_path):
[self._clean_dir(i) for i in dir_path.iterdir() if i.is_dir()]
[i.unlink() for i in dir_path.iterdir() if i.is_file()]
dir_path.rmdir()
def get_transformer(self):
return BulkTransformer(
storage=self.storage,
crypto_controller=self.crypto_controller,
account_manager=self.account_manager,
)
def empty_set(self):
return dict(
snippet_set=[],
group_set=[],
tag_set=[],
taghost_set=[],
pfrule_set=[],
identity_set=[],
sshkeycrypt_set=[],
sshconfig_set=[],
host_set=[],
)
