def set_up(self): self.FILE_NAME = "20141020_2IAEW4.xlsx" self.acceptance_suite = AcceptanceSuite(self.client, self.application) self.pointsale_suite = PointSaleSuite(self.client, self.application) self.application_suite = ApplicationSuite( self.client, self.application) self.provider_suite = ProviderTestSuite(self.client, self.application) self.test_provider_id, _, _, _ = ( self.provider_suite.create_test_provider()) self.invoice_suite = MailInvoiceTestSuite( self.client, self.application) with self.application.app_context(): pointsale = self.pointsale_suite.create_test_pointsale( name=u"ШШК", address=u"Наб. Челны") self.pointsale_id = pointsale.id
class AcceptanceTest(BaseTestCase): def set_up(self): self.FILE_NAME = "20141020_2IAEW4.xlsx" self.acceptance_suite = AcceptanceSuite(self.client, self.application) self.pointsale_suite = PointSaleSuite(self.client, self.application) self.application_suite = ApplicationSuite( self.client, self.application) self.provider_suite = ProviderTestSuite(self.client, self.application) self.test_provider_id, _, _, _ = ( self.provider_suite.create_test_provider()) self.invoice_suite = MailInvoiceTestSuite( self.client, self.application) with self.application.app_context(): pointsale = self.pointsale_suite.create_test_pointsale( name=u"ШШК", address=u"Наб. Челны") self.pointsale_id = pointsale.id def invoice_(self): """ Принимаем накладную из файла FILE_NAME """ self.invoice_suite.handle_invoice( datetime=datetime.now(), file_name=self.FILE_NAME, mail_id=1) return Invoice.query.first().id def price_to_good(self, good_id): return Good.query.filter( Good.id == good_id ).one().price def priceparish_to_good(self, good_id): good = Good.query.get(good_id) return PriceParish.query.filter( PriceParish.commodity_id == good.commodity_id, PriceParish.number_local_from == good.number_local, PriceParish.number_global_from == good.number_global ) def invoice_count(self): return Invoice.query.count() def acceptance_count(self): return Acceptance.query.count() def acceptance_items(self, id): return AcceptanceItems.query.filter( AcceptanceItems.acceptance_id == id ) def acceptance_item(self, id): return AcceptanceItems.query.filter( AcceptanceItems.id == id ).one() def pointsale_items(self, id): return PointSaleItem.query.filter( PointSaleItem.pointsale_id == id ) def invoice_items(self, id): return InvoiceItem.query.filter( InvoiceItem.invoice_id == id ) def success_stories_mail(self, date): invoice_id = self.invoice_() return invoice_id, self.acceptance_suite.create( date=date, type=MAIL, invoice_id=invoice_id, pointsale_id=self.pointsale_id)