def setUp(self):
self.uuid = generate_uuid() # Generate an id except for POST
self.uuid1 = generate_uuid() # Generate id for update
self.data = [{
'id': self.uuid,
'scope_id': 'a05fa20f-662c-4e9d-83b3-8186bea5d680',
'match_fields': 'TOS,PROTOCOL,MAC-ADDRESS',
'collect_fields': 'COLLECT COUNTER,FLOW ACCESS TIMESTAMP',
}]
self.data1 = [{
'id': self.uuid1,
'scope_id': '4b63d4b9-fbd2-4b47-86a1-912ecae3b79e',
'match_fields': 'TOS,SOURCE-ADDRESS,DESTINATION-ADDRESS',
'collect_fields': 'FLOW ACCESS TIMESTAMP,VLAN',
}]
self.data2 = [{
'id': self.uuid,
'scope_id': '8769c99f-afa1-4da0-86d4-0fdd66d028a9',
'match_fields': 'PROTOCOL,VLAN',
'collect_fields': 'COLLECT COUNTER,MAC-ADDRESS',
}]
self.data3 = [{}]
serializer_data = json.loads(json.dumps(self.data))
# for serializerdata in serializer_data:
serializer = NetFlowMonitorSerializer(data=serializer_data[0])
if serializer.valid(self.uuid):
serializer.create(serializer.data)
def test_post_with_invalid_data(self):
"""test case to test create function with invalid data"""
# serializer_data3 = json.loads(json.dumps(self.data3))
# for serializerdata3 in serializer_data3:
serializer = NetFlowMonitorSerializer(data=None)
if serializer.valid(self.uuid):
with self.assertRaises(TypeError):
serializer.create(serializer.data)
def test_post_data(self):
"""test case to test create function with valid data"""
serializer_data1 = json.loads(json.dumps(self.data1))
# for serializerdata1 in serializer_data1:
serializer = NetFlowMonitorSerializer(data=serializer_data1[0])
if serializer.valid(self.uuid):
serializer.create(serializer.data)
else:
print("serializer data is not valid")
self.assertEqual(serializer_data1[0], serializer.data)
def test_update_id(self):
"""test case to test update function with id update"""
obj = NetFlowMonitor.get(self.uuid)
serializer_data1 = json.loads(json.dumps(self.data1))
# for serializerdata1 in serializer_data1:
with self.assertRaises(IDUpdateNotPermitted):
serializer = NetFlowMonitorSerializer(obj,
data=serializer_data1[0],
partial=True,
context={'pk': self.uuid1})
serializer.update(obj, serializer_data1[0])
def test_put_Data(self):
"""test case to test update function with valid data"""
obj = NetFlowMonitor.get(self.uuid)
serializer_data2 = json.loads(json.dumps(self.data2))
# for serializerdata2 in serializer_data2:
serializer = NetFlowMonitorSerializer(obj,
data=serializer_data2[0],
partial=True,
context={'pk': self.uuid})
serializer.update(obj, serializer_data2[0])
serializerdata2 = serializer_data2[0]
serializerdata2['id'] = self.uuid
self.assertEqual(serializerdata2, serializer.data)
def test_invalid_serializer_without_pk(self):
"""test case to test valid function with invalid serializer without pk
"""
record = NetFlowMonitor.get(self.uuid)
serializer = NetFlowMonitorSerializer(record)
self.assertEqual(serializer.valid(None), False)
def test_valid_serializer(self):
"""test case to test valid function with valid serializer"""
serializer_data1 = json.loads(json.dumps(self.data1))
# for serializerdata1 in serializer_data1:
serializer = NetFlowMonitorSerializer(data=serializer_data1[0])
self.assertEqual(serializer.valid(self.uuid), True)