Always import mock stuff from mode.utils.mocks

t/conftest.py

@@ -1,6 +1,6 @@
 import asyncio
-from unittest import mock
 import pytest
+from mode.utils.mocks import Mock
 
 sentinel = object()
 

t/docs/testing/test_figB_agent_to_agent.py

@@ -1,6 +1,6 @@
-from unittest.mock import Mock, patch
 import faust
 import pytest
+from mode.utils.mocks import Mock, patch
 
 app = faust.App('example-test-agent-call')
 

t/functional/serializers/test_registry.py

@@ -1,10 +1,10 @@
 import typing
-from unittest.mock import Mock
 
 import faust
 import pytest
 from faust.exceptions import KeyDecodeError, ValueDecodeError
 from faust.utils import json
+from mode.utils.mocks import Mock
 
 
 class Case(typing.NamedTuple):

t/functional/test_streams.py

@@ -1,11 +1,12 @@
 import asyncio
 from copy import copy
-from unittest.mock import Mock
-from mode.utils.aiter import aiter, anext
-from mode.utils.mocks import AsyncMock
+
+import pytest
 from faust.exceptions import ImproperlyConfigured
 from faust.streams import maybe_forward
-import pytest
+from mode.utils.aiter import aiter, anext
+from mode.utils.mocks import AsyncMock, Mock
+
 from .helpers import channel_empty, message, put
 
 

t/functional/web/conftest.py

@@ -1,7 +1,7 @@
-from unittest.mock import Mock
+import pytest
 from faust.exceptions import SameNode
 from faust.web.site import Website
-import pytest
+from mode.utils.mocks import Mock
 
 BASE_PORT = 6066
 

t/unit/agents/test_actor.py

@@ -1,8 +1,7 @@
-from unittest.mock import Mock
 import pytest
 from faust.agents.actor import Actor, AsyncIterableActor, AwaitableActor
 from faust.types import TP
-from mode.utils.mocks import AsyncMock
+from mode.utils.mocks import AsyncMock, Mock
 
 
 class test_Actor:

t/unit/agents/test_manager.py

@@ -1,7 +1,6 @@
-from unittest.mock import Mock
 import pytest
 from faust.types import TP
-from mode.utils.mocks import AsyncMock
+from mode.utils.mocks import AsyncMock, Mock
 
 
 class test_AgentManager:

t/unit/sensors/test_base.py

@@ -1,7 +1,7 @@
-from unittest.mock import Mock
 import pytest
 from faust.sensors import Sensor
 from faust.types import TP
+from mode.utils.mocks import Mock
 
 TP1 = TP('foo', 0)
 

t/unit/stores/test_base.py

@@ -1,8 +1,8 @@
-from unittest.mock import Mock
 import pytest
-from mode import label
 from faust.stores.base import SerializedStore, Store
 from faust.types import TP
+from mode import label
+from mode.utils.mocks import Mock
 
 
 class MyStore(Store):

t/unit/stores/test_memory.py

@@ -1,7 +1,7 @@
-from unittest.mock import Mock
 import pytest
 from faust.stores.memory import Store
 from faust.types import TP
+from mode.utils.mocks import Mock
 
 
 class test_Store:

t/unit/tables/test_table.py

@@ -1,7 +1,7 @@
-from unittest.mock import Mock, patch
 import faust
 import pytest
 from faust.tables.wrappers import WindowWrapper
+from mode.utils.mocks import Mock, patch
 
 
 class TableKey(faust.Record):

t/unit/tables/test_wrappers.py

@@ -1,11 +1,12 @@
 import operator
 from datetime import datetime
-from unittest.mock import Mock
+
 import faust
 import pytest
 from faust.events import Event
 from faust.exceptions import ImproperlyConfigured
 from faust.tables.wrappers import WindowSet
+from mode.utils.mocks import Mock
 
 DATETIME = datetime.utcnow()
 DATETIME_TS = DATETIME.timestamp()

t/unit/test_events.py

@@ -1,6 +1,5 @@
-from unittest.mock import Mock
 import pytest
-from mode.utils.mocks import AsyncMock
+from mode.utils.mocks import AsyncMock, Mock
 from faust import Event
 
 

t/unit/test_joins.py

@@ -1,7 +1,7 @@
-from unittest.mock import Mock
 import pytest
 from faust import Record
 from faust.joins import InnerJoin, Join, LeftJoin, OuterJoin, RightJoin
+from mode.utils.mocks import Mock
 
 
 class User(Record):

t/unit/test_worker.py

@@ -3,12 +3,12 @@
 import socket
 import warnings
 from pathlib import Path
-from unittest.mock import Mock, patch
+
 import pytest
-from mode.utils.imports import symbol_by_name
-from mode.utils.mocks import AsyncMock
 from faust.worker import WEBSITE_CLS, Worker
 from faust.utils import terminal
+from mode.utils.imports import symbol_by_name
+from mode.utils.mocks import AsyncMock, Mock, patch
 
 
 class CoroEq:

t/unit/utils/terminal/test_spinners.py

@@ -1,6 +1,6 @@
 import sys
-from unittest.mock import Mock, call, patch
 from faust.utils.terminal.spinners import Spinner, SpinnerHandler
+from mode.utils.mocks import Mock, call, patch
 
 
 def spinner(file=None, isatty=True):

t/unit/utils/terminal/test_tables.py

@@ -1,8 +1,9 @@
 from contextlib import ExitStack, contextmanager
-from unittest.mock import Mock, patch
+
 import pytest
 import terminaltables
 from faust.utils.terminal import tables
+from mode.utils.mocks import Mock, patch
 
 TABLE_DATA = [
     ('foo', 1.0, 3.33, 6.66, 9.99),

t/unit/utils/test_venusian.py

@@ -1,5 +1,5 @@
-from unittest.mock import Mock, patch
 from faust.utils import venusian
+from mode.utils import Mock, patch
 
 
 def test_attach():

t/unit/web/test_views.py

@@ -1,6 +1,5 @@
-from unittest.mock import Mock, call
 import pytest
-from mode.utils.mocks import AsyncMock
+from mode.utils.mocks import AsyncMock, Mock, call
 from faust.web.views import Site, View