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)