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
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
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
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