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_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))
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))
