def test_process_cluster(self): self.cluster.add_broker(Broker(3, "brokerhost3.example.com")) self.args.brokers = [1] self.args.to_brokers = [] action = ActionRemove(self.args, self.cluster) action.process_cluster() b2 = self.cluster.brokers[2] b3 = self.cluster.brokers[3] assert self.cluster.topics['testTopic1'].partitions[0].replicas == [b3, b2] assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2, b3] assert self.cluster.topics['testTopic2'].partitions[0].replicas == [b2, b3] assert self.cluster.topics['testTopic2'].partitions[1].replicas == [b3, b2]
def test_process_cluster(self): self.cluster.add_broker(Broker(3, "brokerhost3.example.com")) self.args.brokers = [1] self.args.to_brokers = [] action = ActionRemove(self.args, self.cluster) action.process_cluster() b2 = self.cluster.brokers[2] b3 = self.cluster.brokers[3] assert self.cluster.topics['testTopic1'].partitions[0].replicas == [ b3, b2 ] assert self.cluster.topics['testTopic1'].partitions[1].replicas == [ b2, b3 ] assert self.cluster.topics['testTopic2'].partitions[0].replicas == [ b2, b3 ] assert self.cluster.topics['testTopic2'].partitions[1].replicas == [ b3, b2 ]
def test_configure_args(self): ActionRemove.configure_args(self.subparsers) sys.argv = ['kafka-assigner', 'remove', '-b', '1', '-t', '2'] parsed_args = self.parser.parse_args() assert parsed_args.action == 'remove'
def test_process_cluster_error(self): self.args.brokers = [1] self.args.to_brokers = [] action = ActionRemove(self.args, self.cluster) self.assertRaises(NotEnoughReplicasException, action.process_cluster)
def test_configure_args(self): ActionRemove.configure_args(self.subparsers) sys.argv = ['kafka-assigner', 'remove', '-b', '1', '-t', '2'] parsed_args = self.parser.parse_args() assert parsed_args.action == 'remove'
def test_create_class_no_target(self): self.args.brokers = [1] self.args.to_brokers = [] action = ActionRemove(self.args, self.cluster) assert action.to_brokers == [2]
def test_create_class(self): self.args.brokers = [1] self.args.to_brokers = [2] action = ActionRemove(self.args, self.cluster) assert isinstance(action, ActionRemove)