def test_process_cluster(self): self.args.brokers = [1] action = ActionTrim(self.args, self.cluster) action.process_cluster() b2 = self.cluster.brokers[2] assert self.cluster.topics['testTopic1'].partitions[0].replicas == [b2] assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2] assert self.cluster.topics['testTopic2'].partitions[0].replicas == [b2] assert self.cluster.topics['testTopic2'].partitions[1].replicas == [b2]
def test_process_cluster(self): self.args.brokers = [1] action = ActionTrim(self.args, self.cluster) action.process_cluster() b2 = self.cluster.brokers[2] assert self.cluster.topics['testTopic1'].partitions[0].replicas == [b2] assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2] assert self.cluster.topics['testTopic2'].partitions[0].replicas == [b2] assert self.cluster.topics['testTopic2'].partitions[1].replicas == [b2]
def test_process_cluster_no_change(self): self.cluster.add_broker(Broker("brokerhost3.example.com", id=3)) self.args.brokers = [3] action = ActionTrim(self.args, self.cluster) action.process_cluster() b1 = self.cluster.brokers[1] b2 = self.cluster.brokers[2] assert self.cluster.topics['testTopic1'].partitions[0].replicas == [b1, b2] assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2, b1] assert self.cluster.topics['testTopic2'].partitions[0].replicas == [b2, b1] assert self.cluster.topics['testTopic2'].partitions[1].replicas == [b1, b2]
def test_process_cluster_no_change(self): self.cluster.add_broker(Broker("brokerhost3.example.com", id=3)) self.args.brokers = [3] action = ActionTrim(self.args, self.cluster) action.process_cluster() b1 = self.cluster.brokers[1] b2 = self.cluster.brokers[2] assert self.cluster.topics['testTopic1'].partitions[0].replicas == [ b1, b2 ] assert self.cluster.topics['testTopic1'].partitions[1].replicas == [ b2, b1 ] assert self.cluster.topics['testTopic2'].partitions[0].replicas == [ b2, b1 ] assert self.cluster.topics['testTopic2'].partitions[1].replicas == [ b1, b2 ]