コード例 #1
0
ファイル: broker_test.py プロジェクト: me2d/kafka-utils
    def test_count_partition(self):
        t1 = sentinel.t1
        partitions = set([
            Mock(spec=Partition, topic=t1, replicas=sentinel.r1),
            Mock(spec=Partition, topic=t1, replicas=sentinel.r1),
            Mock(spec=Partition, topic=sentinel.t2, replicas=sentinel.r1),
        ])
        broker = Broker('test-broker', partitions=partitions)

        assert broker.count_partitions(t1) == 2
        assert broker.count_partitions(sentinel.t3) == 0
コード例 #2
0
ファイル: broker_test.py プロジェクト: Veterun/kafka-utils
    def test_count_partition(self):
        t1 = sentinel.t1
        partitions = set([
            Mock(spec=Partition, topic=t1, replicas=sentinel.r1),
            Mock(spec=Partition, topic=t1, replicas=sentinel.r1),
            Mock(spec=Partition, topic=sentinel.t2, replicas=sentinel.r1),
        ])
        broker = Broker('test-broker', partitions=partitions)

        assert broker.count_partitions(t1) == 2
        assert broker.count_partitions(sentinel.t3) == 0
コード例 #3
0
ファイル: broker_test.py プロジェクト: mikewiebe/kafka-utils
    def test_count_partition(self, create_partition):
        p10 = create_partition('t1', 0)
        p11 = create_partition('t1', 1)
        p20 = create_partition('t2', 0)
        p30 = create_partition('t3', 0)
        t1 = p10.topic
        t3 = p30.topic
        broker = Broker('test-broker', partitions=set([p10, p11, p20]))

        assert broker.count_partitions(t1) == 2
        assert broker.count_partitions(t3) == 0
コード例 #4
0
ファイル: broker_test.py プロジェクト: Yelp/kafka-utils
    def test_count_partition(self, create_partition):
        p10 = create_partition('t1', 0)
        p11 = create_partition('t1', 1)
        p20 = create_partition('t2', 0)
        p30 = create_partition('t3', 0)
        t1 = p10.topic
        t3 = p30.topic
        broker = Broker('test-broker', partitions=set([p10, p11, p20]))

        assert broker.count_partitions(t1) == 2
        assert broker.count_partitions(t3) == 0