def setUp(self):
self.whitelist = Whitelist()
class TestWhitelist(unittest.TestCase):
def setUp(self):
self.whitelist = Whitelist()
def test_include(self):
(self.whitelist
.include(['test', 'inc_test'], 'inc_test')
.include(['test2'], 'test2')
)
args = {
'include': '["test"]'
}
out = self.whitelist.process(args)
includes = out['include']
out.pop('include')
# The correct include keys
self.assertTrue(includes['inc_test'])
self.assertFalse(includes['test2'])
# Make sure there are no keys in non-includes
self.assertTrue(len(out) == 0)
def test_include__empty(self):
out = self.whitelist.process({})
self.assertTrue(out.has_key('include'))
def test_include__mixed(self):
(self.whitelist
.include(['test', 'inc_test'], 'inc_test')
.include(['test', 'test2'], 'test2')
)
args = {
'include': '["test"]',
}
out = self.whitelist.process(args)
includes = out['include']
out.pop('include')
# The correct include keys
self.assertTrue(includes['inc_test'])
self.assertTrue(includes['test2'])
self.assertTrue(len(out) == 0)
@unittest.skip("TODO: Whitelists still need fix-up")
def test_include__remove(self):
(self.whitelist
.include(['test', 'inc_test'], 'inc_test')
.include(['test', 'test2'], 'test2')
.remove('test2')
)
args = {
'include': '["test"]',
}
out = self.whitelist.process(args)
includes = out['include']
# The correct include keys
self.assertTrue(includes['inc_test'])
self.assertFalse(includes.has_key('test2'))
self.assertTrue(len(out) == 0)
def test_fields(self):
(self.whitelist
.add(Field(['test'], 'test', int))
.add(Field(['other'], 'test2', int))
.add(Field(['test3'], 'test3', int))
)
args = {
'test': '20',
'other': '"12"',
}
out = self.whitelist.process(args)
self.assertEquals(20, out['test'])
self.assertEquals(12, out['test2'])
def test_fields__duplicate_fields(self):
(self.whitelist
.add(Field(['test'], 'test', int))
)
with self.assertRaises(KeyError):
self.whitelist.add(Field(['key'], 'test'))
