99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from raven.utils.compat 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=list(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': 'test_server',
|
|
'level': 40,
|
|
'checksum': 'acbd18db4cc2f85cedef654fccc4a4d8',
|
|
'modules': {},
|
|
'tags': {},
|
|
'time_spent': None,
|
|
'timestamp': 1336089600,
|
|
'message': 'foo',
|
|
}
|
|
|
|
# The event_id is always overridden
|
|
del msg['event_id']
|
|
|
|
self.assertDictContainsSubset(expected, msg)
|