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