def test_right_rackaware_order_works(self):
self.args.types = ['even', 'rackaware']
action = ActionBalance(self.args, self.cluster)
action.process_cluster()
def test_process_cluster(self, mock_process):
self.args.types = ['count']
action = ActionBalance(self.args, self.cluster)
action.process_cluster()
mock_process.assert_called_once_with()
def test_process_cluster(self, mock_process):
self.args.types = ['count']
action = ActionBalance(self.args, self.cluster)
action.process_cluster()
mock_process.assert_called_once_with()
def test_right_rackaware_order_works(self):
self.args.types = ['even', 'rackaware']
action = ActionBalance(self.args, self.cluster)
action.process_cluster()