def setUp(self): self.device_model = DeviceModel(type=DeviceType.rack_server, name="ziew-X") self.device_model.save() self.device = Device( model=self.device_model, sn='123456789', name='ziew', ) self.device.save()
def add_model(self, name, device_type): dm = DeviceModel() dm.model_type = device_type, dm.name = name dm.save() return dm