cmd/otk-resolve-ostree-commit: test the mock resolver

cmd/otk-resolve-ostree-commit/export_test.go

@@ -3,3 +3,16 @@ package main
 var (
 	Run = run
 )
+
+func MockEnvLookup() (restore func()) {
+	saved := osLookupEnv
+	osLookupEnv = func(key string) (string, bool) {
+		if key == "OTK_UNDER_TEST" {
+			return "1", true
+		}
+		return "", false
+	}
+	return func() {
+		osLookupEnv = saved
+	}
+}

cmd/otk-resolve-ostree-commit/main_test.go

@@ -191,3 +191,35 @@ func TestResolverErrors(t *testing.T) {
 		})
 	}
 }
+
+func TestMockResolve(t *testing.T) {
+	restore := resolver.MockEnvLookup()
+	defer restore()
+
+	assert := assert.New(t)
+
+	inputReq := `
+{
+  "tree": {
+    "ref": "otk/ostree/test",
+    "url": "https://ostree.example.org/repo"
+  }
+}
+`
+	expOutput := `{
+  "tree": {
+    "const": {
+      "ref": "otk/ostree/test",
+      "url": "https://ostree.example.org/repo",
+      "checksum": "5c1c0655481926623eca85109dd4017c8dba320ea1473c42b43005db6d900a60"
+    }
+  }
+}
+`
+	input := bytes.NewBuffer([]byte(inputReq))
+	output := &bytes.Buffer{}
+
+	assert.NoError(resolver.Run(input, output))
+
+	assert.Equal(expOutput, output.String())
+}