def login_as_user(self, user=None, password='******', *args, **kwargs): if not user: user = UserFactory(*args, **kwargs) user.set_password(password) user.save() self.user = user self.client.login(username=user.username, password=password)
def setUp(self): sample_role = VentureRoleFactory() RolePropertyFactory( symbol='my_custom_property_1', type=RolePropertyTypeFactory(symbol='STRING'), role=sample_role, ) self.sample_device = DeviceFactory(venture=sample_role.venture, venture_role=sample_role) self.sample_user = UserFactory()
def setUp(self): self.user = UserFactory( username='******', email='*****@*****.**', is_staff=True, is_superuser=True, ) self.user.set_password('QWEasd') self.user.save() self.net_1 = NetworkFactory( name='test-net-1.dc', address="192.168.1.0/24", ) self.net_1.save() self.net_2 = NetworkFactory( name='test-net-2.dc', address="192.168.56.0/24", ) self.net_2.save()
def _get_user_by_perm(self, perm): group = GroupFactory( name=perm.name, boundperm_set=(BoundPerm(perm=perm),) ) user = UserFactory( is_staff=False, is_superuser=False, groups=(group,), ) return user
def test_user_no_perms(self): """user without perms -> show 403""" no_access_user = UserFactory( is_staff=False, is_superuser=False, ) self.client = login_as_user(no_access_user) response = self.client.get( self.success_login_url, follow=True, **self.request_headers ) self.assertEqual(response.status_code, 403)
def test_signal_sent(self): with mock_signal_receiver(fields_synced_signal) as rec: old_name = self.dev.name author = UserFactory() old_service = self.dev.service service = ServiceCatalogFactory() self.dev.service = service self.dev.save(user=author) rec.assert_called_with( signal=mock.ANY, sender=self.dev, changes=[ChangeTuple('service', old_service, service)], change_author=author, )
def setUp(self): venture_role = VentureRoleFactory() device_environment = DeviceEnvironmentFactory() service = ServiceCatalogFactory() self.mass_deployment = MassDeploymentFactory() self.user = UserFactory() self.data = { "mac": "00:00:00:00:00:00", "ip": "192.168.1.1", "hostname": "testhost.dc2", "preboot": PrebootFactory(), "venture": venture_role.venture, "venture_role": venture_role, "service": service.name, "device_environment": device_environment.name, "asset_identity": None, }
def get_profile(n): """Due to strange logic in lck.django we can't use subfactories to create profiles.""" user = UserFactory() user.save() return user.profile
def setUp(self): user = UserFactory(is_superuser=True) self.client = login_as_user(user)
def login_as_user(self, user=None, password='******', *args, **kwargs): if not user: user = UserFactory(*args, **kwargs) user.set_password(password) user.save() self.client.login(username=user.username, password=password)
class TestNetworksResource(TestCase): def setUp(self): self.user = UserFactory( username='******', email='*****@*****.**', is_staff=True, is_superuser=True, ) self.user.set_password('QWEasd') self.user.save() self.net_1 = NetworkFactory( name='test-net-1.dc', address="192.168.1.0/24", ) self.net_1.save() self.net_2 = NetworkFactory( name='test-net-2.dc', address="192.168.56.0/24", ) self.net_2.save() def _get_networks_names(self, result): return [ network['name'] for network in result['objects'] ] def test_resource(self): path = "/api/v0.9/networks/" # sort by min_ip response = self.client.get( path=path, data={ 'format': 'json', 'username': self.user.username, 'api_key': self.user.api_key.key, 'order_by': 'min_ip', }, format='json', ) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(result['meta']['total_count'], 2) self.assertEqual( self._get_networks_names(result), ['test-net-1.dc', 'test-net-2.dc'], ) # sort by max_ip response = self.client.get( path=path, data={ 'format': 'json', 'username': self.user.username, 'api_key': self.user.api_key.key, 'order_by': '-max_ip', }, format='json', ) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(result['meta']['total_count'], 2) self.assertEqual( self._get_networks_names(result), ['test-net-2.dc', 'test-net-1.dc'], ) # filter by min_ip response = self.client.get( path=path, data={ 'format': 'json', 'username': self.user.username, 'api_key': self.user.api_key.key, 'min_ip__lte': '3232235777', }, format='json', ) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(result['meta']['total_count'], 1) self.assertEqual( self._get_networks_names(result), ['test-net-1.dc'], )
def _prepare_users(self): self.user = UserFactory() self.owner = UserFactory()
class TestNetworksResource(TestCase): def setUp(self): self.user = UserFactory( username='******', email='*****@*****.**', is_staff=True, is_superuser=True, ) self.user.set_password('QWEasd') self.user.save() self.net_1 = NetworkFactory( name='test-net-1.dc', address="192.168.1.0/24", ) self.net_1.save() self.net_2 = NetworkFactory( name='test-net-2.dc', address="192.168.56.0/24", ) self.net_2.save() def _get_networks_names(self, result): return [network['name'] for network in result['objects']] def test_resource(self): path = "/api/v0.9/networks/" # sort by min_ip response = self.client.get( path=path, data={ 'format': 'json', 'username': self.user.username, 'api_key': self.user.api_key.key, 'order_by': 'min_ip', }, format='json', ) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(result['meta']['total_count'], 2) self.assertEqual( self._get_networks_names(result), ['test-net-1.dc', 'test-net-2.dc'], ) # sort by max_ip response = self.client.get( path=path, data={ 'format': 'json', 'username': self.user.username, 'api_key': self.user.api_key.key, 'order_by': '-max_ip', }, format='json', ) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(result['meta']['total_count'], 2) self.assertEqual( self._get_networks_names(result), ['test-net-2.dc', 'test-net-1.dc'], ) # filter by min_ip response = self.client.get( path=path, data={ 'format': 'json', 'username': self.user.username, 'api_key': self.user.api_key.key, 'min_ip__lte': '3232235777', }, format='json', ) self.assertEqual(response.status_code, 200) result = json.loads(response.content) self.assertEqual(result['meta']['total_count'], 1) self.assertEqual( self._get_networks_names(result), ['test-net-1.dc'], )