Butler stream servers can generate client address.

logdog/client/butlerlib/streamclient/client_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 streamclient
 
 import (
         "bytes"
         "errors"
         "io"
@@ skipping 61 matching lines @@
                 })
 
                 Convey(`Can instantiate a new client.`, func() {
                         client, err := reg.NewClient("test:foo")
                         So(err, ShouldBeNil)
                         So(client, ShouldHaveSameTypeAs, &clientImpl{})
 
                         Convey(`That can instantiate new Streams.`, func() {
                                 stream, err := client.NewStream(flags)
                                 So(err, ShouldBeNil)
-                                So(stream, ShouldHaveSameTypeAs, &streamImpl{})
+                                So(stream, ShouldHaveSameTypeAs, &BaseStream{})
 
-                                si := stream.(*streamImpl)
+                                si := stream.(*BaseStream)
                                 So(si.WriteCloser, ShouldHaveSameTypeAs, &testStreamWriteCloser{})
 
                                 tswc := si.WriteCloser.(*testStreamWriteCloser)
                                 So(tswc.addr, ShouldEqual, "foo")
 
                                 Convey(`The stream should have the stream header written to it.`, func() {
                                         So(tswc.Next(len(streamproto.ProtocolFrameHeaderMagic)), ShouldResemble,
                                                 streamproto.ProtocolFrameHeaderMagic)
 
                                         r := recordio.NewReader(tswc, -1)
                                         f, err := r.ReadFrameAll()
                                         So(err, ShouldBeNil)
                                         So(string(f), ShouldResemble, `{"name":"test","timestamp":"0001-02-03T04:05:06.000000007Z"}`)
                                 })
                         })
 
                         Convey(`If the stream fails to write the handshake, it will be closed.`, func() {
                                 tswcErr = errors.New("test error")
                                 _, err := client.NewStream(flags)
                                 So(err, ShouldNotBeNil)
 
                                 So(tswc, ShouldNotBeNil)
                                 So(tswc.closed, ShouldBeTrue)
                         })
                 })
         })
 }