cryptonas-branches/pythonrewrite/bin2/unittests.CryptoBox.py

#!/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()