class FeedbackClientFactoryTestCase(TestCase):
@patch(MODULE + 'ssl.PrivateCertificate.loadPEM', Mock())
@patch(MODULE + 'FeedbackClientFactory.ENDPOINTS', {'pub': ('foo', 'bar')})
def setUp(self):
self.factory = FeedbackClientFactory('pub', __file__)
@patch(MODULE + 'ReconnectingClientFactory.clientConnectionFailed')
def test_client_connection_failed(self, client_connection_failed_mock):
connector = Mock()
reason = Mock()
self.factory.clientConnectionFailed(connector, reason)
client_connection_failed_mock.assert_called_once_with(self.factory,
connector,
reason)
@patch(MODULE + 'ReconnectingClientFactory.clientConnectionLost')
def test_client_connection_lost(self, client_connection_lost_mock):
connector = Mock()
reason = Mock()
self.factory.clientConnectionLost(connector, reason)
client_connection_lost_mock.assert_called_once_with(self.factory,
connector,
reason)
def test_feedbacks_received(self):
feedbacks = Mock()
callback = Mock()
event = self.factory.EVENT_FEEDBACKS_RECEIVED
self.factory.listen(event, callback)
self.factory.feedbacksReceived(feedbacks)
callback.assert_called_once_with(event, self.factory, feedbacks)
def setUp(self):
self.factory = FeedbackClientFactory('pub', __file__)
