diff --git a/tests/mocks.py b/tests/mocks.py index f0ab965..f620da3 100644 --- a/tests/mocks.py +++ b/tests/mocks.py @@ -65,6 +65,7 @@ class MockClient(pydle.client.BasicClient): def _connect(self, hostname, port, *args, **kwargs): self.connection = MockConnection(hostname, port, mock_client=self, mock_server=self._mock_server, eventloop=self.eventloop) self.connection.connect() + self.on_connect() def raw(self, data): self.connection._mock_server.receivedata(data) diff --git a/tests/test__mocks.py b/tests/test__mocks.py index ed26858..be4462e 100644 --- a/tests/test__mocks.py +++ b/tests/test__mocks.py @@ -27,8 +27,12 @@ class Passed: @with_client(connected=False) def test_mock_client_connect(server, client): assert not client.connected + client.on_connect = Mock() client.connect('mock://local', 1337, eventloop=MockEventLoop()) + assert client.connected + assert client.on_connect.called + client.disconnect() assert not client.connected