def setup(self): super(TestUpdateFramework, self).setup() framework = Framework() framework.name = 'Example G-Cloud framework' framework.framework = 'g-cloud' framework.slug = 'example' framework.status = 'open' with self.app.app_context(): db.session.add(framework) db.session.commit()
def _add_framework(request, app, slug, **kwargs): # first whitelist the kwargs to limit what we have to worry about kwargs = {k: v for k, v in iteritems(kwargs) if k in _framework_kwargs_whitelist} if "status" in kwargs and "clarification_questions_open" not in kwargs: kwargs["clarification_questions_open"] = (kwargs["status"] == "open") framework = Framework(slug=slug, name=slug, **kwargs) db.session.add(framework) db.session.commit() def teardown(): with app.app_context(): FrameworkLot.query.filter(FrameworkLot.framework_id == framework.id).delete() Framework.query.filter(Framework.id == framework.id).delete() db.session.commit() request.addfinalizer(teardown) return framework.serialize()
def _add_framework(request, app, slug, **kwargs): # first whitelist the kwargs to limit what we have to worry about kwargs = {k: v for k, v in kwargs.items() if k in _framework_kwargs_whitelist} if "status" in kwargs and "clarification_questions_open" not in kwargs: kwargs["clarification_questions_open"] = (kwargs["status"] == "open") framework = Framework(slug=slug, name=slug, **kwargs) db.session.add(framework) db.session.commit() def teardown(): with app.app_context(): FrameworkLot.query.filter(FrameworkLot.framework_id == framework.id).delete() Framework.query.filter(Framework.id == framework.id).delete() db.session.commit() request.addfinalizer(teardown) return framework.serialize()
def setup(self): super(TestGetService, self).setup() now = pendulum.now('UTC') with self.app.app_context(): db.session.add(Framework( id=123, name="expired", slug="expired", framework="g-cloud", status="expired", )) db.session.commit() db.session.add(FrameworkLot( framework_id=123, lot_id=1 )) db.session.add( Supplier(code=1, name=u"Supplier 1", addresses=[Address(address_line="{} Dummy Street 1", suburb="Dummy", state="ZZZ", postal_code="0000", country='Australia')]) ) db.session.add(Service(service_id="123-published-456", supplier_code=1, updated_at=now, created_at=now, status='published', data={'foo': 'bar'}, lot_id=1, framework_id=1)) db.session.add(Service(service_id="123-disabled-456", supplier_code=1, updated_at=now, created_at=now, status='disabled', data={'foo': 'bar'}, lot_id=1, framework_id=1)) db.session.add(Service(service_id="123-enabled-456", supplier_code=1, updated_at=now, created_at=now, status='enabled', data={'foo': 'bar'}, lot_id=1, framework_id=1)) db.session.add(Service(service_id="123-expired-456", supplier_code=1, updated_at=now, created_at=now, status='enabled', data={'foo': 'bar'}, lot_id=1, framework_id=123)) db.session.commit()
def test_framework_should_not_accept_invalid_status(): app = create_app('test') with app.app_context(), assert_raises(ValidationError): f = Framework( name='foo', slug='foo', framework='g-cloud', status='invalid', ) db.session.add(f) db.session.commit()
def test_framework_should_accept_valid_statuses(): app = create_app('test') with app.app_context(): for i, status in enumerate(Framework.STATUSES): f = Framework( name='foo', slug='foo-{}'.format(i), framework='g-cloud', status=status, ) db.session.add(f) db.session.commit()
def setup_dummy_framework( self, slug, framework_family, name='New Framework', id=None, status='open', clarifications=False, lots=None, has_direct_award=True, has_further_competition=False, ): if lots is None: if framework_family.startswith('g-cloud'): lots = [ Lot.query.filter(Lot.slug == 'cloud-hosting').first(), Lot.query.filter(Lot.slug == 'cloud-software').first(), Lot.query.filter(Lot.slug == 'cloud-support').first(), ] elif framework_family.startswith( 'digital-outcomes-and-specialists'): lots = [ Lot.query.filter(Lot.slug == 'digital-outcomes').first(), Lot.query.filter( Lot.slug == 'digital-specialists').first(), Lot.query.filter( Lot.slug == 'user-research-participants').first(), Lot.query.filter( Lot.slug == 'user-research-studios').first(), ] else: lots = [] framework = Framework( id=id, slug=slug, name=name, framework=framework_family, status=status, clarification_questions_open=clarifications, lots=lots, has_direct_award=has_direct_award, has_further_competition=has_further_competition, ) db.session.add(framework) db.session.commit() return framework.id
def setup(self): super(TestUpdateFramework, self).setup() framework = Framework() framework.name = 'Example G-Cloud framework' framework.framework = 'g-cloud' framework.slug = 'example' framework.status = 'open' with self.app.app_context(): db.session.add(framework) db.session.commit()
def setup_dos_2_framework(self, status='open', clarifications=True): with self.app.app_context(): db.session.add( Framework( id=101, slug=u'digital-outcomes-and-specialists-2', name=u'Digital Outcomes and Specialists 2', framework=u'digital-outcomes-and-specialists', status=status, clarification_questions_open=clarifications, lots=[Lot.query.filter(Lot.slug == 'digital-outcomes').first(), Lot.query.filter(Lot.slug == 'digital-specialists').first(), Lot.query.filter(Lot.slug == 'user-research-participants').first(), Lot.query.filter(Lot.slug == 'user-research-studios').first(), ] ) ) db.session.commit()
def setup_dos_2_framework(self, status='open', clarifications=True): db.session.add( Framework( id=101, slug=u'digital-outcomes-and-specialists-2', name=u'Digital Outcomes and Specialists 2', framework=u'digital-outcomes-and-specialists', status=status, clarification_questions_open=clarifications, lots=[ Lot.query.filter(Lot.slug == 'digital-outcomes').first(), Lot.query.filter( Lot.slug == 'digital-specialists').first(), Lot.query.filter( Lot.slug == 'user-research-participants').first(), Lot.query.filter( Lot.slug == 'user-research-studios').first(), ], has_direct_award=False, has_further_competition=True, )) db.session.commit()
def framework(): return Framework(id=1, slug='digital-marketplace')