def test_get_ip_ranges_empty(
self,
hosting_provider: ac_models.Hostingprovider,
sample_hoster_user: User,
):
"""
Exercise the simplest happy path.
"""
hosting_provider.save()
sample_hoster_user.hostingprovider = hosting_provider
sample_hoster_user.save()
rf = APIRequestFactory()
url_path = reverse("ip-range-list")
request = rf.get(url_path)
request.user = sample_hoster_user
# GET end point for IP Ranges
view = IPRangeViewSet.as_view({"get": "list"})
response = view(request)
# check contents
assert response.status_code == 200
assert len(response.data) == 0
def test_get_ip_ranges_for_hostingprovider_with_active_range(
self,
hosting_provider: ac_models.Hostingprovider,
sample_hoster_user: User,
green_ip: GreencheckIp,
):
hosting_provider.save()
sample_hoster_user.hostingprovider = hosting_provider
sample_hoster_user.save()
rf = APIRequestFactory()
url_path = reverse("ip-range-list")
request = rf.get(url_path)
request.user = sample_hoster_user
# GET end point for IP Ranges
view = IPRangeViewSet.as_view({"get": "list"})
response = view(request)
ip_range, *_ = response.data
assert response.status_code == 200
assert len(response.data) == 1
assert ip_range["ip_start"] == green_ip.ip_start
assert ip_range["ip_end"] == green_ip.ip_end
assert ip_range["hostingprovider"] == green_ip.hostingprovider.id
def test_create_new_ip_range(
self,
hosting_provider: ac_models.Hostingprovider,
sample_hoster_user: User,
green_ip: GreencheckIp,
):
hosting_provider.save()
sample_hoster_user.hostingprovider = hosting_provider
sample_hoster_user.save()
GreencheckIp.objects.count() == 1
rf = APIRequestFactory()
url_path = reverse("ip-range-list")
sample_json = {
"hostingprovider": hosting_provider.id,
"ip_start": "192.168.178.121",
"ip_end": "192.168.178.129",
}
request = rf.post(url_path, sample_json)
request.user = sample_hoster_user
view = IPRangeViewSet.as_view({"post": "create"})
response = view(request)
assert response.status_code == 201
GreencheckIp.objects.count() == 2
assert response.data["ip_start"] == "192.168.178.121"
assert response.data["ip_end"] == "192.168.178.129"
assert response.data["hostingprovider"] == hosting_provider.id
def test_get_ip_range_for_hostingprovider_by_id(
self,
hosting_provider: ac_models.Hostingprovider,
sample_hoster_user: User,
green_ip: GreencheckIp,
):
hosting_provider.save()
sample_hoster_user.hostingprovider = hosting_provider
sample_hoster_user.save()
rf = APIRequestFactory()
url_path = reverse("ip-range-detail", kwargs={"pk": green_ip.id})
request = rf.get(url_path)
request.user = sample_hoster_user
# GET end point for IP Ranges
view = IPRangeViewSet.as_view({"get": "retrieve"})
response = view(request, pk=green_ip.id)
assert response.status_code == 200
assert response.data["ip_start"] == green_ip.ip_start
assert response.data["ip_end"] == green_ip.ip_end
assert response.data["hostingprovider"] == green_ip.hostingprovider.id
def test_get_ip_ranges_without_auth(
self,
hosting_provider: ac_models.Hostingprovider,
sample_hoster_user: User,
):
"""
We don't want to list all the IP ranges we have, so we just show an empty
list for anon users.
"""
hosting_provider.save()
sample_hoster_user.hostingprovider = hosting_provider
sample_hoster_user.save()
rf = APIRequestFactory()
url_path = reverse("ip-range-list")
request = rf.get(url_path)
# set up the viewset, as a views, so it knows what to do when we
# pass in a GET request as defined a couple of lines up
view = IPRangeViewSet.as_view({"get": "list"})
response = view(request)
# check contents
assert response.status_code == 200
assert len(response.data) == 0
def test_get_ip_range_for_user_with_no_hosting_provider(
self,
sample_hoster_user: User,
rf: RequestFactory,
):
sample_hoster_user.save()
rf = APIRequestFactory()
url_path = reverse("ip-range-list")
request = rf.get(url_path)
request.user = sample_hoster_user
# GET end point for IP Ranges
view = IPRangeViewSet.as_view({"get": "list"})
response = view(request)
# check contents
assert response.status_code == 200
assert len(response.data) == 0
def test_delete_existing_ip_range(
self,
hosting_provider: ac_models.Hostingprovider,
sample_hoster_user: User,
green_ip: GreencheckIp,
):
"""
If a user deletes an IP Range, that has been referenced when marking
sites as green, an actual delete will mean that all those greenchecks
are now pointing to a non-existent range.
We do not delete with the API - we hide them.
"""
# arrange
hosting_provider.save()
sample_hoster_user.hostingprovider = hosting_provider
sample_hoster_user.save()
# check that we have what we expect first
assert GreencheckIp.objects.filter(active=True).count() == 1
assert GreencheckIp.objects.count() == 1
rf = APIRequestFactory()
url_path = reverse("ip-range-detail", kwargs={"pk": green_ip.id})
# act
request = rf.delete(url_path, pk=green_ip.id)
request.user = sample_hoster_user
view = IPRangeViewSet.as_view({"delete": "destroy"})
response = view(request, pk=green_ip.id)
# assert
assert response.status_code == 204
assert GreencheckIp.objects.filter(active=True).count() == 0
assert GreencheckIp.objects.filter(active=False).count() == 1
assert GreencheckIp.objects.count() == 1