def test_generated_files(self): due_date = datetime.datetime(2011, 10, 24) sale = self._create_sale(1666, due_date=due_date) generator = NFeGenerator(sale, self.store) # If we generate random cnf, the test will always fail _get_random_cnf = NFeIdentification._get_random_cnf NFeIdentification._get_random_cnf = lambda s: 10000001 # Mimic today behavior _get_today_date = NFeGenerator._get_today_date NFeGenerator._get_today_date = lambda s: due_date generator.sale_id = 1234 generator.payment_ids = [4321] generator.generate() NFeIdentification._get_random_cnf = _get_random_cnf NFeGenerator._get_today_date = _get_today_date basedir = get_tests_datadir('plugins') expected = os.path.join(basedir, "nfe-expected.txt") output = os.path.join(basedir, "nfe-output.txt") if not os.path.isfile(expected): with open(expected, 'wb') as fp: fp.write(strip_accents(generator._as_txt())) return with open(output, 'wb') as fp: fp.write(strip_accents(generator._as_txt())) # Diff and compare diff = diff_files(expected, output) os.unlink(output) self.failIf(diff, '%s\n%s' % ("Files differ, output:", diff))
def test_invalid_cnpj(self): sale = self._create_sale(2666) company = sale.branch.person.company company.cnpj = u'123.321.678/4567-90' generator = NFeGenerator(sale, self.store) generator.sale_id = 2345 generator.payment_ids = [5432] self.assertRaises(ModelDataError, generator.generate)