From 16cf3cec763de85b3dbf1a41db60ee3b204bb9d0 Mon Sep 17 00:00:00 2001
From: Raphael Michel <mail@raphaelmichel.de>
Date: Mon, 12 Oct 2020 11:30:57 +0200
Subject: [PATCH] Add tests for device API

---
 src/tests/api/test_devices.py | 111 ++++++++++++++++++++++++++++++++++
 1 file changed, 111 insertions(+)
 create mode 100644 src/tests/api/test_devices.py

diff --git a/src/tests/api/test_devices.py b/src/tests/api/test_devices.py
new file mode 100644
index 0000000000..b8a931a1ff
--- /dev/null
+++ b/src/tests/api/test_devices.py
@@ -0,0 +1,111 @@
+import pytest
+from django.utils.timezone import now
+from django_scopes import scopes_disabled
+
+from pretix.base.models import Device
+
+
+@pytest.fixture
+def device(organizer, event):
+    t = organizer.devices.create(
+        device_id=1,
+        name='Scanner',
+        hardware_brand="Zebra",
+        unique_serial="UOS3GNZ27O39V3QS",
+        initialization_token="frkso3m2w58zuw70",
+        hardware_model="TC25",
+        software_brand="pretixSCAN",
+        software_version="1.5.1",
+        initialized=now(),
+        all_events=False,
+    )
+    t.limit_events.add(event)
+    return t
+
+
+TEST_DEV_RES = {
+    "device_id": 1,
+    "unique_serial": "UOS3GNZ27O39V3QS",
+    "initialization_token": "frkso3m2w58zuw70",
+    "all_events": False,
+    "limit_events": [
+        "dummy"
+    ],
+    "revoked": False,
+    "name": "Scanner",
+    "created": "2020-09-18T14:17:40.971519Z",
+    "initialized": "2020-09-18T14:17:44.190021Z",
+    "hardware_brand": "Zebra",
+    "hardware_model": "TC25",
+    "software_brand": "pretixSCAN",
+    "software_version": "1.5.1"
+}
+
+
+@pytest.mark.django_db
+def test_device_list(token_client, organizer, event, device):
+    res = dict(TEST_DEV_RES)
+    res["created"] = device.created.isoformat().replace('+00:00', 'Z')
+    res["initialized"] = device.initialized.isoformat().replace('+00:00', 'Z')
+
+    resp = token_client.get('/api/v1/organizers/{}/devices/'.format(organizer.slug))
+    assert resp.status_code == 200
+    assert [res] == resp.data['results']
+
+
+@pytest.mark.django_db
+def test_device_detail(token_client, organizer, event, device):
+    res = dict(TEST_DEV_RES)
+    res["created"] = device.created.isoformat().replace('+00:00', 'Z')
+    res["initialized"] = device.initialized.isoformat().replace('+00:00', 'Z')
+    resp = token_client.get('/api/v1/organizers/{}/devices/{}/'.format(organizer.slug, device.device_id))
+    assert resp.status_code == 200
+    assert res == resp.data
+
+
+TEST_DEVICE_CREATE_PAYLOAD = {
+    "name": "Foobar",
+    "all_events": False,
+    "limit_events": ["dummy"],
+}
+
+
+@pytest.mark.django_db
+def test_device_create(token_client, organizer, event):
+    resp = token_client.post(
+        '/api/v1/organizers/{}/devices/'.format(organizer.slug),
+        TEST_DEVICE_CREATE_PAYLOAD,
+        format='json'
+    )
+    assert resp.status_code == 201
+    with scopes_disabled():
+        d = Device.objects.get(device_id=resp.data['device_id'])
+        assert list(d.limit_events.all()) == [event]
+        assert d.initialization_token
+        assert not d.initialized
+
+
+@pytest.mark.django_db
+def test_device_update(token_client, organizer, event, device):
+    resp = token_client.patch(
+        '/api/v1/organizers/{}/devices/{}/'.format(organizer.slug, device.device_id),
+        {
+            'name': 'bla',
+            'hardware_brand': 'Foo'
+        },
+        format='json'
+    )
+    assert resp.status_code == 200
+    device.refresh_from_db()
+    assert device.hardware_brand == 'Zebra'
+    assert device.name == 'bla'
+
+
+@pytest.mark.django_db
+def test_device_delete(token_client, organizer, event, device):
+    resp = token_client.delete(
+        '/api/v1/organizers/{}/devices/{}/'.format(organizer.slug, device.device_id),
+    )
+    assert resp.status_code == 405
+    with scopes_disabled():
+        assert organizer.devices.count() == 1