def test_idx_device_exists(self): """Testing function idx_exists.""" # Test known working value result = db_device.idx_device_exists(self.expected['idx_device']) self.assertEqual(result, True) # Test known false value result = db_device.idx_device_exists(-1) self.assertEqual(result, False)
def _insert_agent_device(self): """Insert first agent and device in the database. Args: None Returns: None """ # Initialize key variables idx_agent = 1 idx_device = 1 # Add agent if db_agent.idx_agent_exists(idx_agent) is False: # Generate a UID and add a record in the database record = Agent(id_agent=general.encode(self.reserved), name=general.encode(self.reserved)) database = db.Database() database.add(record, 1109) # Add device if db_device.idx_device_exists(idx_device) is False: record = Device(description=general.encode(self.reserved), devicename=general.encode(self.reserved)) database = db.Database() database.add(record, 1106) # Add to Agent / Device table if db_deviceagent.device_agent_exists(idx_device, idx_agent) is False: record = DeviceAgent(idx_device=idx_device, idx_agent=idx_agent) database = db.Database() database.add(record, 1107)