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

1 files changed, 74 insertions, 0 deletions

diff --git a/vendor/github.com/utahta/go-openuri/openuri_test.go b/vendor/github.com/utahta/go-openuri/openuri_test.go
new file mode 100644
index 0000000..7f653e6
--- /dev/null
+++ b/vendor/github.com/utahta/go-openuri/openuri_test.go
@@ -0,0 +1,74 @@
+package openuri
+
+import (
+	"io/ioutil"
+	"net/http"
+	"strings"
+	"testing"
+)
+
+type dummyRoundTripper struct{}
+
+func (d *dummyRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
+	return &http.Response{Status: "dummy", Body: ioutil.NopCloser(strings.NewReader("body dummy"))}, nil
+}
+
+func TestNew(t *testing.T) {
+	_, err := New()
+	if err != nil {
+		t.Error(err)
+	}
+}
+
+func TestWithHTTPClient(t *testing.T) {
+	c, err := New(WithHTTPClient(&http.Client{Transport: &dummyRoundTripper{}}))
+	if err != nil {
+		t.Error(err)
+	}
+
+	resp, _ := c.httpClient.Get("test")
+	if resp.Status != "dummy" {
+		t.Errorf("Expected status dummy, got %s", resp.Status)
+	}
+}
+
+func TestOpen_File(t *testing.T) {
+	o, err := Open("./openuri.go")
+	if err != nil {
+		t.Error(err)
+	}
+
+	b, err := ioutil.ReadAll(o)
+	if err != nil {
+		t.Error(err)
+	}
+
+	if !strings.HasPrefix(string(b), "package openuri") {
+		t.Errorf("Expected open file, go %s", string(b))
+	}
+}
+
+func TestOpen_URL(t *testing.T) {
+	tests := []struct {
+		url string
+	}{
+		{"http://example.com"},
+		{"https://example.com"},
+	}
+
+	for _, test := range tests {
+		o, err := Open(test.url, WithHTTPClient(&http.Client{Transport: &dummyRoundTripper{}}))
+		if err != nil {
+			t.Error(err)
+		}
+
+		b, err := ioutil.ReadAll(o)
+		if err != nil {
+			t.Error(err)
+		}
+
+		if !strings.HasPrefix(string(b), "body dummy") {
+			t.Errorf("Expected open file, go %s", string(b))
+		}
+	}
+}