def test_configure_args(self):
ActionSetRF.configure_args(self.subparsers)
sys.argv = [
'kafka-assigner', 'set-replication-factor', '-t', 'testTopic1',
'-r', '1'
]
parsed_args = self.parser.parse_args()
assert parsed_args.action == 'set-replication-factor'
def test_process_cluster_decrease(self):
self.args.topics = ['testTopic1']
self.args.replication_factor = 1
action = ActionSetRF(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]
assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2]
assert self.cluster.topics['testTopic2'].partitions[0].replicas == [b2, b1]
assert self.cluster.topics['testTopic2'].partitions[1].replicas == [b1, b2]
def test_process_cluster_increase(self):
self.cluster.add_broker(Broker(3, "brokerhost3.example.com"))
self.args.topics = ['testTopic1']
self.args.replication_factor = 3
action = ActionSetRF(self.args, self.cluster)
action.process_cluster()
b1 = self.cluster.brokers[1]
b2 = self.cluster.brokers[2]
b3 = self.cluster.brokers[3]
assert self.cluster.topics['testTopic1'].partitions[0].replicas == [b1, b2, b3]
assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2, b1, b3]
assert self.cluster.topics['testTopic2'].partitions[0].replicas == [b2, b1]
assert self.cluster.topics['testTopic2'].partitions[1].replicas == [b1, b2]
def test_process_cluster_decrease(self):
self.args.topics = ['testTopic1']
self.args.replication_factor = 1
action = ActionSetRF(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]
assert self.cluster.topics['testTopic1'].partitions[1].replicas == [b2]
assert self.cluster.topics['testTopic2'].partitions[0].replicas == [
b2, b1
]
assert self.cluster.topics['testTopic2'].partitions[1].replicas == [
b1, b2
]
def test_process_cluster_increase(self):
self.cluster.add_broker(Broker("brokerhost3.example.com", id=3))
self.args.topics = ['testTopic1']
self.args.replication_factor = 3
action = ActionSetRF(self.args, self.cluster)
action.process_cluster()
b1 = self.cluster.brokers[1]
b2 = self.cluster.brokers[2]
b3 = self.cluster.brokers[3]
assert self.cluster.topics['testTopic1'].partitions[0].replicas == [
b1, b2, b3
]
assert self.cluster.topics['testTopic1'].partitions[1].replicas == [
b2, b1, b3
]
assert self.cluster.topics['testTopic2'].partitions[0].replicas == [
b2, b1
]
assert self.cluster.topics['testTopic2'].partitions[1].replicas == [
b1, b2
]
def test_create_class(self):
self.args.topics = ['testTopic1']
self.args.replication_factor = 1
action = ActionSetRF(self.args, self.cluster)
assert isinstance(action, ActionSetRF)
def test_configure_args(self):
ActionSetRF.configure_args(self.subparsers)
sys.argv = ['kafka-assigner', 'set-replication-factor', '-t', 'testTopic1', '-r', '1']
parsed_args = self.parser.parse_args()
assert parsed_args.action == 'set-replication-factor'