def hospital(request): if request.method == 'GET': hospital=Hospital.objects.all() serializer=HospitalSerializer(hospital,many=True) return Response(serializer.data) elif request.method == 'POST': serializer=HospitalSerializer(data=request.data) if serializer.is_valid(): serializer.save() return Response(serializer.data,status=status.HTTP_201_CREATED) return Response(serializer.errors,status=status.HTTP_400_BAD_REQUEST)
def test_hospital_get_list_viewset(self):
# instantiate client
client = Client()
# login as admin
client.login(username=settings.MQTT['USERNAME'],
password=settings.MQTT['PASSWORD'])
# retrieve hospitals
response = client.get('/api/hospital/', follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = [
HospitalSerializer(self.h1).data,
HospitalSerializer(self.h2).data,
HospitalSerializer(self.h3).data
]
self.assertCountEqual(result, answer)
# logout
client.logout()
# login as testuser1
client.login(username='******', password='******')
# retrieve hospitals
response = client.get('/api/hospital/', follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = [
HospitalSerializer(Hospital.objects.get(id=self.h1.id)).data,
HospitalSerializer(Hospital.objects.get(id=self.h2.id)).data
]
self.assertCountEqual(result, answer)
# logout
client.logout()
# login as testuser2
client.login(username='******', password='******')
# retrieve hospitals
response = client.get('/api/hospital/', follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = []
self.assertCountEqual(result, answer)
# logout
client.logout()
def post(self, request, *args, **kwargs):
serializer = self.serializer_class(data=request.data,
context={'request': request})
serializer.is_valid(raise_exception=True)
user = serializer.validated_data['user']
token = Token.objects.get_or_create(user=user)
if user.is_hospital:
queryset = Hospital.objects.filter(User=user.id).first()
serializer = HospitalSerializer(queryset, many=False)
data = {}
data['token'] = str(token[0])
data.update(serializer.data)
context = {
"message":"Authentication",
"status":True,
"data":data
}
return Response(context,status=status.HTTP_202_ACCEPTED)
else:
context = {
"message":"You are not authenticated!!",
"status":False,
"data":None
}
return Response(context,status=status.HTTP_203_NON_AUTHORITATIVE_INFORMATION)
def test_hospital_serializer(self):
# test HospitalSerializer
for h in (self.h1, self.h2, self.h3):
serializer = HospitalSerializer(h)
result = {
'id': h.id,
'name': h.name,
'number': h.number,
'street': h.street,
'unit': h.unit,
'neighborhood': h.neighborhood,
'city': h.city,
'state': h.state,
'zipcode': h.zipcode,
'country': h.country,
'location': point2str(h.location),
'comment': h.comment,
'updated_by': h.updated_by.id,
'updated_on': date2iso(h.updated_on),
'equipmentholder_id': h.equipmentholder.id
# 'hospitalequipment_set':
# EquipmentItemSerializer(EquipmentItem.objects.filter(equipmentholder_id=h.equipmentholder.id),
# many=True).data
}
logger.debug('answer= {}'.format(serializer.data))
logger.debug('result = {}'.format(result))
self.assertDictEqual(serializer.data, result)
def hospital(request):
if request.method == 'GET':
hospital = Hospital.objects.all()
serializer = HospitalSerializer(hospital, many=True)
return Response(serializer.data)
elif request.method == 'POST':
serializer = HospitalSerializer(data=request.data)
if serializer.is_valid():
serializer.save()
return Response(serializer.data, status=status.HTTP_201_CREATED)
return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
def test_hospital_serializer(self):
# test HospitalSerializer
for h in (self.h1, self.h2, self.h3):
serializer = HospitalSerializer(h)
result = {
'id': h.id,
'name': h.name,
'number': h.number,
'street': h.street,
'unit': h.unit,
'neighborhood': h.neighborhood,
'city': h.city,
'state': h.state,
'zipcode': h.zipcode,
'country': h.country,
'location': point2str(h.location),
'comment': h.comment,
'updated_by': h.updated_by.id,
'updated_on': date2iso(h.updated_on)
}
self.assertDictEqual(serializer.data, result)
def on_hospital(self, clnt, userdata, msg):
try:
logger.debug("on_hospital: msg = '{}:{}'".format(
msg.topic, msg.payload))
# parse topic
user, client, data, hospital_id = self.parse_topic(msg, 4)
except Exception as e:
logger.debug("on_hospital: ParseException '{}'".format(e))
return
try:
# retrieve hospital
hospital = Hospital.objects.get(id=hospital_id)
except Hospital.DoesNotExist:
# send error message to user
self.send_error_message(
user, client, msg.topic, msg.payload,
"Hospital with id '{}' does not exist".format(hospital_id))
return
except Exception as e:
# send error message to user
self.send_error_message(user, client, msg.topic, msg.payload,
"Exception: '{}'".format(e))
return
try:
logger.debug('on_hospital: hospital = {}'.format(hospital))
# update hospital
serializer = HospitalSerializer(hospital, data=data, partial=True)
if serializer.is_valid():
logger.debug('on_hospital: valid serializer')
# save to database
serializer.save(updated_by=user)
else:
logger.debug('on_hospital: INVALID serializer')
# send error message to user
self.send_error_message(user, client, msg.topic, msg.payload,
serializer.errors)
except Exception as e:
logger.debug('on_hospital: serializer EXCEPTION')
# send error message to user
self.send_error_message(user, client, msg.topic, msg.payload, e)
logger.debug('on_hospital: DONE')
def publish_hospital(self, hospital, qos=2, retain=False):
self.publish_topic('hospital/{}/data'.format(hospital.id),
HospitalSerializer(hospital),
qos=qos,
retain=retain)
def test_hospital_get_viewset(self):
# instantiate client
client = Client()
# login as admin
client.login(username=settings.MQTT['USERNAME'],
password=settings.MQTT['PASSWORD'])
# retrieve any hospital
response = client.get('/api/hospital/{}/'.format(str(self.h1.id)),
follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h1.id)).data
self.assertDictEqual(result, answer)
# retrieve any hospital
response = client.get('/api/hospital/{}/'.format(str(self.h2.id)),
follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h2.id)).data
self.assertDictEqual(result, answer)
# retrieve any hospital
response = client.get('/api/hospital/{}/'.format(str(self.h3.id)),
follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h3.id)).data
self.assertDictEqual(result, answer)
# logout
client.logout()
# login as testuser1
client.login(username='******', password='******')
# retrieve own's
response = client.get('/api/hospital/{}/'.format(str(self.h1.id)),
follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h1.id)).data
self.assertDictEqual(result, answer)
response = client.get('/api/hospital/{}/'.format(str(self.h2.id)),
follow=True)
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h2.id)).data
self.assertDictEqual(result, answer)
# retrieve someone else's
response = client.get('/api/hospital/{}/'.format(str(self.h3.id)),
follow=True)
self.assertEqual(response.status_code, 404)
# logout
client.logout()
# login as testuser2
client.login(username='******', password='******')
# retrieve someone else's
response = client.get('/api/hospital/{}/'.format(str(self.h1.id)),
follow=True)
self.assertEqual(response.status_code, 404)
# retrieve someone else's
response = client.get('/api/hospital/{}/'.format(str(self.h2.id)),
follow=True)
self.assertEqual(response.status_code, 404)
# retrieve someone else's
response = client.get('/api/hospital/{}/'.format(str(self.h3.id)),
follow=True)
self.assertEqual(response.status_code, 404)
# logout
client.logout()
def test_hospital_patch_viewset(self):
# instantiate client
client = Client()
# login as admin
client.login(username=settings.MQTT['USERNAME'],
password=settings.MQTT['PASSWORD'])
# retrieve hospital
response = client.get('/api/hospital/{}/'.format(str(self.h1.id)))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(self.h1).data
self.assertDictEqual(result, answer)
# set street hospital
street = 'new street'
response = client.patch('/api/hospital/{}/'.format(str(self.h1.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h1.id)).data
self.assertDictEqual(result, answer)
# retrieve new hospital street
response = client.get('/api/hospital/{}/'.format(str(self.h1.id)))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
self.assertEqual(result['street'], street)
# set hospital location
location = {'latitude': -2., 'longitude': 7.}
response = client.patch('/api/hospital/{}/'.format(str(self.h1.id)),
content_type='application/json',
data=json.dumps({'location': str(location)}))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h1.id)).data
self.assertDictEqual(result, answer)
# retrieve new hospital location
response = client.get('/api/hospital/{}/'.format(str(self.h1.id)))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
self.assertEqual(result['street'], street)
self.assertEqual(result['location'], point2str(location))
# set wrong hospital id
response = client.patch('/api/hospital/100/',
data=json.dumps({'street': street}))
self.assertEqual(response.status_code, 404)
# logout
client.logout()
# login as testuser1
client.login(username='******', password='******')
# retrieve hospital
response = client.get('/api/hospital/{}/'.format(str(self.h2.id)))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(self.h2).data
self.assertDictEqual(result, answer)
# set street hospital
street = 'another street'
response = client.patch('/api/hospital/{}/'.format(str(self.h2.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h2.id)).data
self.assertDictEqual(result, answer)
# retrieve new hospital street
response = client.get('/api/hospital/{}/'.format(str(self.h2.id)))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
self.assertEqual(result['street'], street)
# set hospital location
location = {'latitude': -2., 'longitude': 7.}
response = client.patch('/api/hospital/{}/'.format(str(self.h2.id)),
content_type='application/json',
data=json.dumps(
{'location': point2str(location)}))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
answer = HospitalSerializer(Hospital.objects.get(id=self.h2.id)).data
self.assertDictEqual(result, answer)
# retrieve new hospital location
response = client.get('/api/hospital/{}/'.format(str(self.h2.id)))
self.assertEqual(response.status_code, 200)
result = JSONParser().parse(BytesIO(response.content))
self.assertEqual(result['street'], street)
self.assertEqual(result['location'], point2str(location))
# set status hospital (no permission)
street = 'yet another hospital'
response = client.patch('/api/hospital/{}/'.format(str(self.h1.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 404)
# set street hospital (no permission)
street = 'another one street'
response = client.patch('/api/hospital/{}/'.format(str(self.h3.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 404)
# logout
client.logout()
# login as testuser2
client.login(username='******', password='******')
# set street hospital
street = 'some street'
response = client.patch('/api/hospital/{}/'.format(str(self.h1.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 404)
# set street hospital
response = client.patch('/api/hospital/{}/'.format(str(self.h2.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 404)
# set street hospital
response = client.patch('/api/hospital/{}/'.format(str(self.h2.id)),
content_type='application/json',
data=json.dumps({
'street': street,
}))
self.assertEqual(response.status_code, 404)
# logout
client.logout()
def test_hospital_update_serializer(self):
# superuser first
# Update hospital street
h = self.h1
user = self.u1
street = 'new street'
serializer = HospitalSerializer(h,
data={
'street': street,
},
partial=True)
serializer.is_valid()
serializer.save(updated_by=user)
# test
serializer = HospitalSerializer(h)
result = {
'id': h.id,
'name': h.name,
'number': h.number,
'street': street,
'unit': h.unit,
'neighborhood': h.neighborhood,
'city': h.city,
'state': h.state,
'zipcode': h.zipcode,
'country': h.country,
'comment': h.comment,
'location': point2str(h.location),
'updated_by': user.id,
'updated_on': date2iso(h.updated_on)
}
self.assertDictEqual(serializer.data, result)
# Update hospital location
location = {'latitude': -2., 'longitude': 7.}
serializer = HospitalSerializer(h,
data={
'location': location,
},
partial=True)
serializer.is_valid()
serializer.save(updated_by=user)
# test
serializer = HospitalSerializer(h)
result = {
'id': h.id,
'name': h.name,
'number': h.number,
'street': h.street,
'unit': h.unit,
'neighborhood': h.neighborhood,
'city': h.city,
'state': h.state,
'zipcode': h.zipcode,
'country': h.country,
'comment': h.comment,
'location': point2str(location),
'updated_by': user.id,
'updated_on': date2iso(h.updated_on)
}
self.assertDictEqual(serializer.data, result)
# regular authorized user
# Update hospital street
h = self.h2
user = self.u2
street = 'new street'
serializer = HospitalSerializer(h,
data={
'street': street,
},
partial=True)
serializer.is_valid()
serializer.save(updated_by=user)
# test
serializer = HospitalSerializer(h)
result = {
'id': h.id,
'name': h.name,
'number': h.number,
'street': street,
'unit': h.unit,
'neighborhood': h.neighborhood,
'city': h.city,
'state': h.state,
'zipcode': h.zipcode,
'country': h.country,
'comment': h.comment,
'location': point2str(h.location),
'updated_by': user.id,
'updated_on': date2iso(h.updated_on)
}
self.assertDictEqual(serializer.data, result)
# Update hospital location
location = {'latitude': -2., 'longitude': 7.}
serializer = HospitalSerializer(h,
data={
'location': location,
},
partial=True)
serializer.is_valid()
serializer.save(updated_by=user)
# test
serializer = HospitalSerializer(h)
result = {
'id': h.id,
'name': h.name,
'number': h.number,
'street': h.street,
'unit': h.unit,
'neighborhood': h.neighborhood,
'city': h.city,
'state': h.state,
'zipcode': h.zipcode,
'country': h.country,
'comment': h.comment,
'location': point2str(location),
'updated_by': user.id,
'updated_on': date2iso(h.updated_on)
}
self.assertDictEqual(serializer.data, result)
def test_mqttseed(self):
# seed
from django.core import management
management.call_command('mqttseed', verbosity=1)
print('>> Processing messages...')
# Start client as admin
broker = {
'HOST': 'localhost',
'PORT': 1883,
'KEEPALIVE': 60,
'CLEAN_SESSION': True
}
broker.update(settings.MQTT)
broker['CLIENT_ID'] = 'test_mqttseed_admin'
client = MQTTTestClient(broker)
self.is_connected(client)
qos = 0
# Expect settings
client.expect('settings',
JSONRenderer().render(SettingsView.get_settings()), qos)
# Expect all ambulances
for ambulance in Ambulance.objects.all():
client.expect(
'ambulance/{}/data'.format(ambulance.id),
JSONRenderer().render(AmbulanceSerializer(ambulance).data),
qos)
# Expect all hospitals
for hospital in Hospital.objects.all():
client.expect(
'hospital/{}/data'.format(hospital.id),
JSONRenderer().render(HospitalSerializer(hospital).data), qos)
hospital_equipment = hospital.hospitalequipment_set.values(
'equipment')
equipment = Equipment.objects.filter(id__in=hospital_equipment)
client.expect(
'hospital/{}/metadata'.format(hospital.id),
JSONRenderer().render(
EquipmentSerializer(equipment, many=True).data), qos)
# Expect all hospital equipments
for e in HospitalEquipment.objects.all():
client.expect(
'hospital/{}/equipment/{}/data'.format(e.hospital.id,
e.equipment.name),
JSONRenderer().render(HospitalEquipmentSerializer(e).data),
qos)
# Expect all profiles
for obj in Profile.objects.all():
client.expect(
'user/{}/profile'.format(obj.user.username),
JSONRenderer().render(ExtendedProfileSerializer(obj).data),
qos)
# Subscribed?
self.is_subscribed(client)
# Done?
self.loop(client)
client.wait()
# Repeat with same client
client = MQTTTestClient(broker)
self.is_connected(client)
qos = 0
# Expect settings
client.expect('settings',
JSONRenderer().render(SettingsView.get_settings()), qos)
# Expect all ambulances
for ambulance in Ambulance.objects.all():
client.expect(
'ambulance/{}/data'.format(ambulance.id),
JSONRenderer().render(AmbulanceSerializer(ambulance).data),
qos)
# Expect all hospitals
for hospital in Hospital.objects.all():
client.expect(
'hospital/{}/data'.format(hospital.id),
JSONRenderer().render(HospitalSerializer(hospital).data), qos)
hospital_equipment = hospital.hospitalequipment_set.values(
'equipment')
equipment = Equipment.objects.filter(id__in=hospital_equipment)
client.expect(
'hospital/{}/metadata'.format(hospital.id),
JSONRenderer().render(
EquipmentSerializer(equipment, many=True).data), qos)
# Expect all hospital equipments
for e in HospitalEquipment.objects.all():
client.expect(
'hospital/{}/equipment/{}/data'.format(e.hospital.id,
e.equipment.name),
JSONRenderer().render(HospitalEquipmentSerializer(e).data),
qos)
# Expect all profiles
for obj in Profile.objects.all():
client.expect(
'user/{}/profile'.format(obj.user.username),
JSONRenderer().render(ExtendedProfileSerializer(obj).data),
qos)
# Subscribed?
self.is_subscribed(client)
# Done?
self.loop(client)
client.wait()
# Repeat with same client and different qos
client = MQTTTestClient(broker)
self.is_connected(client)
qos = 2
# Expect settings
client.expect('settings',
JSONRenderer().render(SettingsView.get_settings()), qos)
# Expect all ambulances
for ambulance in Ambulance.objects.all():
client.expect(
'ambulance/{}/data'.format(ambulance.id),
JSONRenderer().render(AmbulanceSerializer(ambulance).data),
qos)
# Expect all hospitals
for hospital in Hospital.objects.all():
client.expect(
'hospital/{}/data'.format(hospital.id),
JSONRenderer().render(HospitalSerializer(hospital).data), qos)
hospital_equipment = hospital.hospitalequipment_set.values(
'equipment')
equipment = Equipment.objects.filter(id__in=hospital_equipment)
client.expect(
'hospital/{}/metadata'.format(hospital.id),
JSONRenderer().render(
EquipmentSerializer(equipment, many=True).data), qos)
# Expect all hospital equipments
for e in HospitalEquipment.objects.all():
client.expect(
'hospital/{}/equipment/{}/data'.format(e.hospital.id,
e.equipment.name),
JSONRenderer().render(HospitalEquipmentSerializer(e).data),
qos)
# Expect all profiles
for obj in Profile.objects.all():
client.expect(
'user/{}/profile'.format(obj.user.username),
JSONRenderer().render(ExtendedProfileSerializer(obj).data),
qos)
# Subscribed?
self.is_subscribed(client)
# Done?
self.loop(client)
client.wait()
# repeat with another user
qos = 0
# Start client as common user
broker['USERNAME'] = '******'
broker['PASSWORD'] = '******'
broker['CLIENT_ID'] = 'test_mqttseed_testuser1'
client = MQTTTestClient(broker)
self.is_connected(client)
# Expect settings
client.expect('settings',
JSONRenderer().render(SettingsView.get_settings()), qos)
# Expect user profile
profile = Profile.objects.get(user__username='******')
client.expect(
'user/testuser1/profile',
JSONRenderer().render(ExtendedProfileSerializer(profile).data),
qos)
# User Ambulances
can_read = profile.ambulances.filter(
can_read=True).values('ambulance_id')
for ambulance in Ambulance.objects.filter(id__in=can_read):
client.expect(
'ambulance/{}/data'.format(ambulance.id),
JSONRenderer().render(AmbulanceSerializer(ambulance).data),
qos)
# User Hospitals
can_read = profile.hospitals.filter(
can_read=True).values('hospital_id')
for hospital in Hospital.objects.filter(id__in=can_read):
client.expect(
'hospital/{}/data'.format(hospital.id),
JSONRenderer().render(HospitalSerializer(hospital).data), qos)
# Expect all user hospital equipments
for e in HospitalEquipment.objects.filter(hospital__id__in=can_read):
client.expect(
'hospital/{}/equipment/{}/data'.format(e.hospital.id,
e.equipment.name),
JSONRenderer().render(HospitalEquipmentSerializer(e).data),
qos)
# Subscribed?
self.is_subscribed(client)
# Done?
self.loop(client)
client.wait()
# repeat with another user
qos = 0
# Start client as common user
broker['USERNAME'] = '******'
broker['PASSWORD'] = '******'
broker['CLIENT_ID'] = 'test_mqttseed_testuser2'
client = MQTTTestClient(broker)
self.is_connected(client)
# Expect settings
client.expect('settings',
JSONRenderer().render(SettingsView.get_settings()), qos)
# Expect user profile
profile = Profile.objects.get(user__username='******')
client.expect(
'user/testuser2/profile',
JSONRenderer().render(ExtendedProfileSerializer(profile).data),
qos)
# User Ambulances
can_read = profile.ambulances.filter(
can_read=True).values('ambulance_id')
for ambulance in Ambulance.objects.filter(id__in=can_read):
client.expect(
'ambulance/{}/data'.format(ambulance.id),
JSONRenderer().render(AmbulanceSerializer(ambulance).data),
qos)
# User Hospitals
can_read = profile.hospitals.filter(
can_read=True).values('hospital_id')
for hospital in Hospital.objects.filter(id__in=can_read):
client.expect(
'hospital/{}/data'.format(hospital.id),
JSONRenderer().render(HospitalSerializer(hospital).data), qos)
# Expect all user hospital equipments
for e in HospitalEquipment.objects.filter(hospital__id__in=can_read):
client.expect(
'hospital/{}/equipment/{}/data'.format(e.hospital.id,
e.equipment.name),
JSONRenderer().render(HospitalEquipmentSerializer(e).data),
qos)
# Subscribed?
self.is_subscribed(client)
# Done?
self.loop(client)
client.wait()