1

# :set encoding=utf8

     2

"""

     3

    Unit tests for svv.ext.urltree

     4

"""

     5

     6

import unittest

     7

     8

from svv.ext import urltree

     9

    10

if False :

    11

    class TestLabelValue (unittest.TestCase) :

    12

        class dummylabel :

    13

            key = 'foo'

    14

    15

        def test_str_default (self) :

    16

            self.assertEqual(str(urltree.LabelValue(self.dummylabel, 'bar', True)), "foo")

    17

    18

        def test_str_value (self) :

    19

            self.assertEqual(str(urltree.LabelValue(self.dummylabel, 'bar', False)), "foo='bar'")

    20

    21

class TestLabel (unittest.TestCase) :

    22

    def setUp (self) :

    23

        self.config = urltree.URLConfig()

    24

    25

    def _test_parse (self, mask, _type, defaults={}, **attrs) :

    26

        label = urltree.Label.parse(mask, defaults, self.config)

    27

    28

        self.assertTrue(isinstance(label, _type))

    29

    30

        for k, v in attrs.iteritems() :

    31

            self.assertEqual(getattr(label, k), v)

    32

    33

    def test_parse_empty (self) :

    34

        self._test_parse("", urltree.EmptyLabel)

    35

    36

    def test_parse_static (self) :

    37

        self._test_parse("foo", urltree.StaticLabel, name="foo")

    38

    39

    def test_parse_static_valid (self) :

    40

        self._test_parse("foo_bar", urltree.StaticLabel, name="foo_bar")

    41

        self._test_parse("foo2", urltree.StaticLabel, name="foo2")

    42

        self._test_parse("2foo.q", urltree.StaticLabel, name="2foo.q")

    43

    44

    def test_parse_static_invalid (self) :

    45

        self.assertRaises(urltree.URLError, urltree.Label.parse, "foo/bar", {}, self.config)

    46

    47

    def test_parse_value_plain (self) :

    48

        self._test_parse("{foo}", urltree.ValueLabel, key="foo", default=None, type_name=None, type=urltree.URLConfig.BUILTIN_TYPES[None])

    49

    50

    def test_parse_value_default1 (self) :

    51

        self._test_parse("{foo=bar1}", urltree.ValueLabel, key="foo", default="bar1", type_name=None)

    52

    53

    def test_parse_value_default2 (self) :

    54

        self._test_parse("{foo}", urltree.ValueLabel, dict(foo="bar2"), key="foo", default="bar2", type_name=None)

    55

    56

    def test_parse_value_default3 (self) :

    57

        self._test_parse("{foo=bar1}", urltree.ValueLabel, dict(foo="bar3"), key="foo", default="bar3", type_name=None)

    58

    59

    def test_parse_value_type_str (self) :

    60

        self._test_parse("{foo:str}", urltree.ValueLabel, key="foo", type_name="str", type=urltree.URLConfig.BUILTIN_TYPES['str'])

    61

    62

    def test_parse_value_type_int (self) :

    63

        self._test_parse("{foo:int}", urltree.ValueLabel, key="foo", type_name="int", type=urltree.URLConfig.BUILTIN_TYPES['int'])

    64

    65

    def test_parse_invalid_type (self) :

    66

        self.assertRaises(KeyError, self._test_parse, "{foo:bar}", None)

    67

    68

class TestEmptyLabel (unittest.TestCase) :

    69

    def setUp (self) :

    70

        self.label = urltree.EmptyLabel()

    71

    72

    def test_eq (self) :

    73

        self.assertTrue(self.label == urltree.EmptyLabel())

    74

        self.assertFalse(self.label == urltree.StaticLabel("foo"))

    75

    76

    def test_match (self) :

    77

        self.assertFalse(self.label.match())

    78

        self.assertFalse(self.label.match("foo"))

    79

        self.assertTrue(self.label.match(""))

    80

    81

    def test_build (self) :

    82

        self.assertEqual(self.label.build({}), "")

    83

    84

    def test_build_default (self) :

    85

        self.assertEqual(self.label.build_default({}), (False, ""))

    86

    87

    def test_str (self) :

    88

        self.assertEqual(str(self.label), "")

    89

    90

class TestStaticLabel (unittest.TestCase) :

    91

    def setUp (self) :

    92

        self.label = urltree.StaticLabel("test")

    93

    94

    def test_eq (self) :

    95

        self.assertTrue(self.label == urltree.StaticLabel("test"))

    96

        self.assertFalse(self.label == urltree.StaticLabel("Test"))

    97

        self.assertFalse(self.label == urltree.EmptyLabel())

    98

    99

    def test_match (self) :

   100

        self.assertFalse(self.label.match())

   101

        self.assertFalse(self.label.match("foo"))

   102

        self.assertTrue(self.label.match("test"))

   103

   104

    def test_build (self) :

   105

        self.assertEqual(self.label.build({}), "test")

   106

   107

    def test_build_default (self) :

   108

        self.assertEqual(self.label.build_default({}), (False, "test"))

   109

   110

    def test_str (self) :

   111

        self.assertEqual(str(self.label), "test")

   112

   113

class TestValueLabel (unittest.TestCase) :

   114

    def setUp (self) :

   115

        self.label = urltree.ValueLabel("test", urltree.URLConfig.BUILTIN_TYPES['int'], None, 'int')

   116

        self.label_default_0 = urltree.ValueLabel("test", urltree.URLConfig.BUILTIN_TYPES['int'], 0, 'int')

   117

        self.label_default_1 = urltree.ValueLabel("test", urltree.URLConfig.BUILTIN_TYPES['int'], 1, 'int')

   118

        self.label_str = urltree.ValueLabel("test", urltree.URLConfig.BUILTIN_TYPES[None], None, None)

   119

        self.label_str_default = urltree.ValueLabel("test", urltree.URLConfig.BUILTIN_TYPES[None], 'def', None)

   120

   121

    def test_eq (self) :

   122

        self.assertTrue(self.label == urltree.ValueLabel("test", None, 1, 'str'))

   123

        self.assertFalse(self.label == urltree.StaticLabel("Test"))

   124

        self.assertFalse(self.label == urltree.EmptyLabel())

   125

   126

    def _check_value (self, label, label_value, value, is_default) :

   127

        _key, _value, _type, _is_default = label_value

   128

   129

        self.assertEqual(_key, label.key)

   130

        self.assertEqual(_type,label.type)

   131

        self.assertEqual(_value, value)

   132

        self.assertEqual(_is_default, is_default)

   133

   134

    def test_match_default_none (self) :

   135

        self.assertEquals(self.label.match(), False)

   136

   137

    def test_match_default_value (self) :

   138

        self._check_value(self.label_default_0, self.label_default_0.match(), 0, True)

   139

        self._check_value(self.label_default_1, self.label_default_1.match(), 1, True)

   140

   141

    def test_match_invalid (self) :

   142

        self.assertEquals(self.label.match("foo"), False)

   143

        self.assertEquals(self.label_default_0.match("foo"), False)

   144

        self.assertEquals(self.label_default_1.match("foo"), False)

   145

   146

    def test_match_valid (self) :

   147

        self._check_value(self.label, self.label.match("0"), 0, False)

   148

        self._check_value(self.label, self.label.match("1"), 1, False)

   149

        self._check_value(self.label_default_0, self.label_default_0.match("1"), 1, False)

   150

        self._check_value(self.label_default_1, self.label_default_1.match("0"), 0, False)

   151

   152

    def test_build (self) :

   153

        self.assertEqual(self.label.build(dict(test=1)), "1")

   154

   155

    def test_build_nodefault (self) :

   156

        self.assertRaises(urltree.URLError, self.label.build, {})

   157

   158

    def test_build_none (self) :

   159

        self.assertRaises(urltree.URLError, self.label.build, dict(test=None))

   160

   161

    def test_build_default (self) :

   162

        self.assertEqual(self.label_default_0.build_default({}), (True, "0"))

   163

        self.assertEqual(self.label_default_1.build_default({}), (True, "1"))

   164

        self.assertEqual(self.label_default_0.build_default(dict(test=0)), (True, "0"))

   165

   166

    def test_build_nonedefault (self) :

   167

        self.assertEqual(self.label_default_1.build_default(dict(test=None)), (True, "1"))

   168

   169

    def test_build_value (self) :

   170

        self.assertEqual(self.label.build_default(dict(test=0)), (False, "0"))

   171

        self.assertEqual(self.label.build_default(dict(test=1)), (False, "1"))

   172

        self.assertEqual(self.label_default_0.build_default(dict(test=1)), (False, "1"))

   173

   174

    def test_str (self) :

   175

        self.assertEqual(str(self.label), "{test:int}")

   176

        self.assertEqual(str(self.label_default_0), "{test:int=0}")

   177

        self.assertEqual(str(self.label_default_1), "{test:int=1}")

   178

        self.assertEqual(str(self.label_str), "{test}")

   179

        self.assertEqual(str(self.label_str_default), "{test=def}")

   180

   181

class TestStringType (unittest.TestCase) :

   182

    def setUp (self) :

   183

        self.type = urltree.URLStringType()

   184

   185

    def test_test (self) :

   186

        self.assertTrue(self.type.test(""))

   187

        self.assertTrue(self.type.test("xxx"))

   188

   189

    def test_parse (self) :

   190

        self.assertEqual(self.type.parse(""), "")

   191

        self.assertEqual(self.type.parse("xxx"), "xxx")

   192

   193

    def test_build (self) :

   194

        self.assertEqual(self.type.build(""), "")

   195

        self.assertEqual(self.type.build("xxx"), "xxx")

   196

   197

        # XXX: unicode?

   198

        self.assertEqual(self.type.build(u"äää"), u"äää")

   199

   200

class TestIntegerType (unittest.TestCase) :

   201

    def setUp (self) :

   202

        self.type = urltree.URLIntegerType()

   203

        self.type_positive = urltree.URLIntegerType(allow_negative=False)

   204

        self.type_nonzero = urltree.URLIntegerType(allow_zero=False)

   205

        self.type_max_5 = urltree.URLIntegerType(max=5)

   206

   207

    def test_test (self) :

   208

        self.assertTrue(self.type.test("1"))

   209

        self.assertFalse(self.type.test("xx"))

   210

        self.assertTrue(self.type_positive.test("1"))

   211

        self.assertFalse(self.type_positive.test("-1"))

   212

        self.assertTrue(self.type_nonzero.test("1"))

   213

        self.assertFalse(self.type_nonzero.test("0"))

   214

        self.assertTrue(self.type_max_5.test("5"))

   215

        self.assertFalse(self.type_max_5.test("6"))

   216

   217

    def test_parse_invalid (self) :

   218

        self.assertRaises(urltree.URLValueError, self.type.parse, "xx")

   219

        self.assertRaises(urltree.URLValueError, self.type_nonzero.parse, "0")

   220

   221

    def test_parse_valid (self) :

   222

        self.assertEqual(self.type.parse("0"), 0)

   223

        self.assertEqual(self.type.parse("2"), 2)

   224

        self.assertEqual(self.type_nonzero.parse("3"), 3)

   225

   226

    def test_append (self) :

   227

        self.assertRaises(urltree.URLValueError, self.type.append, 0, 1)

   228

   229

    def test_build (self) :

   230

        self.assertEqual(self.type.build(0), "0")

   231

        self.assertEqual(self.type.build(5), "5")

   232

        self.assertEqual(self.type_positive.build(1), "1")

   233

        self.assertEqual(self.type_nonzero.build(1), "1")

   234

        self.assertEqual(self.type_max_5.build(5), "5")

   235

   236

    def test_build_invalid (self) :

   237

        self.assertRaises(urltree.URLValueError, self.type_positive.build, -1)

   238

        self.assertRaises(urltree.URLValueError, self.type_nonzero.build, 0)

   239

        self.assertRaises(urltree.URLValueError, self.type_max_5.build, 6)

   240

   241

    def test_build_multi (self) :

   242

        self.assertEqual(self.type.build_multi(0), ["0"])

   243

   244

class TestListType (unittest.TestCase) :

   245

    def setUp (self) :

   246

        self.type = urltree.URLListType()

   247

   248

    def test_parse (self) :

   249

        self.assertEqual(self.type.parse("x"), ["x"])

   250

        self.assertEqual(self.type.parse(""), [""])

   251

   252

    def test_append (self) :

   253

        self.assertEqual(self.type.append(["x"], ["y"]), ["x", "y"])

   254

   255

    def test_build_multi (self) :

   256

        self.assertEqual(self.type.build_multi(["x", "y"]), ["x", "y"])

   257

   258

class TestConfig (unittest.TestCase) :

   259

    def test_init (self) :

   260

        urltree.URLConfig(type_dict=dict(foo=None), reject_extra_args=False)

   261

   262

    def test_get_type (self) :

   263

        self.assertEquals(urltree.URLConfig(dict(foo='xxx')).get_type('foo'), 'xxx')

   264

   265

    def test_get_type_invalid (self) :

   266

        self.assertRaises(KeyError, urltree.URLConfig(dict(foo='xxx')).get_type, 'bar')

   267

   268

    def test_call (self) :

   269

        config = urltree.URLConfig()

   270

        url = config("foo", None)

   271

   272

        self.assertTrue(isinstance(url, urltree.URLRule))

   273

        self.assertTrue(url in config.urls)

   274

   275

    def test_iter (self) :

   276

        config = urltree.URLConfig()

   277

        url1 = config("foo1", None)

   278

        url2 = config("foo2", None)

   279

   280

        urls = list(config)

   281

   282

        self.assertTrue(urls[0].url_spec == "foo1")

   283

        self.assertTrue(urls[1].url_spec == "foo2")

   284

   285

class TestURL (unittest.TestCase) :

   286

    def setUp (self) :

   287

        self.config = urltree.URLConfig(reject_extra_args=False)

   288

#        self.config_strict = urltree.URLConfig(ignore_extra_args=False)

   289

   290

    def _test_label_path (self, mask, *path, **qargs) :

   291

        url = self.config(mask, None)

   292

   293

        # right label path

   294

        self.assertEquals(url.label_path, list(path))

   295

   296

        # right qargs keys

   297

        self.assertEquals(set(url.query_items), set(qargs))

   298

   299

        # right qargs values

   300

        for key, value in qargs.iteritems() :

   301

            self.assertEquals((url.query_items[key].type, url.query_items[key].default), value)

   302

   303

    # __init__

   304

    def test_label_path_empty (self) :

   305

        self._test_label_path("", urltree.EmptyLabel())

   306

   307

    def test_label_path_root (self) :

   308

        self._test_label_path("/", urltree.EmptyLabel())

   309

   310

    def test_label_path_static (self) :

   311

        self._test_label_path("/foo", urltree.StaticLabel("foo"))

   312

   313

    def test_label_path_static2 (self) :

   314

        self._test_label_path("/foo/bar/", urltree.StaticLabel("foo"), urltree.StaticLabel("bar"), urltree.EmptyLabel())

   315

   316

    def test_label_path_mix (self) :

   317

        self._test_label_path("/foo/{bar}", urltree.StaticLabel("foo"), urltree.ValueLabel("bar", None, None, None))

   318

   319

#    def test_query_args_root_empty (self) :

   320

#        self._test_label_path("/?", urltree.EmptyLabel())

   321

   322

    def test_query_args_simple (self) :

   323

        self._test_label_path("/x/?foo", urltree.StaticLabel("x"), foo=(self.config.get_type(None), None))

   324

   325

    def test_query_args_multi (self) :

   326

        self._test_label_path("/x/?foo=0&bar&tee:int=&eee:int", urltree.StaticLabel("x"), 

   327

            foo = (self.config.get_type(None), "0"),

   328

            bar = (self.config.get_type(None), None),

   329

            tee = (self.config.get_type('int'), ''),

   330

            eee = (self.config.get_type('int'), None),

   331

        )

   332

   333

    def test_label_path_mutate (self) :

   334

        l = self.config("xxx", None)

   335

   336

        lp = l.get_label_path()

   337

   338

        lp.pop(0)

   339

   340

        self.assertTrue(len(l.label_path) > len(lp))

   341

   342

    def _setup_handler (self) :

   343

        def _handler (req, **kwargs) :

   344

            return kwargs

   345

   346

        return _handler

   347

   348

    def _setup_execute (self, mask, config) :

   349

        _handler = self._setup_handler()

   350

   351

        url = config(mask, _handler)

   352

   353

        return url

   354

   355

    # XXX: not used?

   356

    class dummy_label :

   357

        def __init__ (self, key, type) :

   358

            self.key = key

   359

            self.type = type

   360

   361

    # XXX: rename to test_evaluate

   362

    def _test_execute (self, mask, values={}, qargs={}, qlist=[], config=None) :

   363

        if not config :

   364

            config = self.config

   365

   366

        # setup

   367

        url = self._setup_execute(mask, config)

   368

   369

        # qargs

   370

        uargs = [(key, value, config.get_type(), is_default) for key, (value, is_default) in values.iteritems()]

   371

        qargs = [(key, [value]) for key, value in qargs.iteritems()] + qlist

   372

   373

        # exec

   374

        out_args, remaining = url.evaluate(uargs, qargs)

   375

   376

        return out_args

   377

   378

    # execute

   379

    def test_execute_empty (self) :

   380

        self.assertEquals(set(self._test_execute("/")), set())

   381

   382

    def test_execute_plain (self) :

   383

        self.assertEquals(set(self._test_execute("/foo")), set())

   384

   385

    def test_execute_simple (self) :

   386

        self.assertEquals(self._test_execute("/foo/{bar}", dict(bar=(0, False))), dict(bar=0))

   387

   388

    def test_execute_multi (self) :

   389

        self.assertEquals(self._test_execute("/foo/{bar}/{quux}", dict(bar=(1, False), quux=(2, False))), dict(bar=1, quux=2))

   390

   391

    def test_execute_default (self) :

   392

        self.assertEquals(self._test_execute("/foo/{bar=0}", dict(bar=("0", True))), dict(bar="0"))

   393

   394

    def test_execute_qargs_default (self) :

   395

        self.assertEquals(self._test_execute("/{foo}/?bar=0", dict(foo=("x", False))), dict(foo="x", bar="0"))

   396

   397

    def test_execute_qargs_default_type (self) :

   398

        self.assertEquals(self._test_execute("/{foo}/?bar:int=0", dict(foo=("x", False))), dict(foo="x", bar=0))

   399

   400

    def test_execute_qargs_default_none (self) :

   401

        self.assertEquals(self._test_execute("/{foo}/?bar=", dict(foo=("x", False))), dict(foo="x"))

   402

   403

    def test_execute_qargs_missing (self) :

   404

        self.assertRaises(urltree.URLValueError, self._test_execute, "/{foo}/?bar", dict(foo=("x", False)))

   405

   406

    def test_execute_qargs_invalid (self) :

   407

        self.assertRaises(urltree.URLValueError, self._test_execute, "/{foo}/?bar:int", dict(foo=("x", False)), dict(bar="x"))

   408

   409

    def test_execute_qargs_simple (self) :

   410

        self.assertEquals(self._test_execute("/{foo}/?bar", dict(foo=("x", False)), dict(bar="y")), dict(foo="x", bar="y"))

   411

   412

    def test_execute_qargs_novalue (self) :

   413

        self.assertRaises(urltree.URLError, self._test_execute, "/{foo}/?bar", dict(foo=("x", False)), dict(bar=''))

   414

   415

    def test_execute_qargs_multi_invalid (self) :

   416

        self.assertRaises(urltree.URLError, self._test_execute, "/{foo}/?bar", dict(foo=("x", False)), qlist=[('bar', ['a', 'b'])])

   417

   418

    def test_execute_qargs_multi_list (self) :

   419

        self.assertEqual(self._test_execute("/{foo}/?bar:list", dict(foo=("x", False)), qlist=[('bar', ['a', 'b'])]), dict(foo='x', bar=['a', 'b']))

   420

   421

    # XXX: fix to check remaining qargs

   422

#    def test_execute_qarg_override_strict (self) :

   423

#        self.assertRaises(urltree.URLError, self._test_execute, "/{foo}", dict(foo=("x1", False)), dict(foo="x2"), config=self.config_strict)

   424

   425

    def test_execute_qarg_override_ignore (self) :

   426

        """Query-arg does not override non-default Label value"""

   427

        self.assertEqual(self._test_execute("/{foo}", dict(foo=("x1", False)), dict(foo="x2")), dict(foo="x1"))

   428

   429

    def test_execute_qarg_override_ok (self) :

   430

        """Query-arg overrides default Label value"""

   431

        self.assertEqual(self._test_execute("/{foo=x1}", dict(foo=("x1", True)), dict(foo="x2")), dict(foo="x2"))

   432

   433

    # build

   434

   435

    def _test_build (self, mask, url, **args) :

   436

        out, qargs = self.config(mask, None).build(**args)

   437

   438

        if qargs :

   439

            out += ('?' + '&'.join('%s=%s' % (key, value) for key, values in qargs for value in values))

   440

   441

        self.assertEquals(out, url)

   442

   443

    def _test_build_fails (self, err, mask, **args) :

   444

        self.assertRaises(err, self.config(mask, None).build, **args)

   445

   446

    def test_build_empty (self) :

   447

        self._test_build("/", "/")

   448

   449

    def test_build_static (self) :

   450

        self._test_build("/foo", "/foo")

   451

   452

    def test_build_simple (self) :

   453

        self._test_build("/foo/{bar}", "/foo/x", bar="x")

   454

   455

    def test_build_multi (self) :

   456

        self._test_build("/foo/{bar}/{quux}", "/foo/x/y", bar="x", quux="y")

   457

   458

    def test_build_missing (self) :

   459

        self._test_build_fails(urltree.URLError, "/foo/{bar}/{quux}", bar="x")

   460

   461

    def test_build_unknown (self) :

   462

        self._test_build_fails(urltree.URLError, "/foo/{bar}/{quux}", bar="x", quux="y", frob="???")

   463

   464

    def test_build_long (self) :

   465

        self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x/y", bar="x", quux="y")

   466

   467

    def test_build_short (self) :

   468

        self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x", bar="x", quux="b")

   469

   470

    def test_build_with_none (self) :

   471

        self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x", bar="x", quux=None)

   472

   473

    def test_build_default (self) :

   474

        self._test_build("/foo/{bar=a}/{quux=b}", "/foo/x", bar="x")

   475

   476

    def test_build_qargs (self) :

   477

        self._test_build("/foo/{bar}/?quux", "/foo/x?quux=a", bar="x", quux="a")

   478

   479

    def test_build_qargs_default (self) :

   480

        self._test_build("/foo/{bar}/?quux", "/foo/x?quux=a", bar="x", quux="a")

   481

   482

    # XXX: this just becomes ...?quux=['a', 'b'] like from str(list)

   483

#    def test_build_qargs_multi_invalid (self) :

   484

#        self._test_build_fails(urltree.URLError, "/foo/{bar}/?quux", bar="x", quux=["a", "b"])

   485

   486

    def test_build_qargs_multi_list (self) :

   487

        self._test_build("/foo/{bar}/?quux:list", "/foo/x?quux=a&quux=b", bar="x", quux=["a", "b"])

   488

   489

    def test_build_qargs_none (self) :

   490

        self._test_build("/foo/{bar}/?quux", "/foo/x", bar="x", quux=None)

   491

   492

class TestTreeBuild (unittest.TestCase) :

   493

    def setUp (self) :

   494

        self.config = urltree.URLConfig(reject_extra_args=False)

   495

   496

    def test_simple_root (self) :

   497

        self.config("/", None)

   498

        self.assertEqual(str(urltree.URLTree(self.config).root), "/[]")

   499

   500

    def test_simple_static (self) :

   501

        self.config("/foo/bar", None)

   502

        self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[bar/[]]]")

   503

   504

    def test_multi_static (self) :

   505

        self.config("/foo/bar", None)

   506

        self.config("/foo/quux", None)

   507

        self.config("/asdf", None)

   508

        self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[bar/[],quux/[]],asdf/[]]")

   509

   510

    def test_simple_value (self) :

   511

        self.config("/foo/{bar}", None)

   512

        self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[{bar}/[]]]")

   513

   514

    def test_deep (self) :

   515

        self.config("/foo/{cc}/a", None)

   516

        self.config("/foo/{cc}/b", None)

   517

        self.assertEqual(str(urltree.URLTree(self.config).root), "/[foo/[{cc}/[a/[],b/[]]]]")

   518

   519

    def test_deep2 (self) :

   520