make sure timeout in context timeout test (#1363)

go-xorm · Jul 23, 2019 · 674c908 · 674c908
1 parent 62d884e
commit 674c908
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/engine_context_test.go b/engine_context_test.go
@@ -20,6 +20,8 @@ func TestPingContext(t *testing.T) {
 	ctx, canceled := context.WithTimeout(context.Background(), time.Nanosecond)
 	defer canceled()
 
+	time.Sleep(time.Nanosecond)
+
 	err := testEngine.(*Engine).PingContext(ctx)
 	assert.Error(t, err)
 	assert.Contains(t, err.Error(), "context deadline exceeded")

diff --git a/session_context_test.go b/session_context_test.go
@@ -26,6 +26,9 @@ func TestQueryContext(t *testing.T) {
 
 	ctx, cancel := context.WithTimeout(context.Background(), time.Nanosecond)
 	defer cancel()
+
+	time.Sleep(time.Nanosecond)
+
 	has, err := testEngine.Context(ctx).Exist(&ContextQueryStruct{Name: "1"})
 	assert.Error(t, err)
 	assert.Contains(t, err.Error(), "context deadline exceeded")