def test_generated_files(self):
due_date = datetime.datetime(2011, 10, 24)
sale = self._create_sale(1666, due_date=due_date)
sale.identifier = 1234
for p in sale.payments:
p.identifier = 4321
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.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 compare_files(sfile, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
sfile.write(output)
expected = get_tests_datadir('plugins', expected)
diff = diff_files(expected, output)
os.unlink(output)
return diff
def compare_sintegra_file(sfile, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
sfile.write(output)
expected = get_tests_datadir(expected)
diff = diff_files(expected, output)
os.unlink(output)
if diff:
raise AssertionError('%s\n%s' % ("Files differ, output:", diff))
def compare_sintegra_file(sfile, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
sfile.write(output)
expected = get_tests_datadir(expected)
diff = diff_files(expected, output)
os.unlink(output)
if diff:
raise AssertionError('%s\n%s' % ("Files differ, output:", diff))
def _compare_files(self, content, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
with open(output, 'w') as fh:
fh.write(content)
expected = self._get_expected(expected, content)
diff = diff_files(expected, output)
os.unlink(output)
if diff:
raise AssertionError('%s\n%s' % ("Files differ, output:", diff))
def _compare_files(self, content, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
with open(output, 'w') as fh:
fh.write(content)
expected = self._get_expected(expected, content)
diff = diff_files(expected, output)
os.unlink(output)
if diff:
raise AssertionError('%s\n%s' % ("Files differ, output:", diff))
def compare_invoice_file(invoice, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
fp = open(output, 'w')
for n, page in enumerate(invoice.generate_pages()):
fp.write('-- PAGE %d - START ----\n' % (n + 1, ))
for line in page:
fp.write(line.tostring())
fp.write('-- PAGE %d - END ----\n' % (n + 1, ))
fp.close()
expected = get_tests_datadir(expected)
diff = diff_files(expected, output)
os.unlink(output)
if diff:
raise AssertionError('%s\n%s' % ("Files differ, output:", diff))
def compare_invoice_file(invoice, basename):
expected = basename + '-expected.txt'
output = basename + '-output.txt'
fp = open(output, 'w')
for n, page in enumerate(invoice.generate_pages()):
fp.write('-- PAGE %d - START ----\n' % (n + 1, ))
for line in page:
fp.write(line.tostring())
fp.write('-- PAGE %d - END ----\n' % (n + 1, ))
fp.close()
expected = get_tests_datadir(expected)
diff = diff_files(expected, output)
os.unlink(output)
if diff:
raise AssertionError('%s\n%s' % ("Files differ, output:", diff))
def _test_generated_files(self, new_client=None):
due_date = datetime.datetime(2011, 10, 24, 0, 0, 0, 0)
sale = self._create_sale(1666, due_date=due_date)
sale.identifier = 1234
if new_client:
sale.client = new_client
for p in sale.payments:
p.identifier = 4321
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 now_datetime behavior
_get_now_datetime = NFeGenerator._get_now_datetime
NFeGenerator._get_now_datetime = lambda s: due_date
generator.generate()
NFeIdentification._get_random_cnf = _get_random_cnf
NFeGenerator._get_now_datetime = _get_now_datetime
basedir = get_tests_datadir('plugins')
if new_client is None:
expected = os.path.join(basedir, "nfe-expected.txt")
elif isinstance(sale.get_client_role(), Individual):
expected = os.path.join(basedir, "individual-nfe-expected.txt")
else:
expected = os.path.join(basedir, "company-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()).encode())
return
with open(output, 'wb') as fp:
fp.write(strip_accents(generator._as_txt()).encode())
# Diff and compare
diff = diff_files(expected, output)
os.unlink(output)
self.assertFalse(diff, '%s\n%s' % ("Files differ, output:", diff))
def _test_generated_files(self, new_client=None):
due_date = datetime.datetime(2011, 10, 24, 0, 0, 0, 0)
sale = self._create_sale(1666, due_date=due_date)
sale.identifier = 1234
if new_client:
sale.client = new_client
for p in sale.payments:
p.identifier = 4321
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 now_datetime behavior
_get_now_datetime = NFeGenerator._get_now_datetime
NFeGenerator._get_now_datetime = lambda s: due_date
generator.generate()
NFeIdentification._get_random_cnf = _get_random_cnf
NFeGenerator._get_now_datetime = _get_now_datetime
basedir = get_tests_datadir('plugins')
if new_client is None:
expected = os.path.join(basedir, "nfe-expected.txt")
elif isinstance(sale.get_client_role(), Individual):
expected = os.path.join(basedir, "individual-nfe-expected.txt")
else:
expected = os.path.join(basedir, "company-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 _diff_expected(self, report_class, expected_name, *args, **kwargs):
basedir = get_tests_datadir('reporting')
expected = os.path.join(basedir, '%s.html' % expected_name)
output = os.path.join(basedir, '%s-tmp.html' % expected_name)
def save_report(filename, *args, **kwargs):
report = report_class(filename, *args, **kwargs)
report.adjust_for_test()
report.save_html(filename)
if not os.path.isfile(expected):
save_report(expected, *args, **kwargs)
return
save_report(output, *args, **kwargs)
# Diff and compare
diff = diff_files(expected, output)
os.unlink(output)
self.failIf(diff, '%s\n%s' % ("Files differ, output:", diff))
