Do not make setup transport managed in acceptTransport

pkg/transport/manager.go

@@ -371,7 +371,7 @@ func (tm *Manager) dialTransport(ctx context.Context, factory Factory, remote ci
 	return tr, entry, nil
 }
 
-func (tm *Manager) acceptTransport(ctx context.Context, factory Factory) (*ManagedTransport, error) {
+func (tm *Manager) acceptTransport(ctx context.Context, factory Factory) (Transport, error) {
 	tr, err := factory.Accept(ctx)
 	if err != nil {
 		return nil, err
@@ -409,28 +409,27 @@ func (tm *Manager) acceptTransport(ctx context.Context, factory Factory) (*Manag
 		oldTr.killWorker()
 	}
 
+	if isSetup {
+		select {
+		case <-tm.doneChan:
+			return nil, io.ErrClosedPipe
+		case tm.SetupTpChan <- tr:
+			return tr, nil
+		}
+	}
+
 	mTr := newManagedTransport(tr, *entry, true)
 
 	tm.mu.Lock()
 	tm.transports[entry.ID] = mTr
 	tm.mu.Unlock()
 
-	if isSetup {
-		select {
-		case <-tm.doneChan:
-			return nil, io.ErrClosedPipe
-		case tm.SetupTpChan <- mTr:
-			go tm.manageTransport(ctx, mTr, factory, remote)
-			return mTr, nil
-		}
-	} else {
-		select {
-		case <-tm.doneChan:
-			return nil, io.ErrClosedPipe
-		case tm.DataTpChan <- mTr:
-			go tm.manageTransport(ctx, mTr, factory, remote)
-			return mTr, nil
-		}
+	select {
+	case <-tm.doneChan:
+		return nil, io.ErrClosedPipe
+	case tm.DataTpChan <- mTr:
+		go tm.manageTransport(ctx, mTr, factory, remote)
+		return mTr, nil
 	}
 }
 

pkg/transport/manager_test.go

@@ -306,7 +306,7 @@ func ExampleMakeTransportID() {
 	// uuid is different for public and private transports
 }
 
-func ExampleManager_CreateTransport() {
+func ExampleManager_CreateDataTransport() {
 	// Repetition is required here to guarantee that correctness does not depends on order of edges
 	for i := 0; i < 4; i++ {
 		pkB, mgrA, err := MockTransportManager()