summaryrefslogtreecommitdiff
path: root/tests/test_util.py
blob: 3f71f2b8e3d88032714c40e0585d95f49933d172 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
import unittest
import warnings

from webob.response import Response
from webob.util import warn_deprecation


class Test_warn_deprecation(unittest.TestCase):
    def setUp(self):

        self.oldwarn = warnings.warn
        warnings.warn = self._warn
        self.warnings = []

    def tearDown(self):

        warnings.warn = self.oldwarn
        del self.warnings

    def _callFUT(self, text, version, stacklevel):

        return warn_deprecation(text, version, stacklevel)

    def _warn(self, text, type, stacklevel=1):
        self.warnings.append(locals())

    def test_multidict_update_warning(self):
        # test warning when duplicate keys are passed
        r = Response()
        r.headers.update([("Set-Cookie", "a=b"), ("Set-Cookie", "x=y")])
        self.assertEqual(len(self.warnings), 1)
        deprecation_warning = self.warnings[0]
        self.assertEqual(deprecation_warning["type"], UserWarning)
        assert "Consider using .extend()" in deprecation_warning["text"]

    def test_multidict_update_warning_unnecessary(self):
        # no warning on normal operation
        r = Response()
        r.headers.update([("Set-Cookie", "a=b")])
        self.assertEqual(len(self.warnings), 0)

    def test_warn_deprecation(self):
        v = "1.3.0"

        self.assertRaises(DeprecationWarning, warn_deprecation, "foo", v[:3], 1)

    def test_warn_deprecation_future_version(self):
        v = "9.9.9"

        warn_deprecation("foo", v[:3], 1)
        self.assertEqual(len(self.warnings), 1)