diff -r fabb71550e51 -r 30af52a271a1 tests/ext_urltree.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/ext_urltree.py Sun Jan 09 00:50:21 2011 +0200 @@ -0,0 +1,625 @@ +# :set encoding=utf8 +""" + Unit tests for svv.ext.urltree +""" + +import unittest + +from svv.ext import urltree + +class TestLabelValue (unittest.TestCase) : + class dummylabel : + key = 'foo' + + def test_str_default (self) : + self.assertEqual(str(urltree.LabelValue(self.dummylabel, 'bar', True)), "foo") + + def test_str_value (self) : + self.assertEqual(str(urltree.LabelValue(self.dummylabel, 'bar', False)), "foo='bar'") + +class TestLabel (unittest.TestCase) : + def setUp (self) : + self.config = urltree.URLConfig() + + def _test_parse (self, mask, _type, defaults={}, **attrs) : + label = urltree.Label.parse(mask, defaults, self.config) + + self.assertTrue(isinstance(label, _type)) + + for k, v in attrs.iteritems() : + self.assertEqual(getattr(label, k), v) + + def test_parse_empty (self) : + self._test_parse("", urltree.EmptyLabel) + + def test_parse_static (self) : + self._test_parse("foo", urltree.StaticLabel, name="foo") + + def test_parse_static_valid (self) : + self._test_parse("foo_bar", urltree.StaticLabel, name="foo_bar") + self._test_parse("foo2", urltree.StaticLabel, name="foo2") + self._test_parse("2foo.q", urltree.StaticLabel, name="2foo.q") + + def test_parse_static_invalid (self) : + self.assertRaises(urltree.URLError, urltree.Label.parse, "foo/bar", {}, self.config) + + def test_parse_value_plain (self) : + self._test_parse("{foo}", urltree.SimpleValueLabel, key="foo", default=None, type_name=None, type=urltree.URLConfig.BUILTIN_TYPES[None]) + + def test_parse_value_default1 (self) : + self._test_parse("{foo=bar1}", urltree.SimpleValueLabel, key="foo", default="bar1", type_name=None) + + def test_parse_value_default2 (self) : + self._test_parse("{foo}", urltree.SimpleValueLabel, dict(foo="bar2"), key="foo", default="bar2", type_name=None) + + def test_parse_value_default3 (self) : + self._test_parse("{foo=bar1}", urltree.SimpleValueLabel, dict(foo="bar3"), key="foo", default="bar3", type_name=None) + + def test_parse_value_type_str (self) : + self._test_parse("{foo:str}", urltree.SimpleValueLabel, key="foo", type_name="str", type=urltree.URLConfig.BUILTIN_TYPES['str']) + + def test_parse_value_type_int (self) : + self._test_parse("{foo:int}", urltree.SimpleValueLabel, key="foo", type_name="int", type=urltree.URLConfig.BUILTIN_TYPES['int']) + + def test_parse_invalid_type (self) : + self.assertRaises(KeyError, self._test_parse, "{foo:bar}", None) + +class TestEmptyLabel (unittest.TestCase) : + def setUp (self) : + self.label = urltree.EmptyLabel() + + def test_eq (self) : + self.assertTrue(self.label == urltree.EmptyLabel()) + self.assertFalse(self.label == urltree.StaticLabel("foo")) + + def test_match (self) : + self.assertFalse(self.label.match()) + self.assertFalse(self.label.match("foo")) + self.assertTrue(self.label.match("")) + + def test_build (self) : + self.assertEqual(self.label.build({}), "") + + def test_build_default (self) : + self.assertEqual(self.label.build_default({}), (False, "")) + + def test_str (self) : + self.assertEqual(str(self.label), "") + +class TestStaticLabel (unittest.TestCase) : + def setUp (self) : + self.label = urltree.StaticLabel("test") + + def test_eq (self) : + self.assertTrue(self.label == urltree.StaticLabel("test")) + self.assertFalse(self.label == urltree.StaticLabel("Test")) + self.assertFalse(self.label == urltree.EmptyLabel()) + + def test_match (self) : + self.assertFalse(self.label.match()) + self.assertFalse(self.label.match("foo")) + self.assertTrue(self.label.match("test")) + + def test_build (self) : + self.assertEqual(self.label.build({}), "test") + + def test_build_default (self) : + self.assertEqual(self.label.build_default({}), (False, "test")) + + def test_str (self) : + self.assertEqual(str(self.label), "test") + +class TestSimpleValueLabel (unittest.TestCase) : + def setUp (self) : + self.label = urltree.SimpleValueLabel("test", 'int', urltree.URLConfig.BUILTIN_TYPES['int'], None) + self.label_default_0 = urltree.SimpleValueLabel("test", 'int', urltree.URLConfig.BUILTIN_TYPES['int'], 0) + self.label_default_1 = urltree.SimpleValueLabel("test", 'int', urltree.URLConfig.BUILTIN_TYPES['int'], 1) + self.label_str = urltree.SimpleValueLabel("test", None, urltree.URLConfig.BUILTIN_TYPES[None], None) + self.label_str_default = urltree.SimpleValueLabel("test", None, urltree.URLConfig.BUILTIN_TYPES[None], 'def') + + def test_eq (self) : + self.assertTrue(self.label == urltree.SimpleValueLabel("test", 'str', None, 1)) + self.assertFalse(self.label == urltree.StaticLabel("Test")) + self.assertFalse(self.label == urltree.EmptyLabel()) + + def _check_value (self, label, label_value, value, is_default) : + self.assertTrue(isinstance(label_value, urltree.LabelValue)) + self.assertEqual(label_value.label, label) + self.assertEqual(label_value.value, value) + self.assertEqual(label_value.is_default, is_default) + + def test_match_default_none (self) : + self.assertEquals(self.label.match(), None) + + def test_match_default_value (self) : + self._check_value(self.label_default_0, self.label_default_0.match(), 0, True) + self._check_value(self.label_default_1, self.label_default_1.match(), 1, True) + + def test_match_invalid (self) : + self.assertEquals(self.label.match("foo"), False) + self.assertEquals(self.label_default_0.match("foo"), False) + self.assertEquals(self.label_default_1.match("foo"), False) + + def test_match_valid (self) : + self._check_value(self.label, self.label.match("0"), 0, False) + self._check_value(self.label, self.label.match("1"), 1, False) + self._check_value(self.label_default_0, self.label_default_0.match("1"), 1, False) + self._check_value(self.label_default_1, self.label_default_1.match("0"), 0, False) + + def test_build (self) : + self.assertEqual(self.label.build(dict(test=1)), "1") + + def test_build_nodefault (self) : + self.assertRaises(urltree.URLError, self.label.build, {}) + + def test_build_none (self) : + self.assertRaises(urltree.URLError, self.label.build, dict(test=None)) + + def test_build_default (self) : + self.assertEqual(self.label_default_0.build_default({}), (True, "0")) + self.assertEqual(self.label_default_1.build_default({}), (True, "1")) + self.assertEqual(self.label_default_0.build_default(dict(test=0)), (True, "0")) + + def test_build_nonedefault (self) : + self.assertEqual(self.label_default_1.build_default(dict(test=None)), (True, "1")) + + def test_build_value (self) : + self.assertEqual(self.label.build_default(dict(test=0)), (False, "0")) + self.assertEqual(self.label.build_default(dict(test=1)), (False, "1")) + self.assertEqual(self.label_default_0.build_default(dict(test=1)), (False, "1")) + + def test_str (self) : + self.assertEqual(str(self.label), "{test:int}") + self.assertEqual(str(self.label_default_0), "{test:int=0}") + self.assertEqual(str(self.label_default_1), "{test:int=1}") + self.assertEqual(str(self.label_str), "{test}") + self.assertEqual(str(self.label_str_default), "{test=def}") + +class TestStringType (unittest.TestCase) : + def setUp (self) : + self.type = urltree.URLStringType() + + def test_test (self) : + self.assertTrue(self.type.test("")) + self.assertTrue(self.type.test("xxx")) + + def test_parse (self) : + self.assertEqual(self.type.parse(""), "") + self.assertEqual(self.type.parse("xxx"), "xxx") + + def test_build (self) : + self.assertEqual(self.type.build(""), "") + self.assertEqual(self.type.build("xxx"), "xxx") + self.assertEqual(self.type.build("äää"), "äää") + +class TestIntegerType (unittest.TestCase) : + def setUp (self) : + self.type = urltree.URLIntegerType() + self.type_positive = urltree.URLIntegerType(allow_negative=False) + self.type_nonzero = urltree.URLIntegerType(allow_zero=False) + self.type_max_5 = urltree.URLIntegerType(max=5) + + def test_test (self) : + self.assertTrue(self.type.test("1")) + self.assertFalse(self.type.test("xx")) + self.assertTrue(self.type_positive.test("1")) + self.assertFalse(self.type_positive.test("-1")) + self.assertTrue(self.type_nonzero.test("1")) + self.assertFalse(self.type_nonzero.test("0")) + self.assertTrue(self.type_max_5.test("5")) + self.assertFalse(self.type_max_5.test("6")) + + def test_parse_invalid (self) : + self.assertRaises(ValueError, self.type.parse, "xx") + self.assertRaises(ValueError, self.type_nonzero.parse, "0") + + def test_parse_valid (self) : + self.assertEqual(self.type.parse("0"), 0) + self.assertEqual(self.type.parse("2"), 2) + self.assertEqual(self.type_nonzero.parse("3"), 3) + + def test_append (self) : + self.assertRaises(urltree.URLError, self.type.append, 0, 1) + + def test_build (self) : + self.assertEqual(self.type.build(0), "0") + self.assertEqual(self.type.build(5), "5") + self.assertEqual(self.type_positive.build(1), "1") + self.assertEqual(self.type_nonzero.build(1), "1") + self.assertEqual(self.type_max_5.build(5), "5") + + def test_build_invalid (self) : + self.assertRaises(ValueError, self.type_positive.build, -1) + self.assertRaises(ValueError, self.type_nonzero.build, 0) + self.assertRaises(ValueError, self.type_max_5.build, 6) + + def test_build_multi (self) : + self.assertEqual(self.type.build_multi(0), ["0"]) + +class TestListType (unittest.TestCase) : + def setUp (self) : + self.type = urltree.URLListType() + + def test_parse (self) : + self.assertEqual(self.type.parse("x"), ["x"]) + self.assertEqual(self.type.parse(""), [""]) + + def test_append (self) : + self.assertEqual(self.type.append(["x"], ["y"]), ["x", "y"]) + + def test_build_multi (self) : + self.assertEqual(self.type.build_multi(["x", "y"]), ["x", "y"]) + +class TestConfig (unittest.TestCase) : + def test_init (self) : + urltree.URLConfig(type_dict=dict(foo=None), ignore_extra_args=True) + + def test_get_type (self) : + self.assertEquals(urltree.URLConfig(dict(foo='xxx')).get_type('foo'), 'xxx') + + def test_get_type_invalid (self) : + self.assertRaises(KeyError, urltree.URLConfig(dict(foo='xxx')).get_type, 'xxx') + + def test_call (self) : + config = urltree.URLConfig() + url = config("foo", None) + + self.assertTrue(isinstance(url, urltree.URL)) + self.assertTrue(url in config.urls) + + def test_iter (self) : + config = urltree.URLConfig() + url1 = config("foo1", None) + url2 = config("foo2", None) + + urls = list(config) + + self.assertTrue(urls[0].url_mask == "foo1") + self.assertTrue(urls[1].url_mask == "foo2") + +class TestURL (unittest.TestCase) : + def setUp (self) : + self.config = urltree.URLConfig(ignore_extra_args=True) + self.config_strict = urltree.URLConfig(ignore_extra_args=False) + + def _test_label_path (self, mask, *path, **qargs) : + url = self.config(mask, None) + + # right label path + self.assertEquals(url.label_path, list(path)) + + # right qargs keys + self.assertEquals(set(url.query_args), set(qargs)) + + # right qargs values + for key, value in qargs.iteritems() : + self.assertEquals(url.query_args[key], value) + + # __init__ + def test_label_path_empty (self) : + self._test_label_path("", urltree.EmptyLabel()) + + def test_label_path_root (self) : + self._test_label_path("/", urltree.EmptyLabel()) + + def test_label_path_static (self) : + self._test_label_path("/foo", urltree.StaticLabel("foo")) + + def test_label_path_static2 (self) : + self._test_label_path("/foo/bar/", urltree.StaticLabel("foo"), urltree.StaticLabel("bar"), urltree.EmptyLabel()) + + def test_label_path_mix (self) : + self._test_label_path("/foo/{bar}", urltree.StaticLabel("foo"), urltree.SimpleValueLabel("bar", None, None, None)) + +# def test_query_args_root_empty (self) : +# self._test_label_path("/?", urltree.EmptyLabel()) + + def test_query_args_simple (self) : + self._test_label_path("/x/?foo", urltree.StaticLabel("x"), foo=(self.config.get_type(None), None)) + + def test_query_args_multi (self) : + self._test_label_path("/x/?foo=0&bar&tee:int=&eee:int", urltree.StaticLabel("x"), + foo = (self.config.get_type(None), "0"), + bar = (self.config.get_type(None), None), + tee = (self.config.get_type('int'), ''), + eee = (self.config.get_type('int'), None), + ) + + def test_label_path_mutate (self) : + l = self.config("xxx", None) + + lp = l.get_label_path() + + lp.pop(0) + + self.assertTrue(len(l.label_path) > len(lp)) + + def _setup_handler (self) : + def _handler (req, **kwargs) : + return kwargs + + return _handler + + def _setup_execute (self, mask, config) : + _handler = self._setup_handler() + + url = config(mask, _handler) + + return url + + class dummyrequest : + def __init__ (self, qargs) : self.qargs = qargs + def get_args (self) : return self.qargs + + class dummy_label : + def __init__ (self, key, type) : + self.key = key + self.type = type + + class dummy_labelvalue : + def __init__ (self, key, value, is_default, type) : + self.label = TestURL.dummy_label(key, type) + self.value = value + self.is_default = is_default + + def _test_execute (self, mask, values={}, qargs={}, qlist=[], config=None) : + if not config : + config = self.config + + # setup + url = self._setup_execute(mask, config) + req = self.dummyrequest(qargs.items() + qlist) + values = [self.dummy_labelvalue(k, v, d, config.get_type()) for k, (v, d) in values.iteritems()] + + # exec + out_args = url.execute(req, values) + + return out_args + + # execute + def test_execute_empty (self) : + self.assertEquals(set(self._test_execute("/")), set()) + + def test_execute_plain (self) : + self.assertEquals(set(self._test_execute("/foo")), set()) + + def test_execute_simple (self) : + self.assertEquals(self._test_execute("/foo/{bar}", dict(bar=(0, False))), dict(bar=0)) + + def test_execute_multi (self) : + self.assertEquals(self._test_execute("/foo/{bar}/{quux}", dict(bar=(1, False), quux=(2, False))), dict(bar=1, quux=2)) + + def test_execute_default (self) : + self.assertEquals(self._test_execute("/foo/{bar=0}", dict(bar=("0", True))), dict(bar="0")) + + def test_execute_qargs_default (self) : + self.assertEquals(self._test_execute("/{foo}/?bar=0", dict(foo=("x", False))), dict(foo="x", bar="0")) + + def test_execute_qargs_default_type (self) : + self.assertEquals(self._test_execute("/{foo}/?bar:int=0", dict(foo=("x", False))), dict(foo="x", bar=0)) + + def test_execute_qargs_default_none (self) : + self.assertEquals(self._test_execute("/{foo}/?bar=", dict(foo=("x", False))), dict(foo="x")) + + def test_execute_qargs_missing (self) : + self.assertRaises(urltree.URLError, self._test_execute, "/{foo}/?bar", dict(foo=("x", False))) + + def test_execute_qargs_invalid (self) : + self.assertRaises(ValueError, self._test_execute, "/{foo}/?bar:int", dict(foo=("x", False)), dict(bar="x")) + + def test_execute_qargs_simple (self) : + self.assertEquals(self._test_execute("/{foo}/?bar", dict(foo=("x", False)), dict(bar="y")), dict(foo="x", bar="y")) + + def test_execute_qargs_novalue (self) : + self.assertRaises(urltree.URLError, self._test_execute, "/{foo}/?bar", dict(foo=("x", False)), dict(bar='')) + + def test_execute_qargs_multi_invalid (self) : + self.assertRaises(urltree.URLError, self._test_execute, "/{foo}/?bar", dict(foo=("x", False)), qlist=[('bar', 'a'), ('bar', 'b')]) + + def test_execute_qargs_multi_list (self) : + self.assertEqual(self._test_execute("/{foo}/?bar:list", dict(foo=("x", False)), qlist=[('bar', 'a'), ('bar', 'b')]), dict(foo='x', bar=['a', 'b'])) + + def test_execute_qarg_override_strict (self) : + self.assertRaises(urltree.URLError, self._test_execute, "/{foo}", dict(foo=("x1", False)), dict(foo="x2"), config=self.config_strict) + + def test_execute_qarg_override_ignore (self) : + self.assertEqual(self._test_execute("/{foo}", dict(foo=("x1", False)), dict(foo="x2")), dict(foo="x1")) + + def test_execute_qarg_override_ok (self) : + self.assertEqual(self._test_execute("/{foo=x1}", dict(foo=("x1", True)), dict(foo="x2")), dict(foo="x2")) + + # build + class dummyrequest_page : + def __init__ (self, page_prefix) : + self.page_prefix = page_prefix + + def _test_build (self, mask, url, **args) : + self.assertEquals(self.config(mask, None).build(self.dummyrequest_page("/index.cgi"), **args), "/index.cgi" + url) + + def _test_build_fails (self, err, mask, **args) : + self.assertRaises(err, self.config(mask, None).build, self.dummyrequest_page("/index.cgi"), **args) + + def test_build_empty (self) : + self._test_build("/", "/") + + def test_build_static (self) : + self._test_build("/foo", "/foo") + + def test_build_simple (self) : + self._test_build("/foo/{bar}", "/foo/x", bar="x") + + def test_build_multi (self) : + self._test_build("/foo/{bar}/{quux}", "/foo/x/y", bar="x", quux="y") + + def test_build_missing (self) : + self._test_build_fails(urltree.URLError, "/foo/{bar}/{quux}", bar="x") + + def test_build_unknown (self) : + self._test_build_fails(urltree.URLError, "/foo/{bar}/{quux}", bar="x", quux="y", frob="???") + + def test_build_long (self) : + self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x/y", bar="x", quux="y") + + def test_build_short (self) : + self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x", bar="x", quux="b") + + def test_build_with_none (self) : + self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x", bar="x", quux=None) + + def test_build_default (self) : + self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x", bar="x") + + def test_build_qargs (self) : + self._test_build("/foo/{bar}/?quux", "/foo/x?quux=a", bar="x", quux="a") + + def test_build_qargs_default (self) : + self._test_build("/foo/{bar}/?quux", "/foo/x?quux=a", bar="x", quux="a") + + # XXX: this just becomes ...?quux=['a', 'b'] like from str(list) +# def test_build_qargs_multi_invalid (self) : +# self._test_build_fails(urltree.URLError, "/foo/{bar}/?quux", bar="x", quux=["a", "b"]) + + def test_build_qargs_multi_list (self) : + self._test_build("/foo/{bar}/?quux:list", "/foo/x?quux=a&quux=b", bar="x", quux=["a", "b"]) + + def test_build_qargs_none (self) : + self._test_build("/foo/{bar}/?quux", "/foo/x", bar="x", quux=None) + +class TestTreeBuild (unittest.TestCase) : + def setUp (self) : + self.config = urltree.URLConfig(ignore_extra_args=True) + + def test_simple_root (self) : + self.config("/", None) + self.assertEqual(str(urltree.URLTree(self.config).root), "/[]") + + def test_simple_static (self) : + self.config("/foo/bar", None) + self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[bar/[]]]") + + def test_multi_static (self) : + self.config("/foo/bar", None) + self.config("/foo/quux", None) + self.config("/asdf", None) + self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[bar/[],quux/[]],asdf/[]]") + + def test_simple_value (self) : + self.config("/foo/{bar}", None) + self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[{bar}/[]]]") + + def test_deep (self) : + self.config("/foo/{cc}/a", None) + self.config("/foo/{cc}/b", None) + self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[{cc}/[a/[],b/[]]]]") + + def test_deep2 (self) : + self.config("/foo/{cc}/a/x", None) + self.config("/foo/{cc}/b", None) + self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[{cc}/[a/[x/[]],b/[]]]]") + + def test_ambig_simple (self) : + self.config("/foo", None) + self.config("/foo", None) + + self.assertRaises(urltree.URLError, urltree.URLTree, self.config) + +class TestTreeMatch (unittest.TestCase) : + def setUp (self) : + self.config = urltree.URLConfig(ignore_extra_args=True) + + self.root =self.config("/", None) + self.bar = self.config("/bar", None) + self.quux = self.config("/quux/{xyz}", None) + self.quux_boo = self.config("/quux/{xyz}/boo/{opt=no}", None) + self.quux_yes = self.config("/quux/{xyz}/yes", None) + + self.tree = urltree.URLTree(self.config) + + def _test_match (self, path, url, **values) : + t_url, t_values = self.tree.match(path) + + self.assertEqual(t_url, url) + + self.assertEqual(set(v.label.key for v in t_values), set(values)) + + for v in t_values : + self.assertEqual(v.value, values[v.label.key]) + + def test_root (self) : + self._test_match("", self.root) + + def test_bar (self) : + self._test_match("bar", self.bar) + + def test_bar_slash (self) : + self._test_match("bar/", self.bar) + + def test_quux (self) : + self._test_match("quux/a", self.quux, xyz="a") + + def test_quux_missing (self) : + self.assertRaises(urltree.URLError, self._test_match, "quux/", None) + + def test_quux_boo (self) : + self._test_match("quux/a/boo/x", self.quux_boo, xyz="a", opt="x") + + def test_quux_default (self) : + self._test_match("quux/a/boo", self.quux_boo, xyz="a", opt="no") + + def test_yes (self) : + self._test_match("quux/a/yes", self.quux_yes, xyz="a") + +class TestTreeHandler (unittest.TestCase) : + def _build_handler (self, name) : + def _handler (req, **args) : + return name, args + + return _handler + + def setUp (self) : + self.config = urltree.URLConfig(ignore_extra_args=True) + + self.root =self.config("/", self._build_handler('root')) + self.bar = self.config("/bar", self._build_handler('bar')) + self.quux = self.config("/quux/{xyz}", self._build_handler('quux')) + self.quux_boo = self.config("/quux/{xyz}/boo/{opt=no}", self._build_handler('quux_boo')) + + self.tree = urltree.URLTree(self.config) + + class dummyrequest_page : + def __init__ (self, page_name, qargs) : + self.page_name = page_name + self.qargs = qargs + + def get_page_name (self) : return self.page_name + def get_args (self) : return self.qargs + + def _test_handle (self, path, name, qargs={}, **args) : + req = self.dummyrequest_page(path, qargs.iteritems()) + + h_name, h_args = self.tree.handle_request(req) + + self.assertEqual(h_name, name) + self.assertEqual(h_args, args) + + def test_root (self) : + self._test_handle("", 'root') + + def test_bar (self) : + self._test_handle("bar", 'bar') + + def test_quux (self) : + self._test_handle("quux/a", 'quux', xyz='a') + + def test_quux_boo (self) : + self._test_handle("quux/a/boo/b", 'quux_boo', xyz='a', opt='b') + + def test_quux_boo_default (self) : + self._test_handle("quux/a/boo", 'quux_boo', xyz='a', opt='no') + + def test_quux_boo_qarg (self) : + self._test_handle("quux/a/boo", 'quux_boo', dict(opt='yes'), xyz='a', opt='yes') + +if __name__ == '__main__' : + unittest.main() +