def test_multi_param_query(): company = Company() session.add(company) obj1 = crud.create(session, company, MyModel, name='obj1') obj2 = crud.create(session, company, MyModel, name='obj2') eq_({obj1, obj2}, set(crud.object_list(session, company, MyModel, name=['obj1', 'obj2'])))
def test_crud(): company = Company() session.add(company) assert [] == crud.object_list(session, company, MyModel) obj = crud.create(session, company, MyModel, name='xxx') assert obj.name == 'xxx' session.commit() read_obj = crud.read(session, company, MyModel, obj.id) assert read_obj == obj assert 1 == crud.object_count(session, company, MyModel) assert [read_obj] == crud.object_list(session, company, MyModel) crud.update(session, company, MyModel, obj.id, name='aaa') session.commit() update_obj = crud.read(session, company, MyModel, obj.id) assert update_obj.name == 'aaa' crud.delete(session, company, MyModel, obj.id) delete_obj = crud.read(session, company, MyModel, obj.id) assert delete_obj is None assert 0 == crud.object_count(session, company, MyModel)