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)
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 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'], )