def setUp(self): super(TestBeaconJobs, self).setUp() self.yesterday = datetime.datetime.today() - datetime.timedelta(days=1) today = datetime.datetime.today() tomorrow = datetime.datetime.today() + datetime.timedelta(days=1) self.category = CategoryFactory.create() self.admin = UserFactory.create() self.opportunity = OpportunityFactory.create( is_public=True, planned_publish=today, planned_submission_start=today, planned_submission_end=tomorrow, categories=set([self.category]), created_by=self.admin, published_at=today ) self.opportunity2 = OpportunityFactory.create( is_public=True, planned_publish=self.yesterday, planned_submission_start=today, planned_submission_end=tomorrow, publish_notification_sent=True, categories=set([self.category]), created_by=self.admin, published_at=self.yesterday ) self.opportunity3 = OpportunityFactory.create( is_public=False, planned_publish=today, planned_submission_start=today, planned_submission_end=tomorrow, publish_notification_sent=False, categories=set([self.category]), created_by=self.admin, published_at=today ) self.opportunity4 = OpportunityFactory.create( is_public=True, planned_publish=self.yesterday, planned_submission_start=self.yesterday, planned_submission_end=today, publish_notification_sent=True, categories=set([self.category]), created_by=self.admin, published_at=self.yesterday ) VendorFactory.create(opportunities=set([self.opportunity])) VendorFactory.create(categories=set([self.category]))
def setUp(self): self.asker = VendorFactory.create(email='*****@*****.**') super(TestOpportunityQuestions, self).setUp() self.q1 = QuestionFactory.create(opportunity=self.opportunity3, asked_by=self.asker) QuestionFactory.create(answer_text='an answer', opportunity=self.opportunity3) self.opportunity3.created_by = self.staff self.opportunity3.save() self.answer_url = '/beacon/admin/opportunities/{}/questions/{}'.format(self.opportunity3.id, self.q1.id) self.question_url = '/beacon/admin/opportunities/{}/questions'.format(self.opportunity3.id)
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') + '/test/mock/nigp.csv') self.department1 = DepartmentFactory(name='test') AcceptedEmailDomainsFactory.create(domain='foo.com') self.admin = UserFactory.create( email='*****@*****.**', roles=[RoleFactory.create(name='admin')] ) self.staff = UserFactory.create( email='*****@*****.**', roles=[RoleFactory.create(name='staff')] ) self.document = RequiredBidDocumentFactory.create() self.vendor = VendorFactory.create(email='*****@*****.**', business_name='foo2') self.opportunity1 = OpportunityFactory.create( 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), vendor_documents_needed=[self.document.id], categories=(Category.query.first(),) ).save() self.opportunity2 = OpportunityFactory.create( 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=(Category.query.first(),), enable_qa=False, qa_start=None, qa_end=None ).save() self.opportunity3 = OpportunityFactory.create( 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=(Category.query.first(),) ).save() self.opportunity4 = OpportunityFactory.create( 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=(Category.query.first(),) ).save()
def test_signup_download_staff(self): # insert some vendors categories = Category.query.all() VendorFactory.create(categories=(categories[0],)) VendorFactory.create(categories=categories) self.login_user(self.staff) request = self.client.get('/beacon/admin/signups') self.assertEquals(request.mimetype, 'text/tsv') self.assertEquals( request.headers.get('Content-Disposition'), 'attachment; filename=vendors-{}.tsv'.format(datetime.date.today()) ) # python adds an extra return character to the end, # so we chop it off. we should have the header row and # the two rows we inserted above tsv_data = request.data.split('\n')[:-1] self.assertEquals(len(tsv_data), 4) for row in tsv_data: self.assertEquals(len(row.split('\t')), 11)
def test_signup_download_staff(self): # insert some vendors categories = Category.query.all() VendorFactory.create(categories=(categories[0], )) VendorFactory.create(categories=categories) self.login_user(self.staff) request = self.client.get('/beacon/admin/signups') self.assertEquals(request.mimetype, 'text/tsv') self.assertEquals( request.headers.get('Content-Disposition'), 'attachment; filename=vendors-{}.tsv'.format( datetime.date.today())) # python adds an extra return character to the end, # so we chop it off. we should have the header row and # the two rows we inserted above tsv_data = request.data.split('\n')[:-1] self.assertEquals(len(tsv_data), 4) for row in tsv_data: self.assertEquals(len(row.split('\t')), 11)
def setUp(self): super(TestOpportunityPublic, self).setUp() self.opportunity3.is_public = False self.opportunity3.categories = set([Category.query.all()[-1]]) self.opportunity3.planned_submission_end = datetime.datetime.today() + datetime.timedelta(days=5) self.vendor = VendorFactory.create( business_name='foobar', email='*****@*****.**', categories=set([Category.query.all()[-1]]) ) db.session.commit() self.opportunity1.created_by = self.staff self.opportunity3.created_by = self.staff
def setUp(self): super(TestOpportunityPublic, self).setUp() self.opportunity3.is_public = False self.opportunity3.categories = set([Category.query.all()[-1]]) self.opportunity3.planned_submission_end = datetime.datetime.today( ) + datetime.timedelta(days=5) self.vendor = VendorFactory.create(business_name='foobar', email='*****@*****.**', categories=set( [Category.query.all()[-1]])) db.session.commit() self.opportunity1.created_by = self.staff self.opportunity3.created_by = self.staff