def draft_service(request, app, supplier_framework): with app.app_context(): supplier_id = Supplier.query.filter( Supplier.supplier_id == supplier_framework['supplierId']).first() framework = Framework.query.filter( Framework.slug == supplier_framework['frameworkSlug']).first() lot = Lot.query.filter(Lot.slug == 'digital-specialists').first() fl_query = {'framework_id': framework.id, 'lot_id': lot.id} fl = FrameworkLot(**fl_query) ds = DraftService( framework=framework, lot=lot, service_id="1234567890", supplier=supplier_id, data={}, status='submitted', updated_at=datetime.now(), created_at=datetime.now(), lot_one_service_limit=lot.one_service_limit, ) db.session.add(fl) db.session.add(ds) db.session.commit() def teardown(): with app.app_context(): FrameworkLot.query.filter(*[ getattr(FrameworkLot, k) == v for k, v in fl_query.items() ]).delete(synchronize_session=False) Service.query.filter( Service.service_id == ds.service_id).delete( synchronize_session=False) DraftService.query.filter( DraftService.service_id == ds.service_id).delete( synchronize_session=False) db.session.commit() request.addfinalizer(teardown) with mock.patch('app.models.main.url_for', autospec=lambda i, **values: 'test.url/test'): Service.create_from_draft(ds, 'enabled') return ds.serialize()
def draft_service(request, app, supplier_framework): with app.app_context(): supplier_id = Supplier.query.filter(Supplier.supplier_id == supplier_framework['supplierId']).first() framework = Framework.query.filter(Framework.slug == supplier_framework['frameworkSlug']).first() lot = Lot.query.filter(Lot.slug == 'digital-specialists').first() fl_query = { 'framework_id': framework.id, 'lot_id': lot.id } fl = FrameworkLot(**fl_query) ds = DraftService( framework=framework, lot=lot, service_id="1234567890", supplier=supplier_id, data={}, status='submitted', updated_at=datetime.now(), created_at=datetime.now() ) db.session.add(fl) db.session.add(ds) db.session.commit() def teardown(): with app.app_context(): FrameworkLot.query.filter( *[getattr(FrameworkLot, k) == v for k, v in fl_query.items()] ).delete(synchronize_session=False) Service.query.filter(Service.service_id == ds.service_id).delete(synchronize_session=False) DraftService.query.filter(DraftService.service_id == ds.service_id).delete(synchronize_session=False) db.session.commit() request.addfinalizer(teardown) with mock.patch('app.models.main.url_for', autospec=lambda i, **values: 'test.url/test'): Service.create_from_draft(ds, 'enabled') return ds.serialize()