def test_partition_offset_fetch_request(self): topic = mock.Mock() topic.name = "test_topic" partition = mock.Mock() partition.topic = topic partition.id = 12345 op = OwnedPartition(partition) request = op.build_offset_fetch_request() self.assertEqual(request.topic_name, topic.name) self.assertEqual(request.partition_id, partition.id)