You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
53 lines
2.0 KiB
53 lines
2.0 KiB
9 years ago
|
#!/usr/bin/python3
|
||
|
|
# (C) 2014 - 2016 Canonical Ltd.
|
||
|
|
#
|
||
|
|
# This program is free software; you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU General Public License as published by
|
||
|
|
# the Free Software Foundation; either version 2 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
PROJECT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
|
|
sys.path.insert(0, PROJECT_DIR)
|
||
|
|
|
||
|
|
from britney2.utils import get_component, allowed_component
|
||
|
|
from britney2.consts import MAIN, RESTRICTED, UNIVERSE, MULTIVERSE
|
||
|
|
|
||
|
|
|
||
|
|
class UtilTests(unittest.TestCase):
|
||
|
|
|
||
|
|
def test_get_component(self):
|
||
|
|
self.assertEqual(get_component('utils'), MAIN)
|
||
|
|
self.assertEqual(get_component('utils'), MAIN)
|
||
|
|
self.assertEqual(get_component('restricted/admin'), RESTRICTED)
|
||
|
|
self.assertEqual(get_component('universe/web'), UNIVERSE)
|
||
|
|
self.assertEqual(get_component('multiverse/libs'), MULTIVERSE)
|
||
|
|
|
||
|
|
def test_allowed_component(self):
|
||
|
|
self.assertTrue(allowed_component(MAIN, MAIN))
|
||
|
|
self.assertFalse(allowed_component(MAIN, UNIVERSE))
|
||
|
|
self.assertFalse(allowed_component(MAIN, MULTIVERSE))
|
||
|
|
self.assertFalse(allowed_component(MAIN, RESTRICTED))
|
||
|
|
|
||
|
|
self.assertTrue(allowed_component(RESTRICTED, MAIN))
|
||
|
|
self.assertFalse(allowed_component(RESTRICTED, UNIVERSE))
|
||
|
|
self.assertFalse(allowed_component(RESTRICTED, MULTIVERSE))
|
||
|
|
self.assertTrue(allowed_component(RESTRICTED, RESTRICTED))
|
||
|
|
|
||
|
|
self.assertTrue(allowed_component(UNIVERSE, MAIN))
|
||
|
|
self.assertTrue(allowed_component(UNIVERSE, UNIVERSE))
|
||
|
|
self.assertFalse(allowed_component(UNIVERSE, MULTIVERSE))
|
||
|
|
self.assertFalse(allowed_component(UNIVERSE, RESTRICTED))
|
||
|
|
|
||
|
|
self.assertTrue(allowed_component(MULTIVERSE, MAIN))
|
||
|
|
self.assertTrue(allowed_component(MULTIVERSE, UNIVERSE))
|
||
|
|
self.assertTrue(allowed_component(MULTIVERSE, MULTIVERSE))
|
||
|
|
self.assertTrue(allowed_component(MULTIVERSE, RESTRICTED))
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|