Kitchen: Generate LogDog prefix from Swarming.

go/src/infra/tools/kitchen/cook_logdog_test.go

+// Copyright 2016 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+package main
+
+import (
+        "testing"
+
+        "github.com/luci/luci-go/common/environ"
+        "github.com/luci/luci-go/common/logdog/types"
+
+        . "github.com/luci/luci-go/common/testing/assertions"
+        . "github.com/smartystreets/goconvey/convey"
+)
+
+func TestCookLogDogPrefix(t *testing.T) {
+        Convey(`With a fake environment`, t, func() {
+                env := environ.New([]string{
+                        "SWARMING_SERVER=server.appspot.com",
+                        "SWARMING_TASK_ID=1234567890abcdef",
+                })
+
+                Convey(`Can generate a LogDog prefix`, func() {
+                        p, err := getLogdogPrefix(env)
+                        So(err, ShouldBeNil)
+                        So(p, ShouldEqual, types.StreamName("swarm/server.appspot.com/1234567890abcdef"))
+                })
+
+                Convey(`If Swarming server is missing from the environment, will fail.`, func() {
+                        env.Set("SWARMING_SERVER", "")
+
+                        _, err := getLogdogPrefix(env)
+                        So(err, ShouldErrLike, "failed to get swarming task parameters")
+                })
+
+                Convey(`If Swarming task ID is missing from the environment, will fail.`, func() {
+                        env.Set("SWARMING_TASK_ID", "")
+
+                        _, err := getLogdogPrefix(env)
+                        So(err, ShouldErrLike, "failed to get swarming task parameters")
+                })
+        })
+}