2022-11-11 19:05:24 -07:00
// Copyright 2013-2022 The Cobra 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 cobra
import (
"bytes"
2022-11-28 14:39:12 -07:00
"fmt"
2022-11-11 19:05:24 -07:00
"log"
"os"
"testing"
)
func TestGenNushellCompletion ( t * testing . T ) {
2022-11-26 15:39:29 -07:00
rootCmd := & Command { Use : "kubectl" , Run : emptyRun }
2022-11-11 19:05:24 -07:00
rootCmd . PersistentFlags ( ) . String ( "server" , "s" , "The address and port of the Kubernetes API server" )
rootCmd . PersistentFlags ( ) . BoolP ( "skip-headers" , "" , false , "The address and port of the Kubernetes API serverIf true, avoid header prefixes in the log messages" )
getCmd := & Command {
Use : "get" ,
Short : "Display one or many resources" ,
ArgAliases : [ ] string { "pods" , "nodes" , "services" , "replicationcontrollers" , "po" , "no" , "svc" , "rc" } ,
ValidArgs : [ ] string { "pod" , "node" , "service" , "replicationcontroller" } ,
Run : emptyRun ,
}
rootCmd . AddCommand ( getCmd )
buf := new ( bytes . Buffer )
2022-11-28 14:39:12 -07:00
assertNoErr ( t , rootCmd . GenNushellCompletion ( buf , true ) )
2022-11-11 19:05:24 -07:00
output := buf . String ( )
2022-11-28 14:39:12 -07:00
check ( t , output , fmt . Sprintf ( "let cobra_apps = [\"%[1]s\"]" , rootCmd . Name ( ) ) )
2022-11-11 19:05:24 -07:00
}
func TestGenNushellCompletionFile ( t * testing . T ) {
2022-11-26 15:39:29 -07:00
err := os . Mkdir ( "./tmp" , 0 o755 )
2022-11-11 19:05:24 -07:00
if err != nil {
log . Fatal ( err . Error ( ) )
}
defer os . RemoveAll ( "./tmp" )
rootCmd := & Command { Use : "root" , Args : NoArgs , Run : emptyRun }
child := & Command {
Use : "child" ,
ValidArgsFunction : validArgsFunc ,
Run : emptyRun ,
}
rootCmd . AddCommand ( child )
2022-11-28 14:39:12 -07:00
assertNoErr ( t , rootCmd . GenNushellCompletionFile ( "./tmp/test" , true ) )
2022-11-11 19:05:24 -07:00
}
func TestFailGenNushellCompletionFile ( t * testing . T ) {
2022-11-26 15:39:29 -07:00
err := os . Mkdir ( "./tmp" , 0 o755 )
2022-11-11 19:05:24 -07:00
if err != nil {
log . Fatal ( err . Error ( ) )
}
defer os . RemoveAll ( "./tmp" )
2022-11-26 15:39:29 -07:00
f , _ := os . OpenFile ( "./tmp/test" , os . O_CREATE , 0 o400 )
2022-11-11 19:05:24 -07:00
defer f . Close ( )
rootCmd := & Command { Use : "root" , Args : NoArgs , Run : emptyRun }
child := & Command {
Use : "child" ,
ValidArgsFunction : validArgsFunc ,
Run : emptyRun ,
}
rootCmd . AddCommand ( child )
2022-11-28 14:39:12 -07:00
got := rootCmd . GenNushellCompletionFile ( "./tmp/test" , true )
2022-11-11 19:05:24 -07:00
if got == nil {
t . Error ( "should raise permission denied error" )
}
if os . Getenv ( "MSYSTEM" ) == "MINGW64" {
if got . Error ( ) != "open ./tmp/test: Access is denied." {
t . Errorf ( "got: %s, want: %s" , got . Error ( ) , "open ./tmp/test: Access is denied." )
}
} else {
if got . Error ( ) != "open ./tmp/test: permission denied" {
t . Errorf ( "got: %s, want: %s" , got . Error ( ) , "open ./tmp/test: permission denied" )
}
}
}