summaryrefslogtreecommitdiff
path: root/tests/context/tests.py
blob: 9332655cd61ef02d685a8507f8c0f52d1d902048 (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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
import threading
from raven.utils.testutils import TestCase

from raven.base import Client
from raven.context import Context


class ContextTest(TestCase):
    def test_simple(self):
        context = Context()
        context.merge({'foo': 'bar'})
        context.merge({'biz': 'baz'})
        context.merge({'biz': 'boz'})
        assert context.get() == {
            'foo': 'bar',
            'biz': 'boz',
        }

    def test_tags(self):
        context = Context()
        context.merge({'tags': {'foo': 'bar'}})
        context.merge({'tags': {'biz': 'baz'}})
        assert context.get() == {
            'tags': {
                'foo': 'bar',
                'biz': 'baz',
            }
        }

    def test_extra(self):
        context = Context()
        context.merge({'extra': {'foo': 'bar'}})
        context.merge({'extra': {'biz': 'baz'}})
        assert context.get() == {
            'extra': {
                'foo': 'bar',
                'biz': 'baz',
            }
        }

    def test_thread_binding(self):
        client = Client()
        called = []

        class TestContext(Context):

            def activate(self):
                Context.activate(self)
                called.append('activate')

            def deactivate(self):
                called.append('deactivate')
                Context.deactivate(self)

        # The main thread activates the context but clear does not
        # deactivate.
        context = TestContext(client)
        context.clear()
        assert called == ['activate']

        # But another thread does.
        del called[:]

        def test_thread():
            # This activate is unnecessary as the first activate happens
            # automatically
            context.activate()
            context.clear()
        t = threading.Thread(target=test_thread)
        t.start()
        t.join()
        assert called == ['activate', 'activate', 'deactivate']