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)