def test_add_file(self):
si = SatoriImage()
si.add_file("/etc/shadow")
# print(si)
ret = si.get_dir_contents("/etc")
# print(ret)
self.assertTrue('shadow' in ret)
def test_listing_non_dir(self):
si = SatoriImage()
si.add_file("/etc/shadow")
try:
con = si.get_dir_contents('/etc/shadow')
except NotADirectoryError:
self.assertTrue(True)
def test_multiple_files(self):
si = SatoriImage()
si.add_file("/etc/shadow")
si.add_file("/etc/passwd")
si.add_file("/etc/sudoers")
con = si.get_dir_contents('/etc')
self.assertTrue(len(con) == 3)
self.assertTrue('shadow' in con)
self.assertTrue('passwd' in con)
self.assertTrue('sudoers' in con)
def test_paths(self):
si = SatoriImage()
si.add_file("/etc/shadow")
si.add_file("/etc/dir/")
self.assertTrue(si.is_dir("/etc/"))
self.assertTrue(si.is_dir("/"))
self.assertTrue(si.is_dir("/etc/dir"))
self.assertTrue(si.is_dir("/etc/dir/"))
self.assertTrue(si.is_dir("/etc///dir/"))
self.assertTrue(si.is_dir("//"))
self.assertTrue(si.is_dir("/etc///"))
self.assertTrue(si.is_dir("//etc"))
class test_Serializers(unittest.TestCase):
def setUp(self):
# filenames = os.random(4)
self.image = SatoriImage()
self.image.add_file("/etc/passwd") # Add a file
str_pkl_com = SatoriPickler()
str_pkl = SatoriPickler(compress=False)
str_jsn_com = SatoriJsoner()
str_jsn = SatoriJsoner(compress=False)
self.filers = [
str_jsn_com,
str_jsn,
str_pkl_com,
str_pkl,
]
try:
os.mkdir("gen_images")
except FileExistsError:
pass
os.chdir("gen_images")
for serializer in self.filers:
serializer.write(self.image, 'test')
os.listdir('.')
def test_create_image(self):
for serializer in self.filers:
image = serializer.read('test', suffixed=False)
self.assertTrue(image == self.image)
def tearDown(self):
for filename in os.listdir('.'):
os.unlink(filename)
# print("Cleaning '%s'" % filename)
pass
os.chdir("..")
os.rmdir('gen_images')
def test_creating_directories(self):
si = SatoriImage()
si.add_file('/etc/sshd/') # trailing '/' creates dir
self.assertTrue(si.is_dir('/etc/sshd'))
class test_EmptyImage(unittest.TestCase):
def setUp(self):
self.si = SatoriImage()
def test_add_file(self):
si = SatoriImage()
si.add_file("/etc/shadow")
# print(si)
ret = si.get_dir_contents("/etc")
# print(ret)
self.assertTrue('shadow' in ret)
def test_non_existent_file(self):
self.si.add_file("/etc/shadow")
try:
self.si.get_dir_contents('/asjnas')
except FileNotFoundError:
self.assertTrue(True)
def test_multiple_files(self):
si = SatoriImage()
si.add_file("/etc/shadow")
si.add_file("/etc/passwd")
si.add_file("/etc/sudoers")
con = si.get_dir_contents('/etc')
self.assertTrue(len(con) == 3)
self.assertTrue('shadow' in con)
self.assertTrue('passwd' in con)
self.assertTrue('sudoers' in con)
def test_paths(self):
si = SatoriImage()
si.add_file("/etc/shadow")
si.add_file("/etc/dir/")
self.assertTrue(si.is_dir("/etc/"))
self.assertTrue(si.is_dir("/"))
self.assertTrue(si.is_dir("/etc/dir"))
self.assertTrue(si.is_dir("/etc/dir/"))
self.assertTrue(si.is_dir("/etc///dir/"))
self.assertTrue(si.is_dir("//"))
self.assertTrue(si.is_dir("/etc///"))
self.assertTrue(si.is_dir("//etc"))
def test_listing_non_dir(self):
si = SatoriImage()
si.add_file("/etc/shadow")
try:
con = si.get_dir_contents('/etc/shadow')
except NotADirectoryError:
self.assertTrue(True)
def test_creating_directories(self):
si = SatoriImage()
si.add_file('/etc/sshd/') # trailing '/' creates dir
self.assertTrue(si.is_dir('/etc/sshd'))
# si.listdir('/etc/sshd')
def test_setting_mult_attrs(self):
si = SatoriImage()
si.set_multiple_attributes('/tmp/test', ('ext1', 'value1'),
('ext2', 'value2'), ('ext3', 'value3'),
force_create=True)
self.assertTrue(si.get_attribute('/tmp/test', 'ext2') == 'value2')
# def test_stat(self):
# si = SatoriImage()
def test_classes(self):
si = SatoriImage()
si.add_class("class1", section=_DATA_SECTION)
try:
si.add_class("class1", section=_DATA_SECTION)
except KeyError:
self.assertTrue(True) # Readding a class throws exception
cd = si.get_class('class1', section=_DATA_SECTION)
self.assertTrue(not cd) # New class yields empty dict
try:
cd2 = si.get_class('class2', section=_DATA_SECTION)
except KeyError:
self.assertTrue(True) # Non-existent class throws Exception
# Existent classes are compatible
sysinfo = si.get_class('filesystem', section=_DATA_SECTION)