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