Implement MR remarks

SkynetLabs · Mar 16, 2022 · f394f74 · f394f74
1 parent 9a4a807
commit f394f74
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/database/database.go b/database/database.go
@@ -170,7 +170,7 @@ func NewTestDB(ctx context.Context, dbName string, logger *logrus.Logger) *DB {
 	}
 
 	// create a context with timeout and purge the database
-	ctx, cancel := context.WithTimeout(context.Background(), MongoDefaultTimeout)
+	ctx, cancel := context.WithTimeout(ctx, MongoDefaultTimeout)
 	defer cancel()
 	err = db.Purge(ctx)
 	if err != nil {

diff --git a/main.go b/main.go
@@ -13,7 +13,6 @@ import (
 	"github.com/SkynetLabs/blocker/api"
 	"github.com/SkynetLabs/blocker/blocker"
 	"github.com/SkynetLabs/blocker/database"
-	"github.com/SkynetLabs/blocker/skyd"
 	"github.com/SkynetLabs/blocker/syncer"
 	"github.com/joho/godotenv"
 	"github.com/sirupsen/logrus"