def test_model_save(self): self.assertEqual(0, len(context.model_query(TestModel).all())) t = _insert_test_object() self.assertEqual(1, len(context.model_query(TestModel).all())) db_t = context.model_query(TestModel).first() self.assertEqual(t.id, db_t.id) self.assertEqual(t.test_field, db_t.test_field)
def test_model_update(self): _insert_test_object() t = context.current().session.query(TestModel).first() context.model_update(t, test_field=42) db_t = context.model_query(TestModel).first() self.assertEqual(t.id, db_t.id) self.assertEqual(42, db_t.test_field)
def _model_filter_by(model, **kwargs): return ctx.model_query(model, project_only=True).filter_by(**kwargs)