entrouvert
/
debian-django-statsd-mozilla
17
0
Fork 0
Code Issues Pull Requests Releases Activity

Cleanup tests imports for patched methods

This commit is contained in:
Fabián Ezequiel Gallina 2014-04-30 16:21:18 -03:00
parent 73986a0fc8
commit 04aac83d83
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
django_statsd/tests.py
View File

@ -19,6 +19,11 @@ import mock
from nose.tools import eq_
from django_statsd.clients import get_client, statsd
from django_statsd.patches import utils
from django_statsd.patches.db import (
patched_callproc,
patched_execute,
patched_executemany,
)
from django_statsd import middleware
cfg = {
@ -471,7 +476,6 @@ class TestPatchMethod(TestCase):
class TestCursorWrapperPatching(TestCase):
def test_patched_callproc_calls_timer(self):
from django_statsd.patches.db import patched_callproc
with mock.patch.object(statsd, 'timer') as timer:
db = mock.Mock(executable_name='name', alias='alias')
instance = mock.Mock(db=db)
@ -479,7 +483,6 @@ class TestCursorWrapperPatching(TestCase):
self.assertEqual(timer.call_count, 1)
def test_patched_execute_calls_timer(self):
from django_statsd.patches.db import patched_execute
with mock.patch.object(statsd, 'timer') as timer:
db = mock.Mock(executable_name='name', alias='alias')
instance = mock.Mock(db=db)
@ -487,7 +490,6 @@ class TestCursorWrapperPatching(TestCase):
self.assertEqual(timer.call_count, 1)
def test_patched_executemany_calls_timer(self):
from django_statsd.patches.db import patched_executemany
with mock.patch.object(statsd, 'timer') as timer:
db = mock.Mock(executable_name='name', alias='alias')
instance = mock.Mock(db=db)