istio

1
// Copyright Istio Authors
2
//
3
// Licensed under the Apache License, Version 2.0 (the "License");
4
// you may not use this file except in compliance with the License.
5
// You may obtain a copy of the License at
6
//
7
//     http://www.apache.org/licenses/LICENSE-2.0
8
//
9
// Unless required by applicable law or agreed to in writing, software
10
// distributed under the License is distributed on an "AS IS" BASIS,
11
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
// See the License for the specific language governing permissions and
13
// limitations under the License.
14

15
package flag
16

17
import (
18
	"testing"
19

20
	"github.com/spf13/pflag"
21

22
	"istio.io/istio/pkg/test"
23
	"istio.io/istio/pkg/test/util/assert"
24
)
25

26
func TestBind(t *testing.T) {
27
	type Options struct {
28
		A, B, C string
29
		Env     string
30
		Bool    bool
31
	}
32
	opts := Options{
33
		B: "b-def",
34
	}
35
	test.SetEnvForTest(t, "TEST_ENV", "from-env")
36
	fs := pflag.NewFlagSet("test", pflag.ContinueOnError)
37
	Bind(fs, "a", "", "desc", &opts.A)
38
	Bind(fs, "bool", "", "desc", &opts.Bool)
39
	BindEnv(fs, "test-env", "", "desc", &opts.Env)
40
	fs.Set("a", "a-set")
41
	fs.Set("bool", "true")
42
	assert.Equal(t, opts, Options{
43
		A:    "a-set",
44
		B:    "b-def",
45
		Env:  "from-env",
46
		Bool: true,
47
	})
48
}
49