fix UT

Signed-off-by: lance6716 <lance6716@gmail.com>
pingcap · Feb 21, 2023 · 12a92a7 · 12a92a7
1 parent 0edabe5
commit 12a92a7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/executor/loadremotetest/BUILD.bazel b/executor/loadremotetest/BUILD.bazel
@@ -10,6 +10,7 @@ go_test(
         "one_sqldump_test.go",
         "util_test.go",
     ],
+    embedsrcs = ["test.parquet"],
     flaky = True,
     deps = [
         "//executor",

diff --git a/executor/loadremotetest/one_parquet_test.go b/executor/loadremotetest/one_parquet_test.go
@@ -15,17 +15,17 @@
 package loadremotetest
 
 import (
+	_ "embed"
 	"fmt"
-	"os"
 
 	"github.com/fsouza/fake-gcs-server/fakestorage"
 	"github.com/pingcap/tidb/testkit"
 )
 
-func (s *mockGCSSuite) TestLoadParquet() {
-	content, err := os.ReadFile("test.parquet")
-	s.Require().NoError(err)
+//go:embed test.parquet
+var content []byte
 
+func (s *mockGCSSuite) TestLoadParquet() {
 	s.tk.MustExec("DROP DATABASE IF EXISTS load_csv;")
 	s.tk.MustExec("CREATE DATABASE load_csv;")
 	s.tk.MustExec("CREATE TABLE load_csv.t (" +