def test_next_number(self): """test next_number function""" base = '000' expect = '001' result = next_number('000', base) self.assertEqual(expect, result) expect = '10' result = next_number('09') self.assertEqual(expect, result) self.assertRaises(Exception, next_number, '99')
def test_create_dir(self): """test create_dir function""" topdir = '/tmp' system = 'das_test' try: clean_dirs(os.path.join(topdir, system)) os.removedirs(os.path.join(topdir, system)) except: pass base = '00' idir = create_dir(topdir, system, base=base, filesperdir=5) expect = '%s/%s/%s/%s/%s/%s' % \ (topdir, system, yyyymmdd(), hour(), base, base) self.assertEqual(expect, idir) # create 5 files in our recent dir and check that new dir will be created for i in range(0, 6): fdesc = open(os.path.join(idir, str(i)), 'w') fdesc.write(str(i)) fdesc.close() jdir = create_dir(topdir, system, base=base, filesperdir=5) expect2 = '%s/%s/%s/%s/%s/%s' % \ (topdir, system, yyyymmdd(), hour(), base, next_number(base, base)) self.assertEqual(expect2, jdir) for i in range(0, 6): os.remove(os.path.join(idir, str(i))) ndir = '%s/%s/%s/%s/%s' % (topdir, system, yyyymmdd(), hour(), base) clean_dirs(ndir) ndir = '%s/%s/%s/%s' % (topdir, system, yyyymmdd(), hour()) clean_dirs(ndir) ndir = '%s/%s/%s' % (topdir, system, yyyymmdd()) clean_dirs(ndir) ndir = '%s/%s' % (topdir, system) clean_dirs(ndir) os.removedirs(os.path.join(topdir, system))