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()