| /* |
| Copyright 2017 The Kubernetes Authors. |
| |
| Licensed under the Apache License, Version 2.0 (the "License"); |
| you may not use this file except in compliance with the License. |
| You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| package util |
| |
| import ( |
| "testing" |
| ) |
| |
| func TestValidateExactArgNumber(t *testing.T) { |
| var tests = []struct { |
| args, supportedArgs []string |
| expectedErr bool |
| }{ |
| { // one arg given and one arg expected |
| args: []string{"my-node-1234"}, |
| supportedArgs: []string{"node-name"}, |
| expectedErr: false, |
| }, |
| { // two args given and two args expected |
| args: []string{"my-node-1234", "foo"}, |
| supportedArgs: []string{"node-name", "second-toplevel-arg"}, |
| expectedErr: false, |
| }, |
| { // too few supplied args |
| args: []string{}, |
| supportedArgs: []string{"node-name"}, |
| expectedErr: true, |
| }, |
| { // too few non-empty args |
| args: []string{""}, |
| supportedArgs: []string{"node-name"}, |
| expectedErr: true, |
| }, |
| { // too many args |
| args: []string{"my-node-1234", "foo"}, |
| supportedArgs: []string{"node-name"}, |
| expectedErr: true, |
| }, |
| } |
| for _, rt := range tests { |
| actual := ValidateExactArgNumber(rt.args, rt.supportedArgs) |
| if (actual != nil) != rt.expectedErr { |
| t.Errorf( |
| "failed ValidateExactArgNumber:\n\texpected error: %t\n\t actual error: %t", |
| rt.expectedErr, |
| (actual != nil), |
| ) |
| } |
| } |
| } |