you can now request a device, thinking we need a mapping table again :p
This commit is contained in:
parent
69af967ae6
commit
48c90c95f5
|
@ -8,7 +8,6 @@ class Task(object):
|
||||||
for field in ('id', 'name', 'owner', 'status'):
|
for field in ('id', 'name', 'owner', 'status'):
|
||||||
setattr(self, field, kwargs.get(field, None))
|
setattr(self, field, kwargs.get(field, None))
|
||||||
|
|
||||||
|
|
||||||
class DeviceSerializer(serializers.ModelSerializer):
|
class DeviceSerializer(serializers.ModelSerializer):
|
||||||
class Meta:
|
class Meta:
|
||||||
model = Device
|
model = Device
|
||||||
|
|
|
@ -7,8 +7,8 @@ from test_plus.test import TestCase
|
||||||
from rest_framework.test import APIRequestFactory
|
from rest_framework.test import APIRequestFactory
|
||||||
from rest_framework.test import RequestsClient
|
from rest_framework.test import RequestsClient
|
||||||
|
|
||||||
from mhackspace.rfid.models import Device
|
from mhackspace.rfid.models import Device, Rfid
|
||||||
from mhackspace.user.models import User
|
from mhackspace.users.models import User
|
||||||
|
|
||||||
|
|
||||||
# http://www.django-rest-framework.org/api-guide/testing/
|
# http://www.django-rest-framework.org/api-guide/testing/
|
||||||
|
@ -22,11 +22,15 @@ class MigrationTestCase(TestCase):
|
||||||
call_command('migrate', 'rfid', stdout=out)
|
call_command('migrate', 'rfid', stdout=out)
|
||||||
self.assertIn("... OK\n", out.getvalue())
|
self.assertIn("... OK\n", out.getvalue())
|
||||||
|
|
||||||
|
|
||||||
class ApiTests(TestCase):
|
class ApiTests(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
Device(name='device01').save()
|
self.device = Device(name='device01')
|
||||||
User(name='User01').save()
|
self.device.save()
|
||||||
Rfid(code=1, user=1).save()
|
self.user = User(name='User01')
|
||||||
|
self.user.save()
|
||||||
|
self.rfid = Rfid(code=1, user=self.user)
|
||||||
|
self.rfid.save()
|
||||||
|
|
||||||
def testAuth(self):
|
def testAuth(self):
|
||||||
factory = APIRequestFactory()
|
factory = APIRequestFactory()
|
||||||
|
@ -35,12 +39,13 @@ class ApiTests(TestCase):
|
||||||
def testSamsMadness(self):
|
def testSamsMadness(self):
|
||||||
client = RequestsClient()
|
client = RequestsClient()
|
||||||
response = client.post(
|
response = client.post(
|
||||||
'http://127.0.0.1:8180/api/v1/rfidAuth/?format=json',
|
'http://127.0.0.1:8180/api/v1/rfidAuth/',
|
||||||
data={'rfid':'1', 'device': '1'})
|
data={'rfid':'1', 'device': '1'})
|
||||||
print(response)
|
# print(response.json())
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
self.assertEquals(response.json().get('results'), [{'name': 'device01'}])
|
self.assertEquals(
|
||||||
|
response.json(),
|
||||||
|
[{'rfid': self.rfid.code, 'name': 'device01', 'device_id': self.device.identification}])
|
||||||
|
|
||||||
def testAuthUserWithDevice(self):
|
def testAuthUserWithDevice(self):
|
||||||
client = RequestsClient()
|
client = RequestsClient()
|
||||||
|
@ -48,11 +53,9 @@ class ApiTests(TestCase):
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
self.assertEquals(response.json().get('results'), [{'name': 'device01'}])
|
self.assertEquals(response.json().get('results'), [{'name': 'device01'}])
|
||||||
|
|
||||||
|
|
||||||
def testFetchDeviceList(self):
|
def testFetchDeviceList(self):
|
||||||
client = RequestsClient()
|
client = RequestsClient()
|
||||||
response = client.get('http://127.0.0.1:8180/api/v1/rfid/?format=json')
|
response = client.get('http://127.0.0.1:8180/api/v1/rfid/?format=json')
|
||||||
assert response.status_code == 200
|
assert response.status_code == 200
|
||||||
self.assertEquals(response.json().get('results'), [{'name': 'device01'}])
|
self.assertEquals(response.json().get('results'), [{'name': 'device01'}])
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
from rest_framework.response import Response
|
from rest_framework.response import Response
|
||||||
from rest_framework.views import APIView
|
from rest_framework.views import APIView
|
||||||
from rest_framework import viewsets
|
from rest_framework import viewsets
|
||||||
|
from rest_framework import status
|
||||||
from mhackspace.rfid.models import Device, Rfid
|
from mhackspace.rfid.models import Device, Rfid
|
||||||
from mhackspace.rfid.serializers import DeviceSerializer, AuthSerializer
|
from mhackspace.rfid.serializers import DeviceSerializer, AuthSerializer
|
||||||
from django.shortcuts import get_list_or_404, get_object_or_404
|
from django.shortcuts import get_list_or_404, get_object_or_404
|
||||||
|
@ -13,20 +14,20 @@ class DeviceViewSet(viewsets.ModelViewSet):
|
||||||
|
|
||||||
# https://medium.com/django-rest-framework/django-rest-framework-viewset-when-you-don-t-have-a-model-335a0490ba6f
|
# https://medium.com/django-rest-framework/django-rest-framework-viewset-when-you-don-t-have-a-model-335a0490ba6f
|
||||||
class AuthUserWithDeviceViewSet(viewsets.ViewSet):
|
class AuthUserWithDeviceViewSet(viewsets.ViewSet):
|
||||||
# http_method_names = ['get', 'post', 'head']
|
http_method_names = ['post']
|
||||||
serializer_class = AuthSerializer
|
serializer_class = AuthSerializer
|
||||||
|
|
||||||
def list(self, request):
|
def list(self, request):
|
||||||
serializer = AuthSerializer(instance={'name': '1','rfid': '1', 'device_id': '1'})
|
serializer = AuthSerializer(
|
||||||
|
instance={'name': '1', 'rfid': '1', 'device_id': '1'})
|
||||||
return Response(serializer.data)
|
return Response(serializer.data)
|
||||||
|
|
||||||
def post(self, request, format=None):
|
def post(self, request, format=None):
|
||||||
rfid = Rfid.objects.get(code=request.GET.get('rfid_id'))
|
try:
|
||||||
|
rfid = Rfid.objects.get(code=request.data.get('rfid'))
|
||||||
print(rfid.user.device__set(device=request.GET.get('rfid_id')))
|
device = Device.objects.get(user=rfid.user, identifier=request.data.get('device_id'))
|
||||||
# = get_object_or_404(Disease, pk=disease_id)
|
except:
|
||||||
|
return Response(status=status.HTTP_404_NOT_FOUND)
|
||||||
|
serializer = AuthSerializer(
|
||||||
# Device(rfid, device)
|
instance={'name': device.name, 'rfid': rfid.code, 'device_id': device.identifier})
|
||||||
serializer = AuthSerializer(instance={'name': '1', 'rfid': '1', 'device_id': '1'})
|
return Response(serializer.data, status=200)
|
||||||
return Response(serializer.data)
|
|
||||||
|
|
Loading…
Reference in New Issue