diff --git a/computers/tests/test_computer_form_views.py b/computers/tests/test_computer_form_views.py index 3bb0baa..683b8f5 100644 --- a/computers/tests/test_computer_form_views.py +++ b/computers/tests/test_computer_form_views.py @@ -46,3 +46,119 @@ def test_computer_update_view(create_admin_user): assert response.status_code == 302 computer.refresh_from_db() assert computer.name == data['name'] + + +def test_ram_relation_create_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + ram = mixer.blend('computers.Ram') + data = { + 'computer': computer.id, + 'ram': ram.id, + 'amount': 1 + } + url = '/computer/{}/create/ram-relation/'.format(computer.id) + response = client.post(url, data) + assert response.status_code == 302 + + +def test_cpu_relation_create_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + cpu = mixer.blend('computers.Cpu') + data = { + 'computer': computer.id, + 'cpu': cpu.id, + 'amount': 1 + } + url = '/computer/{}/create/cpu-relation/'.format(computer.id) + response = client.post(url, data) + assert response.status_code == 302 + + +def test_gpu_relation_create_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + gpu = mixer.blend('computers.gpu') + data = { + 'computer': computer.id, + 'gpu': gpu.id, + 'amount': 1 + } + url = '/computer/{}/create/gpu-relation/'.format(computer.id) + response = client.post(url, data) + assert response.status_code == 302 + + +def test_disk_relation_create_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + disk = mixer.blend('computers.Disk') + data = { + 'computer': computer.id, + 'disk': disk.id, + 'amount': 1 + } + url = '/computer/{}/create/disk-relation/'.format(computer.id) + response = client.post(url, data) + assert response.status_code == 302 + + +def test_ram_relation_delete_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + ram = mixer.blend('computers.Ram') + ram_relation = mixer.blend('computers.ComputerRamRelation', + computer=computer, ram=ram, amount=1) + url = '/delete/ram-relation/{}/'.format(ram_relation.id) + response = client.post(url) + assert response.status_code == 302 + + +def test_cpu_relation_delete_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + cpu = mixer.blend('computers.Cpu') + cpu_relation = mixer.blend('computers.ComputerCpuRelation', + computer=computer, cpu=cpu, amount=1) + url = '/delete/cpu-relation/{}/'.format(cpu_relation.id) + response = client.post(url) + assert response.status_code == 302 + + +def test_gpu_relation_delete_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + gpu = mixer.blend('computers.Gpu') + gpu_relation = mixer.blend('computers.ComputerGpuRelation', + computer=computer, gpu=gpu, amount=1) + url = '/delete/gpu-relation/{}/'.format(gpu_relation.id) + response = client.post(url) + assert response.status_code == 302 + + +def test_disk_relation_delete_view(create_admin_user): + create_admin_user() + client = Client() + client.login(username="pharma-admin", password="password") + computer = mixer.blend('computers.Computer', customer=mixer.SELECT) + disk = mixer.blend('computers.disk') + disk_relation = mixer.blend('computers.ComputerDiskRelation', + computer=computer, disk=disk, amount=1) + url = '/delete/disk-relation/{}/'.format(disk_relation.id) + response = client.post(url) + assert response.status_code == 302