def test_send_versioned_request_if_not_supported_by_server(self): self.cc.stream_features[...] = roster_xso.RosterVersioningFeature() response = roster_xso.Query() self.cc.send.return_value = response run_coroutine(self.cc.before_stream_established()) call, = self.cc.send.mock_calls _, call_args, call_kwargs = call iq_request, = call_args self.assertEqual("foobar", iq_request.payload.ver)
def test_process_none_response_to_versioned_request(self): self.cc.stream_features[...] = roster_xso.RosterVersioningFeature() self.cc.send.return_value = None cb = unittest.mock.Mock() cb.return_value = True self.s.on_initial_roster_received.connect(cb) run_coroutine(self.cc.before_stream_established()) call, = self.cc.send.mock_calls _, call_args, call_kwargs = call iq_request, = call_args self.assertEqual("foobar", iq_request.payload.ver) self.assertIn(self.user1, self.s.items) self.assertIn(self.user2, self.s.items) self.assertSequenceEqual([ unittest.mock.call(), ], cb.mock_calls)