def setUp(self): from flask_migrate import upgrade upgrade() # insert the users/roles self.admin_role = get_a_role('admin') self.superadmin_role = get_a_role('superadmin') self.admin_user = insert_a_user(email='*****@*****.**', role=self.admin_role) self.superadmin_user = insert_a_user(email='*****@*****.**', role=self.superadmin_role) # insert the companies/contracts self.company_1 = insert_a_company(name='ship', insert_contract=False) company_2 = insert_a_company(name='boat', insert_contract=False) contract_type = ContractTypeFactory.create(name='test') self.contract_type2 = ContractTypeFactory.create(name='test2') self.contract1 = ContractBaseFactory.create( description='vessel', companies=[company_2], line_items=[LineItem(description='NAVY')], expiration_date=datetime.datetime.today() + datetime.timedelta(1), is_archived=False, financial_id='123', contract_type=contract_type) ContractBaseFactory.create( description='sail', financial_id='456', companies=[self.company_1], line_items=[LineItem(description='sunfish')], is_archived=False, expiration_date=datetime.datetime.today() + datetime.timedelta(1), contract_type=contract_type) ContractBaseFactory.create( description='sunfish', financial_id='789', properties=[ ContractPropertyFactory.create(key='foo', value='engine') ], expiration_date=datetime.datetime.today() + datetime.timedelta(1), is_archived=False, contract_type=contract_type) ContractBaseFactory.create( description='sunfish', financial_id='012', properties=[ ContractPropertyFactory.create(key='foo', value='engine') ], expiration_date=datetime.datetime.today() - datetime.timedelta(1), is_archived=False, contract_type=self.contract_type2) # db.session.execute(''' # REFRESH MATERIALIZED VIEW CONCURRENTLY search_view # ''') db.session.commit()
def setUp(self): from flask_migrate import upgrade upgrade() # insert the users/roles self.admin_role = get_a_role('admin') self.superadmin_role = get_a_role('superadmin') self.admin_user = insert_a_user(email='*****@*****.**', role=self.admin_role) self.superadmin_user = insert_a_user(email='*****@*****.**', role=self.superadmin_role) # insert the companies/contracts self.company_1 = insert_a_company(name='ship', insert_contract=False) company_2 = insert_a_company(name='boat', insert_contract=False) contract_type = ContractTypeFactory.create(name='test') self.contract_type2 = ContractTypeFactory.create(name='test2') self.contract1 = ContractBaseFactory.create( description='vessel', companies=[company_2], line_items=[LineItem(description='NAVY')], expiration_date=datetime.datetime.today() + datetime.timedelta(1), is_archived=False, financial_id='123', contract_type=contract_type ) ContractBaseFactory.create( description='sail', financial_id='456', companies=[self.company_1], line_items=[LineItem(description='sunfish')], is_archived=False, expiration_date=datetime.datetime.today() + datetime.timedelta(1), contract_type=contract_type ) ContractBaseFactory.create( description='sunfish', financial_id='789', properties=[ContractPropertyFactory.create(key='foo', value='engine')], expiration_date=datetime.datetime.today() + datetime.timedelta(1), is_archived=False, contract_type=contract_type ) ContractBaseFactory.create( description='sunfish', financial_id='012', properties=[ContractPropertyFactory.create(key='foo', value='engine')], expiration_date=datetime.datetime.today() - datetime.timedelta(1), is_archived=False, contract_type=self.contract_type2 ) # db.session.execute(''' # REFRESH MATERIALIZED VIEW CONCURRENTLY search_view # ''') db.session.commit()
def setUp(self): super(TestOpportunitiesAdminBase, self).setUp() try: mkdir(current_app.config.get('UPLOAD_DESTINATION')) except OSError: rmtree(current_app.config.get('UPLOAD_DESTINATION')) mkdir(current_app.config.get('UPLOAD_DESTINATION')) import_nigp(current_app.config.get('PROJECT_ROOT') + '/purchasing_test/mock/nigp.csv') self.admin_role = insert_a_role('admin') self.staff_role = insert_a_role('staff') self.department1 = DepartmentFactory(name='test') self.opportunity_type = ContractTypeFactory.create(allow_opportunities=True) self.admin = insert_a_user(email='*****@*****.**', role=self.admin_role) self.staff = insert_a_user(email='*****@*****.**', role=self.staff_role) self.document = insert_a_document() self.vendor = Vendor.create(email='*****@*****.**', business_name='foo2') self.opportunity1 = insert_an_opportunity( contact=self.admin, created_by=self.staff, title=u'tést unïcode title', description=u'tést unïcode déscription', is_public=True, is_archived=False, planned_publish=datetime.date.today() + datetime.timedelta(1), planned_submission_start=datetime.date.today() + datetime.timedelta(2), planned_submission_end=datetime.datetime.today() + datetime.timedelta(2), documents=[self.document.id], categories=set([Category.query.first()]) ) self.opportunity2 = insert_an_opportunity( contact=self.admin, created_by=self.staff, is_public=True, is_archived=False, planned_publish=datetime.date.today(), planned_submission_start=datetime.date.today() + datetime.timedelta(2), planned_submission_end=datetime.datetime.today() + datetime.timedelta(2), categories=set([Category.query.first()]) ) self.opportunity3 = insert_an_opportunity( contact=self.admin, created_by=self.staff, is_public=True, is_archived=False, planned_publish=datetime.date.today() - datetime.timedelta(2), planned_submission_start=datetime.date.today() - datetime.timedelta(2), planned_submission_end=datetime.datetime.today() - datetime.timedelta(1), categories=set([Category.query.first()]) ) self.opportunity4 = insert_an_opportunity( contact=self.admin, created_by=self.staff, is_public=True, is_archived=False, planned_publish=datetime.date.today() - datetime.timedelta(1), planned_submission_start=datetime.date.today(), planned_submission_end=datetime.datetime.today() + datetime.timedelta(2), title='TEST TITLE!', categories=set([Category.query.first()]) )
def setUp(self): super(TestConductorSetup, self).setUp() # create a conductor and general staff person self.county_type = ContractTypeFactory.create(**{ 'name': 'County', 'allow_opportunities': True, 'managed_by_conductor': True }) self.department = DepartmentFactory.create(**{'name': 'test department'}) self.conductor_role_id = insert_a_role('conductor') self.staff_role_id = insert_a_role('staff') self.conductor = insert_a_user(email='*****@*****.**', role=self.conductor_role_id) self.staff = insert_a_user(email='*****@*****.**', role=self.staff_role_id) self.conductor2 = insert_a_user(email='*****@*****.**', role=self.conductor_role_id) # create three stages, and set up a flow between them self.stage1 = insert_a_stage( name='stage1', send_notifs=True, post_opportunities=True, default_message='i am a default message' ) self.stage2 = insert_a_stage(name='stage2', send_notifs=True, post_opportunities=False) self.stage3 = insert_a_stage(name='stage3', send_notifs=False, post_opportunities=False) self.flow = insert_a_flow(stage_ids=[self.stage1.id, self.stage2.id, self.stage3.id]) self.flow2 = insert_a_flow(name='test2', stage_ids=[self.stage1.id, self.stage3.id, self.stage2.id]) self.simple_flow = insert_a_flow(name='simple', stage_ids=[self.stage1.id]) # create two contracts self.contract1 = insert_a_contract( contract_type=self.county_type, description='scuba supplies', financial_id=123, expiration_date=datetime.date.today(), properties=[{'key': 'Spec Number', 'value': '123'}], is_visible=True, department=self.department, has_metrics=True ) self.contract2 = insert_a_contract( contract_type=self.county_type, description='scuba repair', financial_id=456, expiration_date=datetime.date.today() + datetime.timedelta(120), properties=[{'key': 'Spec Number', 'value': '456'}], is_visible=True, has_metrics=True ) self.category = CategoryFactory.create() self.login_user(self.conductor) self.detail_view = '/conductor/contract/{}/stage/{}' self.transition_view = '/conductor/contract/{}/stage/{}/'