internal: Use the public module for mocking

(cherry picked from commit 0e38cc5508d0880013dff280b5119da4da4119e8)
This commit is contained in:
Stein Magnus Jodal 2019-03-19 23:48:14 +01:00
parent fb77f0c307
commit 5726fb9d70

View File

@ -236,7 +236,7 @@ class ServerTest(unittest.TestCase):
self.assertFalse( self.assertFalse(
network.Server.maximum_connections_exceeded(self.mock)) network.Server.maximum_connections_exceeded(self.mock))
@patch('pykka.registry.ActorRegistry.get_by_class') @patch('pykka.ActorRegistry.get_by_class')
def test_number_of_connections(self, get_by_class): def test_number_of_connections(self, get_by_class):
self.mock.protocol = sentinel.protocol self.mock.protocol = sentinel.protocol