tsmon: read hostname and region from config file

common/tsmon/iface_test.go

 // Copyright 2016 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 tsmon
 
 import (
         "testing"
 
         "golang.org/x/net/context"
@@ skipping 17 matching lines @@
                         err := InitializeFromFlags(c, &tsmonFlags)
                         So(err, ShouldBeNil)
                         defer Shutdown(c)
 
                         s := GetState(c).S
                         So(s.DefaultTarget(), ShouldHaveSameTypeAs, (*target.NetworkDevice)(nil))
                         target := s.DefaultTarget().(*target.NetworkDevice).AsProto()
                         So(target.GetHostname(), ShouldEqual, "autogen:test-m4")
                 })
 
+                Convey("with predefined autogenerated hostnames (Device)", func() {
+                        tsmonFlags.Target.AutoGenHostname = true
+                        tsmonFlags.Target.DeviceHostname = "test-m5"
+                        err := InitializeFromFlags(c, &tsmonFlags)
+                        So(err, ShouldBeNil)
+                        defer Shutdown(c)
+
+                        s := GetState(c).S
+                        So(s.DefaultTarget(), ShouldHaveSameTypeAs, (*target.NetworkDevice)(nil))
+                        target := s.DefaultTarget().(*target.NetworkDevice).AsProto()
+                        So(target.GetHostname(), ShouldEqual, "autogen:test-m5")
+                })
+
+                Convey("with predefined autogenerated hostnames (Task)", func() {
+                        tsmonFlags.Target.AutoGenHostname = true
+                        tsmonFlags.Target.TaskHostname = "test-m5"
+                        tsmonFlags.Target.TargetType = target.TaskType
+                        tsmonFlags.Target.TaskServiceName = "test-service"
+                        tsmonFlags.Target.TaskJobName = "test-job"
+                        err := InitializeFromFlags(c, &tsmonFlags)
+                        So(err, ShouldBeNil)
+                        defer Shutdown(c)
+
+                        s := GetState(c).S
+                        So(s.DefaultTarget(), ShouldHaveSameTypeAs, (*target.Task)(nil))
+                        target := s.DefaultTarget().(*target.Task).AsProto()
+                        So(target.GetHostName(), ShouldEqual, "autogen:test-m5")
+                })
+
                 Convey("with static hostnames", func() {
                         err := InitializeFromFlags(c, &tsmonFlags)
                         So(err, ShouldBeNil)
                         defer Shutdown(c)
 
                         s := GetState(c).S
                         So(s.DefaultTarget(), ShouldHaveSameTypeAs, (*target.NetworkDevice)(nil))
                         target := s.DefaultTarget().(*target.NetworkDevice).AsProto()
                         So(target.GetHostname(), ShouldEqual, "test-m4")
                 })
 
+                Convey("with predefined static hostnames (Device)", func() {
+                        tsmonFlags.Target.DeviceHostname = "test-m5"
+                        err := InitializeFromFlags(c, &tsmonFlags)
+                        So(err, ShouldBeNil)
+                        defer Shutdown(c)
+
+                        s := GetState(c).S
+                        So(s.DefaultTarget(), ShouldHaveSameTypeAs, (*target.NetworkDevice)(nil))
+                        target := s.DefaultTarget().(*target.NetworkDevice).AsProto()
+                        So(target.GetHostname(), ShouldEqual, "test-m5")
+                })
+
+                Convey("with predefined static hostnames (Task)", func() {
+                        tsmonFlags.Target.TaskHostname = "test-m5"
+                        tsmonFlags.Target.TargetType = target.TaskType
+                        tsmonFlags.Target.TaskServiceName = "test-service"
+                        tsmonFlags.Target.TaskJobName = "test-job"
+                        err := InitializeFromFlags(c, &tsmonFlags)
+                        So(err, ShouldBeNil)
+                        defer Shutdown(c)
+
+                        s := GetState(c).S
+                        So(s.DefaultTarget(), ShouldHaveSameTypeAs, (*target.Task)(nil))
+                        target := s.DefaultTarget().(*target.Task).AsProto()
+                        So(target.GetHostName(), ShouldEqual, "test-m5")
+                })
+
         })
 }