def test_count_preferred_replica(self, partition): p1 = partition b1 = Broker('test-broker', partitions=set([p1])) p1.add_replica(b1) p2 = Mock(spec=Partition, topic=sentinel.t1, replicas=[sentinel.b2]) b1.add_partition(p2) assert b1.count_preferred_replica() == 1
def test_add_partition(self, create_partition): p10 = create_partition('t1', 0) p20 = create_partition('t2', 0) broker = Broker('test-broker', partitions=set([p10])) broker.add_partition(p20) assert broker.partitions == set([p10, p20]) assert p20.replicas == [broker]