def test_should_not_close_connection_to_broadcaster_when_first_call(self, mock_reactor): mock_receiver = Mock(Receiver) mock_broadcaster = Mock() mock_receiver.broadcaster = mock_broadcaster Receiver._refresh_connection(mock_receiver, 123, first_call=True) self.assertEquals(None, mock_broadcaster.client.sendClose.call_args)
def test_should_queue_call_to_refresh_connection(self, mock_reactor): mock_receiver = Mock(Receiver) mock_receiver.broadcaster = Mock() Receiver._refresh_connection(mock_receiver, 123) self.assertEquals( call(123, mock_receiver._refresh_connection), mock_reactor.callLater.call_args)