fixed test

skycoin · Nov 7, 2019 · 031bedd · 031bedd
1 parent 82c7047
commit 031bedd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pkg/transport/discovery.go b/pkg/transport/discovery.go
@@ -3,6 +3,7 @@ package transport
 import (
 	"context"
 	"errors"
+	"fmt"
 	"sync"
 	"time"
 
@@ -90,7 +91,7 @@ func (td *mockDiscoveryClient) DeleteTransport(ctx context.Context, id uuid.UUID
 
 	_, ok := td.entries[id]
 	if !ok {
-		return errors.New("transport not found")
+		return fmt.Errorf("transport with id: %s not found in transport discovery", id)
 	}
 
 	delete(td.entries, id)

diff --git a/pkg/transport/manager_test.go b/pkg/transport/manager_test.go
@@ -142,8 +142,7 @@ func TestNewManager(t *testing.T) {
 
 		m2.DeleteTransport(tp2.Entry.ID)
 		entry, err = tpDisc.GetTransportByID(context.TODO(), tpID)
-		require.NoError(t, err)
-		assert.False(t, entry.IsUp)
+		require.Contains(t, err.Error(), "not found")
 	})
 }