tests: Fire on_connect() when connecting MockClient.

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)

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