Skip to content

Commit 6ecd3d7

Browse files
authored
Refactoring. (#169)
1 parent 2f8ee9a commit 6ecd3d7

File tree

2 files changed

+12
-14
lines changed

2 files changed

+12
-14
lines changed

mocket/async_mocket.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,7 @@ async def wrapper(test, cls=Mocketizer, truesocket_recording_dir=None, *args, **
66
instance = args[0] if args else None
77
namespace = None
88
if truesocket_recording_dir:
9-
namespace = ".".join(
10-
(
11-
instance.__class__.__module__,
12-
instance.__class__.__name__,
13-
test.__name__,
14-
)
15-
)
9+
namespace = Mocketizer.get_namespace(test, instance)
1610
async with cls(
1711
instance,
1812
namespace=namespace,

mocket/mocket.py

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -598,6 +598,16 @@ def __init__(self, instance=None, namespace=None, truesocket_recording_dir=None)
598598
self.truesocket_recording_dir = truesocket_recording_dir
599599
self.namespace = namespace or text_type(id(self))
600600

601+
@staticmethod
602+
def get_namespace(test, instance):
603+
return ".".join(
604+
(
605+
instance.__class__.__module__,
606+
instance.__class__.__name__,
607+
test.__name__,
608+
)
609+
)
610+
601611
def enter(self):
602612
Mocket.enable(
603613
namespace=self.namespace,
@@ -635,13 +645,7 @@ def wrapper(test, cls=Mocketizer, truesocket_recording_dir=None, *args, **kwargs
635645
instance = args[0] if args else None
636646
namespace = None
637647
if truesocket_recording_dir:
638-
namespace = ".".join(
639-
(
640-
instance.__class__.__module__,
641-
instance.__class__.__name__,
642-
test.__name__,
643-
)
644-
)
648+
namespace = Mocketizer.get_namespace(test, instance)
645649
with cls(
646650
instance,
647651
namespace=namespace,

0 commit comments

Comments
 (0)