Extract context test

Author: shamcode

tests/context_test.go

@@ -0,0 +1,35 @@
+package tests
+
+import (
+	"context"
+	"errors"
+	"testing"
+
+	asserts "github.com/shamcode/assert"
+	"github.com/shamcode/simd/executor"
+	"github.com/shamcode/simd/indexes/hash"
+	"github.com/shamcode/simd/namespace"
+	"github.com/shamcode/simd/query"
+)
+
+func Test_Context(t *testing.T) {
+	// Arrange
+	store := namespace.CreateNamespace[*User]()
+	store.AddIndex(hash.NewComparableHashIndex(userID, true))
+	asserts.Success(t, store.Insert(&User{ //nolint:exhaustruct
+		ID:     1,
+		Name:   "First",
+		Status: StatusActive,
+		Score:  10,
+	}))
+
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	// Act
+	_, err := executor.CreateQueryExecutor[*User](store).FetchTotal(ctx, query.NewBuilder[*User]().Query())
+
+	// Assert
+	asserts.Equals(t, "context canceled", err.Error(), "check error")
+	asserts.Equals(t, true, errors.Is(err, context.Canceled), "error is context.Canceled")
+}

tests/simd_test.go

@@ -3,7 +3,6 @@ package tests
 
 import (
 	"context"
-	"errors"
 	"regexp"
 	"testing"
 
@@ -427,22 +426,3 @@ func Test_FetchAllAndTotal(t *testing.T) { //nolint:maintidx
 		})
 	}
 }
-
-func Test_Context(t *testing.T) {
-	store := namespace.CreateNamespace[*User]()
-	store.AddIndex(hash.NewComparableHashIndex(userID, true))
-	asserts.Success(t, store.Insert(&User{
-		ID:     1,
-		Name:   "First",
-		Status: StatusActive,
-		Score:  10,
-	}))
-
-	ctx, cancel := context.WithCancel(context.Background())
-	cancel()
-
-	_, err := executor.CreateQueryExecutor[*User](store).FetchTotal(ctx, query.NewBuilder[*User]().Query())
-
-	asserts.Equals(t, "context canceled", err.Error(), "check error")
-	asserts.Equals(t, true, errors.Is(err, context.Canceled), "error is context.Canceled")
-}