[logdog/common/types] make StreamAddr implement flag.Value.

logdog/common/types/streamaddr_test.go

 // Copyright 2015 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 types
 
 import (
+        "flag"
         "fmt"
         "net/url"
         "testing"
 
         . "github.com/luci/luci-go/common/testing/assertions"
         . "github.com/smartystreets/goconvey/convey"
 )
 
 func TestStreamAddr(t *testing.T) {
         t.Parallel()
@@ skipping 32 matching lines @@
                         })
                 }
 
                 for _, tc := range failures {
                         Convey(fmt.Sprintf(`Failure: %q fails like: %q`, tc.s, tc.err), func() {
                                 _, err := ParseURL(tc.s)
                                 So(err, ShouldErrLike, tc.err)
                         })
                 }
         })
+
+        Convey(`StreamAddr is a flag.Value`, t, func() {
+                fs := flag.NewFlagSet("testing", flag.ContinueOnError)
+                a := &StreamAddr{}
+
+                fs.Var(a, "addr", "its totally an address of a thing")
+
+                Convey(`good`, func() {
+                        So(fs.Parse([]string{"-addr", "logdog://host/project/a/+/b"}), ShouldBeNil)
+                        So(a, ShouldResemble, &StreamAddr{
+                                "host",
+                                "project",
+                                "a/+/b",
+                        })
+                })
+
+                Convey(`bad`, func() {
+                        So(fs.Parse([]string{"-addr", "://host/project/a/+/b"}), ShouldErrLike,
+                                "failed to parse URL")
+                })
+        })
 }