def test_import_from_file(self):
CP = CronJobParser("./testdata/crontab")
config_raw = CP.get()
self.assertEqual(len(config_raw), 8)
config = CP.config
self.assertEqual(len(config), 2)
self.assertTrue("assignments" in config)
self.assertTrue("cronjobs" in config)
self.assertEqual(len(config.get("cronjobs")), 6)
self.assertTrue("PATH" in config.get("assignments"))
self.assertTrue("SHELL" in config.get("assignments"))
def test_save(self):
CP = CronJobParser(content=CRONTAB)
config = CP.config
self.assertEqual(len(config.get("cronjobs")), 6)
cronj = config.get("cronjobs")[0]
self.assertEqual(cronj.minute, "17")
self.assertEqual(cronj.dow, "*")
# now save it to a file
CP.save(outfile="./testdata/tmpfile")
# Read the file again!
CP2 = CronJobParser("./testdata/tmpfile")
config = CP2.config
self.assertEqual(len(config.get("cronjobs")), 6)
cronj = config.get("cronjobs")[0]
self.assertEqual(cronj.minute, "17")
self.assertEqual(cronj.dow, "*")
os.unlink("./testdata/tmpfile")
