def test_detele(self, mock_db):
""" Delete record."""
mock_srv_obj = mock.MagicMock()
ServerOp.delete(mock_srv_obj)
self.assertTrue(mock_db.session.delete.called)
self.assertTrue(mock_db.session.commit.called)
def case_delete(self, refresh_db_before):
""" Delete record."""
server_name = "TestServer"
ServerStatusOp.add("Status")
ServerTypeOp.add("TypeOne")
ServerOp.add(server_name, "Status", "TypeOne")
get_srv = ServerOp.get()
self.assertTrue(len(get_srv) is 1)
self.assertEqual(get_srv[0].name, server_name)
ServerOp.delete(get_srv[0])
get_empty = ServerOp.get()
self.assertFalse(get_empty)