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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
# -*- coding: utf-8 -*-
from unittest2 import TestCase
from raven.base import Client
# Some internal stuff to extend the transport layer
from raven.transport import Transport
import datetime
import calendar
import pytz
class DummyScheme(Transport):
scheme = ['mock']
def __init__(self, parsed_url):
self.check_scheme(parsed_url)
self._parsed_url = parsed_url
def send(self, data, headers):
"""
Sends a request to a remote webserver
"""
self._data = data
self._headers = headers
def compute_scope(self, url, scope):
netloc = url.hostname
netloc += ':%s' % url.port
path_bits = url.path.rsplit('/', 1)
if len(path_bits) > 1:
path = path_bits[0]
else:
path = ''
project = path_bits[-1]
if not all([netloc, project, url.username, url.password]):
raise ValueError('Invalid Sentry DSN: %r' % url.geturl())
server = '%s://%s%s/api/store/' % (url.scheme, netloc, path)
# Note that these variables in the scope are not actually used
# for anything w.r.t the DummyTransport
scope.update({
'SENTRY_SERVERS': [server],
'SENTRY_PROJECT': project,
'SENTRY_PUBLIC_KEY': url.username,
'SENTRY_SECRET_KEY': url.password,
})
return scope
class TransportTest(TestCase):
def setUp(self):
try:
Client.register_scheme('mock', DummyScheme)
except:
pass
def test_custom_transport(self):
c = Client(dsn="mock://some_username:some_password@localhost:8143/1")
data = dict(a=42, b=55, c=range(50))
c.send(**data)
expected_message = c.encode(data)
self.assertIn('mock://localhost:8143/api/store/', Client._registry._transports)
mock_cls = Client._registry._transports['mock://localhost:8143/api/store/']
assert mock_cls._data == expected_message
def test_build_then_send(self):
c = Client(dsn="mock://some_username:some_password@localhost:8143/1",
name="test_server")
mydate = datetime.datetime(2012, 5, 4, tzinfo=pytz.utc)
d = calendar.timegm(mydate.timetuple())
msg = c.build_msg('raven.events.Message', message='foo', date=d)
expected = {
'project': '1',
'sentry.interfaces.Message': {'message': 'foo', 'params': ()},
'server_name': u'test_server',
'level': 40,
'checksum': 'acbd18db4cc2f85cedef654fccc4a4d8',
'site': None,
'tags': None,
'time_spent': None,
'timestamp': 1336089600,
'message': 'foo',
}
# The event_id is always overridden
del msg['event_id']
self.assertDictContainsSubset(expected, msg)
|
