milo: Add SwarmBucket templating.

buildbucket/swarmbucket/subst/subst_test.go

+// Copyright 2017 The LUCI Authors. All rights reserved.
+// Use of this source code is governed under the Apache License, Version 2.0
+// that can be found in the LICENSE file.
+
+package subst
+
+import (
+        "testing"
+
+        . "github.com/luci/luci-go/common/testing/assertions"
+        . "github.com/smartystreets/goconvey/convey"
+)
+
+func TestResolve(t *testing.T) {
+        t.Parallel()
+
+        Convey(`Testing substitution resolution`, t, func() {
+                var s Subst
+
+                shouldResolve := func(v string) string {
+                        val, err := s.Resolve(v)
+                        So(err, ShouldBeNil)
+                        return val
+                }
+
+                Convey(`Can resolve a string without any substitutions.`, func() {
+                        So(shouldResolve(""), ShouldEqual, "")
+                        So(shouldResolve("foo"), ShouldEqual, "foo")
+                        So(shouldResolve(`\${foo}`), ShouldEqual, `\${foo}`)
+                        So(shouldResolve(`\${foo`), ShouldEqual, `\${foo`)
+                })
+
+                Convey(`Will error if there is an incomplete substitution.`, func() {
+                        _, err := s.Resolve("foo-${bar")
+                        So(err, ShouldErrLike, "incomplete template")
+                })
+
+                Convey(`With substitutions defined, can apply.`, func() {
+                        s.SwarmingRunID = "12345"
+
+                        So(shouldResolve(""), ShouldEqual, "")
+                        So(shouldResolve("${swarming_run_id}"), ShouldEqual, "12345")
+                        So(shouldResolve("foo/${swarming_run_id}"), ShouldEqual, "foo/12345")
+                        So(shouldResolve("foo/${swarming_run_id}/bar"), ShouldEqual, "foo/12345/bar")
+                        So(shouldResolve("${swarming_run_id}/bar"), ShouldEqual, "12345/bar")
+                        So(shouldResolve("foo/${swarming_run_id}/bar/${swarming_run_id}/baz"), ShouldEqual, "foo/12345/bar/12345/baz")
+                        So(shouldResolve(`foo/${swarming_run_id}/bar/\${swarming_run_id}/baz`), ShouldEqual, `foo/12345/bar/\${swarming_run_id}/baz`)
+                })
+        })
+}