close the Tracer when the Transport is closed

server_test.go

@@ -203,6 +203,7 @@ var _ = Describe("Server", func() {
  })
 
  AfterEach(func() {
+ tracer.EXPECT().Close()
  tr.Close()
  })
 
@@ -1429,7 +1430,10 @@ var _ = Describe("Server", func() {
  serv.connHandler = phm
  })
 
- AfterEach(func() { tr.Close() })
+ AfterEach(func() {
+ tracer.EXPECT().Close()
+ tr.Close()
+ })
 
  It("passes packets to existing connections", func() {
  connID := protocol.ParseConnectionID([]byte{1, 2, 3, 4, 5, 6, 7, 8})

transport.go

@@ -111,6 +111,7 @@ type Transport struct {
  MaxHandshakes int
 
  // A Tracer traces events that don't belong to a single QUIC connection.
+ // Tracer.Close is called when the transport is closed.
  Tracer *logging.Tracer
 
  handlerMap packetHandlerManager
@@ -366,6 +367,9 @@ func (t *Transport) close(e error) {
  if t.server != nil {
  t.server.close(e, false)
  }
+ if t.Tracer != nil && t.Tracer.Close != nil {
+ t.Tracer.Close()
+ }
  t.closed = true
 }
 

transport_test.go

@@ -141,6 +141,7 @@ var _ = Describe("Transport", func() {
  Eventually(dropped).Should(BeClosed())
 
  // shutdown
+ tracer.EXPECT().Close()
  close(packetChan)
  tr.Close()
  })
@@ -391,6 +392,7 @@ var _ = Describe("Transport", func() {
  Eventually(done).Should(BeClosed())
 
  // shutdown
+ tracer.EXPECT().Close()
  close(packetChan)
  tr.Close()
  })