68 lines
1.7 KiB
Python
68 lines
1.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
from mock import Mock
|
|
from raven.utils.compat import TestCase
|
|
from raven.utils import six
|
|
|
|
from raven.utils.stacks import get_culprit, get_stack_info
|
|
|
|
|
|
class Context(object):
|
|
def __init__(self, dict):
|
|
self.dict = dict
|
|
|
|
__getitem__ = lambda s, *a: s.dict.__getitem__(*a)
|
|
__setitem__ = lambda s, *a: s.dict.__setitem__(*a)
|
|
iterkeys = lambda s, *a: six.iterkeys(s.dict, *a)
|
|
|
|
|
|
class GetCulpritTest(TestCase):
|
|
def test_empty_module(self):
|
|
culprit = get_culprit([{
|
|
'module': None,
|
|
'function': 'foo',
|
|
}])
|
|
assert culprit == '? in foo'
|
|
|
|
def test_empty_function(self):
|
|
culprit = get_culprit([{
|
|
'module': 'foo',
|
|
'function': None,
|
|
}])
|
|
assert culprit == 'foo in ?'
|
|
|
|
def test_no_module_or_function(self):
|
|
culprit = get_culprit([{}])
|
|
assert culprit is None
|
|
|
|
def test_all_params(self):
|
|
culprit = get_culprit([{
|
|
'module': 'package.name',
|
|
'function': 'foo',
|
|
}])
|
|
assert culprit == 'package.name in foo'
|
|
|
|
|
|
class GetStackInfoTest(TestCase):
|
|
def test_bad_locals_in_frame(self):
|
|
frame = Mock()
|
|
frame.f_locals = Context({
|
|
'foo': 'bar',
|
|
'biz': 'baz',
|
|
})
|
|
frame.f_lineno = 1
|
|
frame.f_globals = {}
|
|
frame.f_code.co_filename = __file__
|
|
frame.f_code.co_name = __name__
|
|
|
|
frames = [(frame, 1)]
|
|
results = get_stack_info(frames)
|
|
self.assertEquals(len(results), 1)
|
|
result = results[0]
|
|
assert 'vars' in result
|
|
assert result['vars'] == {
|
|
"u'foo'": "u'bar'",
|
|
"u'biz'": "u'baz'",
|
|
}
|