def test_copy_dir(self):
dir = self.src.directories.create({
'description': uuid4().hex,
'name': uuid4().hex,
'status': 'DISABLED',
'custom_data': {'hi': 'there'},
})
migrator = DirectoryMigrator(destination_client=self.dst, source_directory=dir)
copied_dir = migrator.copy_dir()
self.assertTrue(copied_dir)
self.assertEqual(dir.description, copied_dir.description)
self.assertEqual(dir.name, copied_dir.name)
self.assertEqual(dir.provider.provider_id, copied_dir.provider.provider_id)
self.assertEqual(dir.status, copied_dir.status)
dir.delete()
copied_dir.delete()
dir = self.src.directories.create({
'description': uuid4().hex,
'name': uuid4().hex,
'status': 'DISABLED',
'provider': {
'provider_id': 'google',
'client_id': 'test',
'client_secret': 'test',
'redirect_uri': 'https://test.com/callback',
}
})
migrator = DirectoryMigrator(destination_client=self.dst, source_directory=dir)
copied_dir = migrator.copy_dir()
self.assertTrue(copied_dir)
self.assertEqual(dir.description, copied_dir.description)
self.assertEqual(dir.name, copied_dir.name)
self.assertEqual(dir.provider.provider_id, copied_dir.provider.provider_id)
self.assertEqual(dir.provider.client_id, copied_dir.provider.client_id)
self.assertEqual(dir.provider.client_secret, copied_dir.provider.client_secret)
self.assertEqual(dir.provider.redirect_uri, copied_dir.provider.redirect_uri)
self.assertEqual(dir.status, copied_dir.status)
dir.delete()
copied_dir.delete()
def test_copy_custom_data(self):
dir = self.src.directories.create({
'custom_data': {'hi': 'there'},
'description': uuid4().hex,
'name': uuid4().hex,
'status': 'DISABLED',
})
migrator = DirectoryMigrator(destination_client=self.dst, source_directory=dir)
copied_dir = migrator.copy_dir()
copied_data = migrator.copy_custom_data()
self.assertEqual(copied_data['hi'], 'there')
dir.delete()
copied_dir.delete()
def test_copy_strength(self):
dir = self.src.directories.create({'name': uuid4().hex})
strength = dir.password_policy.strength
strength.max_length = 101
strength.save()
migrator = DirectoryMigrator(destination_client=self.dst, source_directory=dir)
copied_dir = migrator.copy_dir()
copied_strength = migrator.copy_strength()
for key in copied_strength.writable_attrs:
self.assertEqual(copied_strength[key], strength[key])
dir.delete()
copied_dir.delete()
