def test_read_flag(self): flag = "%s/test_read_flag" % self.tmpdir ts = datetime.now() utils.set_reboot_flag(flag_loc=flag) fts = utils.get_reboot_flag_timestamp(flag) fmt_str = "%Y-%m-%d %H:%M:%S" self.assertEqual(fts.strftime(fmt_str), ts.strftime(fmt_str))
def test_set_flag(self): flag = "%s/test_set_flag" % self.tmpdir utils.set_reboot_flag(flag_loc=flag) self.assertTrue(os.path.exists(flag))