def test_apiary_view(client, set_up_apiary):
apiary = Apiary.objects.first()
apiary_count = Apiary.objects.count()
response = client.get(f'/apiary_view/{apiary.id}')
assert response.status_code == 200
assert apiary_count == 5
assert response.context['apiary'] == apiary
def test_scanresults(client, hostscan):
# Test rendering
scanresults_view = reverse("portscanner.hosts:results")
response = client.get(scanresults_view)
assert response.status_code == 200
assert hostscan.target.label in response.content.decode()
# Test search
response = client.post(scanresults_view, {"label": hostscan.target.label})
assert response.status_code == 302
assert response.url == f"/results/{hostscan.target.label}"
def test_scanresults_host(client, hostscan):
# Test rendering
scanresults_view = reverse("portscanner.hosts:results",
args=(hostscan.target.label, ))
assert scanresults_view == f"/results/{hostscan.target.label}"
response = client.get(scanresults_view, {"label": "testhost.example.com"})
assert response.status_code == 200
assert hostscan.target.label in response.content.decode()
# Test search
response = client.post(scanresults_view, {"label": "testhost.example.com"})
assert response.status_code == 302
assert response.url == "/results/testhost.example.com"
def test_hostscan(client, monkeypatch):
# Test rendering
monkeypatch.setattr(PortScanner, "scan", mockNmap().scan)
monkeypatch.setattr(PortScanner, "all_hosts", mockNmap().all_hosts)
hostscan_view = reverse("portscanner.hosts:hostscan")
response = client.get(hostscan_view)
assert response.status_code == 200
# Test scan
response = client.post(hostscan_view, {"label": "192.168.1.1"})
assert response.status_code == 302
assert response.url == "/results/192.168.1.1"
host = models.Host.objects.get(label="192.168.1.1")
hostscan = models.HostScan.objects.get(target__label="192.168.1.1")
assert set(["22/tcp", "80/tcp"]) == set(hostscan.ports["open"])
def test_beehive_view(client, set_up_beehive):
beehive = BeeHive.objects.first()
response = client.get(f'/beehive_view/{beehive.id}')
assert response.status_code == 200
def test_beehive_list_view(client, set_up_beehive):
response = client.get('/beehive_view/')
beehive = BeeHive.objects.all()
assert response.status_code == 200
assert len(response.context['beehive']) == 5
def test_beehive_create(client):
response = client.get('/beehive_create/')
assert response.status_code == 200
def test_apiary_create(client):
response = client.get('/apiary_create/')
assert response.status_code == 200
def test_apiary_list_view(client, set_up_apiary):
response = client.get('/apiary_view/')
apiary = Apiary.objects.all()
assert response.status_code == 200
assert len(response.context['apiary']) == 5
def test_add_user_view(client):
response = client.get('/add_user/')
assert response.status_code == 200
def test_login_view(client):
response = client.get('/login_view/')
assert response.status_code == 200
def test_dashboard_view(client):
response = client.get('/dashboard/')
assert response.status_code == 200
def test_beemother_list_view(client, set_up_beefamily):
response = client.get('/beemother_view/')
beemother = BeeMother.objects.all()
assert response.status_code == 200
assert len(response.context['beemother']) == 5
def test_index_view(client):
response = client.get('/')
assert response.status_code == 200
