diff options
author | Dan Duvall <dduvall@wikimedia.org> | 2018-03-06 20:31:58 -0800 |
---|---|---|
committer | Dan Duvall <dduvall@wikimedia.org> | 2018-03-19 15:55:16 -0700 |
commit | eb9b69dd3d710cb7afa1dfb6e23a5987842b21cc (patch) | |
tree | 049b11cc885e4e9f54aac8981c91a1bf3620e7af /vendor/github.com/pborman/getopt/util_test.go | |
parent | 6896e655eb5cc88b90e66979bc2d862eb92cbb9f (diff) | |
download | blubber-eb9b69dd3d710cb7afa1dfb6e23a5987842b21cc.tar.gz |
Allow for configuration policies
Summary:
Implements a rough interface for validating configuration against
arbitrary policy rules. Policies are provided as YAML and passed via the
command line as file paths or remote URIs.
The format of policies is:
enforcements:
- path: <path>
rule: <rule>
Where `<path>` is a YAML-ish path to a config field and `<rule>` is any
expression our config validator understands (expressions built in by the
validator library and custom tags defined in `config.validation.go`).
Example policy:
enforcements:
- path: variants.production.base
rule: oneof=debian:jessie debian:stretch
- path: variants.production.runs.as
rule: ne=foo
- path: variants.production.node.dependencies
rule: isfalse
Command flag parsing was implemented in `main.go` to support the new
`--policy=uri` flag and improve existing handling of `--version` and the
usage statement.
Test Plan: Run `go test ./...`.
Reviewers: thcipriani, demon, hashar, mmodell, #release-engineering-team
Reviewed By: thcipriani, #release-engineering-team
Tags: #release-engineering-team
Differential Revision: https://phabricator.wikimedia.org/D999
Diffstat (limited to 'vendor/github.com/pborman/getopt/util_test.go')
-rw-r--r-- | vendor/github.com/pborman/getopt/util_test.go | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/vendor/github.com/pborman/getopt/util_test.go b/vendor/github.com/pborman/getopt/util_test.go new file mode 100644 index 0000000..79a732a --- /dev/null +++ b/vendor/github.com/pborman/getopt/util_test.go @@ -0,0 +1,87 @@ +// Copyright 2013 Google Inc. All rights reserved. +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +package getopt + +import ( + "bytes" + "fmt" + "path" + "reflect" + "runtime" + "strings" +) + +var errorString string + +func reset() { + CommandLine.shortOptions = make(map[rune]*option) + CommandLine.longOptions = make(map[string]*option) + CommandLine.options = nil + CommandLine.args = nil + CommandLine.program = "" + errorString = "" +} + +func parse(args []string) { + err := CommandLine.Getopt(args, nil) + if err != nil { + b := &bytes.Buffer{} + + fmt.Fprintln(b, CommandLine.program+": "+err.Error()) + CommandLine.PrintUsage(b) + errorString = b.String() + } +} + +func badSlice(a, b []string) bool { + if len(a) != len(b) { + return true + } + for x, v := range a { + if b[x] != v { + return true + } + } + return false +} + +func loc() string { + _, file, line, _ := runtime.Caller(1) + return fmt.Sprintf("%s:%d", path.Base(file), line) +} + +func (o *option) Equal(opt *option) bool { + if o.value != nil && opt.value == nil { + return false + } + if o.value == nil && opt.value != nil { + return false + } + if o.value != nil && o.value.String() != opt.value.String() { + return false + } + + oc := *o + optc := *opt + oc.value = nil + optc.value = nil + return reflect.DeepEqual(&oc, &optc) +} + +func newStringValue(s string) *stringValue { return (*stringValue)(&s) } + +func checkError(err string) string { + switch { + case err == errorString: + return "" + case err == "": + return fmt.Sprintf("unexpected error %q", errorString) + case errorString == "": + return fmt.Sprintf("did not get expected error %q", err) + case !strings.HasPrefix(errorString, err): + return fmt.Sprintf("got error %q, want %q", errorString, err) + } + return "" +} |