def test_is_open(self, _find): _impl = Mock() plugin = Mock() plugin.Connection.return_value = _impl _find.return_value = plugin url = TEST_URL connection = Connection(url) is_open = connection.is_open() _impl.is_open.assert_called_once_with() self.assertEqual(is_open, _impl.is_open.return_value)
def test_close(self, _find): _impl = Mock() plugin = Mock() plugin.Connection.return_value = _impl _find.return_value = plugin url = TEST_URL connection = Connection(url) # soft connection.close() _impl.close.assert_called_once_with()
def test_open(self, _find): _impl = Mock() plugin = Mock() plugin.Connection.return_value = _impl _find.return_value = plugin url = TEST_URL retry = 1234 connection = Connection(url, retry) connection.open() _impl.open.assert_called_once_with() self.assertEqual(_impl.retry, retry)
def test_init(self, _find): _impl = Mock() plugin = Mock() plugin.Connection.return_value = _impl _find.return_value = plugin url = TEST_URL retry = 1234 connection = Connection(url, retry) _find.assert_called_with(url) self.assertEqual(connection.url, url) self.assertEqual(connection.retry, retry) self.assertEqual(connection._impl, _impl)