def test_get_rack_none(self):
serializer = EquipmentSerializer()
rack = {
'rack': None,
'rackunit': None
}
self.assertEqual(rack, serializer.get_rack(None))
def test_get_attrs_existent(self):
equipment = Equipment.objects.get(serial='G123456')
serializer = EquipmentSerializer()
attrs = {
'key': 'value'
}
self.assertEqual(attrs, serializer.get_attrs(equipment))
def test_get_ipmi_none(self):
serializer = EquipmentSerializer()
ipmi = {
'IPMI Hostname': None,
'IPMI Mac': None
}
self.assertEqual(ipmi, serializer.get_ipmi(None))
def test_get_ipmi_nonexistent(self):
equipment = Equipment.objects.get(serial='R123457')
serializer = EquipmentSerializer()
ipmi = {
'IPMI Hostname': None,
'IPMI Mac': None
}
self.assertEqual(ipmi, serializer.get_ipmi(equipment))
def test_get_ipmi_existent(self):
equipment = Equipment.objects.get(serial='G123456')
serializer = EquipmentSerializer()
ipmi = {
'IPMI Hostname': equipment.servermanagement.hostname,
'IPMI Mac': equipment.servermanagement.mac
}
self.assertEqual(ipmi, serializer.get_ipmi(equipment))
def test_get_rack_nonexistent(self):
equipment = Equipment.objects.get(serial='C1234567')
serializer = EquipmentSerializer()
rack = {
'rack': None,
'rackunit': None
}
self.assertEqual(rack, serializer.get_rack(equipment))
def test_get_rack_existent(self):
equipment = Equipment.objects.get(serial='G123456')
serializer = EquipmentSerializer()
rack = {
'rack': equipment.rack.name,
'rackunit': equipment.unit
}
self.assertEqual(rack, serializer.get_rack(equipment))
def EquipmentList(request):
jsonList = []
jsonList.append(request.data)
#SHOW ALL EQUIPMENTS 1 AND 0 STATUS
if request.method == 'GET':
queryset = Equipment.objects.all()
serializer_class = EquipmentSerializer(queryset, many=True)
return Response(serializer_class.data)
#ADD NEW EQUIPMENT
elif request.method == 'POST':
c = jsonList[0]['name']
search = Equipment.objects.filter(name__iexact = c)
if search:
return Response("Equipment alredy exist")
else:
queryset = EquipmentSerializer(data=request.data)
if queryset.is_valid():
queryset.save()
return Response(queryset.data, status=status.HTTP_201_CREATED)
return Response(queryset.errors, status=status.HTTP_400_BAD_REQUEST)
def test_get_dc_name_nonexistent(self):
equipment = Equipment.objects.get(serial='C1234567')
serializer = EquipmentSerializer()
name = None
self.assertEqual(name, serializer.get_dc_name(equipment))
def test_get_dc_name_none(self):
serializer = EquipmentSerializer()
name = None
self.assertEqual(name, serializer.get_dc_name(None))
def test_get_attrs_none(self):
serializer = EquipmentSerializer()
attrs = {}
self.assertEqual(attrs, serializer.get_attrs(None))
def test_get_dc_name_existent(self):
equipment = Equipment.objects.get(serial='G123456')
serializer = EquipmentSerializer()
name = equipment.dc.name
self.assertEqual(name, serializer.get_dc_name(equipment))
def test_get_attrs_nonexistent(self):
equipment = Equipment.objects.get(serial='C1234567')
serializer = EquipmentSerializer()
attrs = {}
self.assertEqual(attrs, serializer.get_attrs(equipment))
def test_get_allocation_existent(self):
equipment = Equipment.objects.get(serial='G123456')
allocation = equipment.allocation
serializer = EquipmentSerializer()
self.assertEqual(allocation, serializer.get_allocation(equipment))
def test_get_allocation_nonexistent(self):
equipment = Equipment.objects.get(serial='R123457')
serializer = EquipmentSerializer()
self.assertEqual(None, serializer.get_allocation(equipment))
def test_get_allocation_none(self):
serializer = EquipmentSerializer()
self.assertEqual(None, serializer.get_allocation(None))
def EquipmentUnavailable(request):
queryset = Equipment.objects.filter(status = 0).all()
serializer_class = EquipmentSerializer(queryset, many=True)
return Response(serializer_class.data)
