lars
707fb71476
unified errorExit function instead of explicite sys.exit calls add config_file argument to CryptoBoxProps add getCapacity method to CryptoBoxContainer removed "classes.txt"
122 lines
3.7 KiB
Python
Executable file
122 lines
3.7 KiB
Python
Executable file
#!/usr/bin/env python2.4
|
|
|
|
import unittest
|
|
|
|
class CryptoBoxPropsDeviceTests(unittest.TestCase):
|
|
import CryptoBox
|
|
cb = CryptoBox.CryptoBoxProps()
|
|
|
|
def testAllowedDevices(self):
|
|
'''isDeviceAllowed should accept permitted devices'''
|
|
self.assertTrue(self.cb.isDeviceAllowed("/dev/loop"))
|
|
self.assertTrue(self.cb.isDeviceAllowed("/dev/loop1"))
|
|
self.assertTrue(self.cb.isDeviceAllowed("/dev/loop/urgd"))
|
|
self.assertTrue(self.cb.isDeviceAllowed("/dev/usb/../loop1"))
|
|
|
|
def testDeniedDevices(self):
|
|
'''isDeviceAllowed should fail with not explicitly allowed devices'''
|
|
self.assertFalse(self.cb.isDeviceAllowed("/dev/hda"))
|
|
self.assertFalse(self.cb.isDeviceAllowed("/dev/loopa/../hda"))
|
|
self.assertFalse(self.cb.isDeviceAllowed("/"))
|
|
|
|
|
|
class CryptoBoxPropsConfigTests(unittest.TestCase):
|
|
'''test here if everything with the config turns right'''
|
|
import os
|
|
import CryptoBox
|
|
|
|
files = {
|
|
"configFileOK" : "cbox-test_ok.conf",
|
|
"configFileBroken" : "cbox-test_broken.conf",
|
|
"nameDBFile" : "cryptobox_names.db",
|
|
"logFile" : "cryptobox.log",
|
|
"tmpdir" : "cryptobox-mnt" }
|
|
tmpdirname = ""
|
|
filenames = {}
|
|
configContentOK = """
|
|
[Main]
|
|
AllowedDevices = /dev/loop
|
|
DefaultVolumePrefix = "Data "
|
|
DataDir = %s
|
|
NameDatabase = cryptobox_names.db
|
|
[System]
|
|
User = 1000
|
|
Group = 1000
|
|
MountParentDir = %s/mnt
|
|
DefaultCipher = aes-cbc-essiv:sha256
|
|
[Log]
|
|
Level = debug
|
|
Destination = file
|
|
#Details = %s/cryptobox.log
|
|
Details = /tmp/cryptobox.log
|
|
[Programs]
|
|
blkid = /sbin/blkid
|
|
cryptsetup = /sbin/cryptsetup
|
|
super = /usr/bin/super
|
|
CryptoBoxRootActions = CryptoBoxRootActions"""
|
|
|
|
configContentBroken = """
|
|
[Main]
|
|
AllowedDevices = /dev/loop
|
|
DefaultVolumePrefix = "Data "
|
|
#DataDir = %s
|
|
NameDatabase = cryptobox_names.db
|
|
[System]
|
|
User = 1000
|
|
Group = 1000
|
|
MountParentDir = %s/mnt
|
|
DefaultCipher = aes-cbc-essiv:sha256
|
|
[Log]
|
|
Level = debug
|
|
Destination = file
|
|
#Details = %s/cryptobox.log
|
|
Details = /tmp/cryptobox.log
|
|
[Programs]
|
|
blkid = /sbin/blkid
|
|
cryptsetup = /sbin/cryptsetup
|
|
super = /usr/bin/super
|
|
CryptoBoxRootActions = CryptoBoxRootActions"""
|
|
|
|
def setUp(self):
|
|
'''generate all files in tmp and remember the names'''
|
|
import tempfile
|
|
os = self.os
|
|
self.tmpdirname = tempfile.mkdtemp(prefix="cbox-")
|
|
for file in self.files.keys():
|
|
self.filenames[file] = os.path.join(self.tmpdirname, self.files[file])
|
|
cf = open(self.filenames["configFileOK"], "w")
|
|
cf.write(self.configContentOK % (self.tmpdirname, self.tmpdirname, self.tmpdirname))
|
|
cf.close()
|
|
cf = open(self.filenames["configFileBroken"], "w")
|
|
cf.write(self.configContentBroken % (self.tmpdirname, self.tmpdirname, self.tmpdirname))
|
|
cf.close()
|
|
|
|
|
|
def tearDown(self):
|
|
'''remove the created tmpfiles'''
|
|
os = self.os
|
|
os.chdir(self.tmpdirname)
|
|
# remove temp files
|
|
for file in self.filenames.values():
|
|
if os.path.exists(file):
|
|
os.remove(file)
|
|
# remove temp dir
|
|
os.rmdir(self.tmpdirname)
|
|
|
|
|
|
def testConfigInit(self):
|
|
'''Check various branches of config file loading'''
|
|
self.assertRaises("ConfigError", self.CryptoBox.CryptoBoxProps,"/invalid/path/to/config/file")
|
|
self.assertRaises("ConfigError", self.CryptoBox.CryptoBoxProps,"/etc/shadow")
|
|
self.CryptoBox.CryptoBoxProps()
|
|
self.CryptoBox.CryptoBoxProps(self.filenames["configFileOK"])
|
|
self.assertRaises("ConfigError", self.CryptoBox.CryptoBoxProps,[])
|
|
self.assertRaises("ConfigError", self.CryptoBox.CryptoBoxProps,self.filenames["configFileBroken"])
|
|
# TODO: check details of different ConfigError-exceptions
|
|
# TODO: use different kind of broken setups ...
|
|
self.assertTrue(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|