api: use clock package

shellhub-io · Jun 11, 2021 · 5fa3e65 · 5fa3e65
1 parent fd60cc8
commit 5fa3e65
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/store/mongo/migrations/migration_23_test.go b/api/store/mongo/migrations/migration_23_test.go
@@ -3,9 +3,9 @@ package migrations
 import (
 	"context"
 	"testing"
-	"time"
 
 	"github.com/shellhub-io/shellhub/api/pkg/dbtest"
+	"github.com/shellhub-io/shellhub/pkg/clock"
 	"github.com/shellhub-io/shellhub/pkg/models"
 	"github.com/stretchr/testify/assert"
 	migrate "github.com/xakep666/mongo-migrate"
@@ -48,7 +48,7 @@ func TestMigration23(t *testing.T) {
 		UID:      "uid",
 		Identity: &models.DeviceIdentity{MAC: "mac"},
 		TenantID: "tenant",
-		LastSeen: time.Now(),
+		LastSeen: clock.Now(),
 	}
 	_, err = db.Client().Database("test").Collection("devices").InsertOne(context.TODO(), device)
 	assert.NoError(t, err)
@@ -58,7 +58,7 @@ func TestMigration23(t *testing.T) {
 		UID:      "uid2",
 		Identity: &models.DeviceIdentity{MAC: "mac"},
 		TenantID: "tenant2",
-		LastSeen: time.Now(),
+		LastSeen: clock.Now(),
 	}
 	_, err = db.Client().Database("test").Collection("devices").InsertOne(context.TODO(), device)
 	assert.NoError(t, err)