github.com/cloudfoundry/libcfbuildpack@v1.91.23/internal/replace_args.go (about) 1 /* 2 * Copyright 2018-2020 the original author or authors. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * https://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package internal 18 19 import ( 20 "os" 21 "testing" 22 ) 23 24 // ReplaceArgs replaces the current command line arguments (os.Args) with a new collection of values. Returns a 25 // function suitable for use with defer in order to reset the previous values 26 // 27 // defer ReplaceArgs(t, "alpha")() 28 func ReplaceArgs(t *testing.T, args ...string) func() { 29 t.Helper() 30 31 previous := os.Args 32 os.Args = args 33 34 return func() { os.Args = previous } 35 }