def __init__(self, name, partitions): self.name = name self.partitions = [] self.cluster = None for i in range(partitions): self.add_partition(Partition(self, i))
def test_add_partition_to_topic(self): partition2 = Partition(self.topic, 1) self.topic.add_partition(partition2) assert len(self.topic.partitions) == 2 assert self.topic.partitions[1].topic == self.topic assert self.topic.partitions[1].num == 1 assert self.topic.partitions[1].replicas == [] assert self.topic.partitions[1].size == 0
def test_partition_inequality_on_partition_num(self): partition2 = Partition(self.topic, 1) assert self.topic.partitions[0] != partition2
def test_partition_equality_with_different_replicas(self): partition2 = Partition(self.topic, 0) broker = Broker(1, 'testhost1') partition2.replicas = [broker] assert self.topic.partitions[0] == partition2
def test_partition_equality(self): partition2 = Partition(self.topic, 0) assert self.topic.partitions[0] == partition2