diff --git a/cromosoma/webmodel.py b/cromosoma/webmodel.py index ca781f3..e3d18c7 100644 --- a/cromosoma/webmodel.py +++ b/cromosoma/webmodel.py @@ -466,7 +466,13 @@ class WebModel: if self.yes_reset_conditions: self.reset_conditions() - return result + if result.rowcount>0: + + return True + else: + + return False + # Method for create sql tables diff --git a/tests/webmodeltest.py b/tests/webmodeltest.py index 88d9367..d420d5f 100644 --- a/tests/webmodeltest.py +++ b/tests/webmodeltest.py @@ -65,6 +65,12 @@ class TestWebModelMethods(unittest.TestCase): self.assertTrue(model.element_exists(1)) + model.conditions=['WHERE id=%s', [2]] + + self.assertFalse(model.delete()) + + self.assertTrue(model.delete()) + self.assertTrue(model.drop()) if __name__ == '__main__':