コード例 #1
0
ファイル: test_producer.py プロジェクト: benjolitz/aiokafka
    def test_produce_many_gzip(self):
        start_offset = yield from self.current_offset(self.topic, 0)

        message1 = create_gzip_message([
            ("Gzipped 1 %d" % i).encode('utf-8') for i in range(100)])
        message2 = create_gzip_message([
            ("Gzipped 2 %d" % i).encode('utf-8') for i in range(100)])

        yield from self.assert_produce_request([message1, message2],
                                               start_offset, 200)
コード例 #2
0
    def test_produce_many_gzip(self):
        start_offset = yield from self.current_offset(self.topic, 0)

        message1 = create_gzip_message([("Gzipped 1 %d" % i).encode('utf-8')
                                        for i in range(100)])
        message2 = create_gzip_message([("Gzipped 2 %d" % i).encode('utf-8')
                                        for i in range(100)])

        yield from self.assert_produce_request([message1, message2],
                                               start_offset, 200)
コード例 #3
0
ファイル: test_producer.py プロジェクト: benjolitz/aiokafka
    def test_produce_100k_gzipped(self):
        start_offset = yield from self.current_offset(self.topic, 0)

        msgs1 = [("Gzipped batch 1, message %d" % i).encode('utf-8')
                 for i in range(50000)]
        yield from self.assert_produce_request([create_gzip_message(msgs1)],
                                               start_offset, 50000, )

        msgs2 = [("Gzipped batch 1, message %d" % i).encode('utf-8')
                 for i in range(50000)]

        yield from self.assert_produce_request([create_gzip_message(msgs2)],
                                               start_offset + 50000, 50000, )
コード例 #4
0
 def send_gzip_message(self, partition, messages):
     message = create_gzip_message([(self.msg(str(msg)), None)
                                    for msg in messages])
     produce = ProduceRequestPayload(self.topic,
                                     partition,
                                     messages=[message])
     resp, = self.client.send_produce_request([produce])
     self.assertEqual(resp.error, 0)
コード例 #5
0
    def test_produce_100k_gzipped(self):
        start_offset = yield from self.current_offset(self.topic, 0)

        msgs1 = [("Gzipped batch 1, message %d" % i).encode('utf-8')
                 for i in range(50000)]
        yield from self.assert_produce_request(
            [create_gzip_message(msgs1)],
            start_offset,
            50000,
        )

        msgs2 = [("Gzipped batch 1, message %d" % i).encode('utf-8')
                 for i in range(50000)]

        yield from self.assert_produce_request(
            [create_gzip_message(msgs2)],
            start_offset + 50000,
            50000,
        )
コード例 #6
0
    def test_produce_100k_gzipped(self):
        start_offset = self.current_offset(self.topic, 0)

        self.assert_produce_request([
            create_gzip_message([
                (("Gzipped batch 1, message %d" % i).encode('utf-8'), None)
                for i in range(50000)])
            ],
            start_offset,
            50000,
        )

        self.assert_produce_request([
            create_gzip_message([
                (("Gzipped batch 1, message %d" % i).encode('utf-8'), None)
                for i in range(50000)])
            ],
            start_offset+50000,
            50000,
        )
コード例 #7
0
    def test_produce_100k_gzipped(self):
        start_offset = self.current_offset(self.topic, 0)

        self.assert_produce_request([
            create_gzip_message([
                (("Gzipped batch 1, message %d" % i).encode('utf-8'), None)
                for i in range(50000)])
            ],
            start_offset,
            50000,
        )

        self.assert_produce_request([
            create_gzip_message([
                (("Gzipped batch 1, message %d" % i).encode('utf-8'), None)
                for i in range(50000)])
            ],
            start_offset+50000,
            50000,
        )
コード例 #8
0
ファイル: test_producer.py プロジェクト: benjolitz/aiokafka
    def test_produce_mixed(self):
        start_offset = yield from self.current_offset(self.topic, 0)

        msg_count = 1 + 100 + 100
        messages = [
            create_message(b"Just a plain message"),
            create_gzip_message([
                ("Gzipped %d" % i).encode('utf-8') for i in range(100)]),
            create_snappy_message([
                b"Snappy " + bytes(i) for i in range(100)])]

        yield from self.assert_produce_request(messages, start_offset,
                                               msg_count)
コード例 #9
0
    def test_produce_mixed(self):
        start_offset = yield from self.current_offset(self.topic, 0)

        msg_count = 1 + 100 + 100
        messages = [
            create_message(b"Just a plain message"),
            create_gzip_message([("Gzipped %d" % i).encode('utf-8')
                                 for i in range(100)]),
            create_snappy_message([b"Snappy " + bytes(i) for i in range(100)])
        ]

        yield from self.assert_produce_request(messages, start_offset,
                                               msg_count)
コード例 #10
0
    def test_produce_mixed(self):
        start_offset = self.current_offset(self.topic, 0)

        msg_count = 1+100
        messages = [
            create_message(b"Just a plain message"),
            create_gzip_message([
                (("Gzipped %d" % i).encode('utf-8'), None) for i in range(100)]),
        ]

        # All snappy integration tests fail with nosnappyjava
        if False and has_snappy():
            msg_count += 100
            messages.append(create_snappy_message([("Snappy %d" % i, None) for i in range(100)]))

        self.assert_produce_request(messages, start_offset, msg_count)
コード例 #11
0
    def test_produce_mixed(self):
        start_offset = self.current_offset(self.topic, 0)

        msg_count = 1+100
        messages = [
            create_message(b"Just a plain message"),
            create_gzip_message([
                (("Gzipped %d" % i).encode('utf-8'), None) for i in range(100)]),
        ]

        # All snappy integration tests fail with nosnappyjava
        if False and has_snappy():
            msg_count += 100
            messages.append(create_snappy_message([("Snappy %d" % i, None) for i in range(100)]))

        self.assert_produce_request(messages, start_offset, msg_count)
コード例 #12
0
 def send_gzip_message(self, partition, messages):
     message = create_gzip_message([(self.msg(str(msg)), None) for msg in messages])
     produce = ProduceRequestPayload(self.topic, partition, messages = [message])
     resp, = self.client.send_produce_request([produce])
     self.assertEqual(resp.error, 0)