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 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"]))
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()
def get_transformer(self):
return BulkTransformer(
storage=self.storage,
crypto_controller=self.crypto_controller,
account_manager=self.account_manager,
)