def testRunEnterDaysLessThanOne(self):
schedulefiledata = FileTester.read_file(
FileTester.test_enter_lessthan1
)
tempschedulefile = FileTester.write_to_temp_file(
FileTester.test_enter_lessthan1,
schedulefiledata
)
schedulefile = ScheduleFile(tempschedulefile)
templedgerfile = FileTester.create_temp_file('')
ledgerfile = LedgerFile(templedgerfile)
scheduler = Scheduler(ledgerfile, schedulefile)
scheduler.run()
ledgerfile.write_file()
schedulefile.write_file()
schedulefile_actual = FileTester.read_file(tempschedulefile)
schedulefile_expected = FileTester.read_file(
FileTester.test_enter_lessthan1
)
os.remove(templedgerfile)
os.remove(tempschedulefile)
self.assertEqual(
schedulefile_expected,
schedulefile_actual
)
def run_it(self, before_date, after_date, schedule, enter_days=7):
schedulefiledata = self.get_schedule_file(
util.get_date_string(before_date),
schedule,
enter_days
)
tempschedulefile = FileTester.write_to_temp_file(
FileTester.testdir + 'run_it_schedule_file',
schedulefiledata
)
schedulefile = ScheduleFile(tempschedulefile)
templedgerfile = FileTester.write_to_temp_file(
FileTester.testdir + 'run_it_ledger_file',
''
)
ledgerfile = LedgerFile(templedgerfile)
scheduler = Scheduler(ledgerfile, schedulefile)
scheduler.run()
ledgerfile.write_file()
schedulefile.write_file()
schedulefile_actual = FileTester.read_file(tempschedulefile)
schedulefile_expected = self.get_schedule_file(
util.get_date_string(after_date),
schedule,
enter_days
)
os.remove(templedgerfile)
os.remove(tempschedulefile)
self.assertEqual(schedulefile_expected, schedulefile_actual)
