瀏覽代碼

fix bucket test

AlteredCoder 5 年之前
父節點
當前提交
950344a398
共有 1 個文件被更改,包括 5 次插入1 次删除
  1. 5 1
      pkg/leakybucket/buckets_test.go

+ 5 - 1
pkg/leakybucket/buckets_test.go

@@ -22,6 +22,10 @@ type TestFile struct {
 	Results []types.Event `yaml:"results,omitempty"`
 }
 
+var (
+	configDir = "./data"
+)
+
 func TestBucket(t *testing.T) {
 
 	var envSetting = os.Getenv("TEST_ONLY")
@@ -76,7 +80,7 @@ func testOneBucket(t *testing.T, dir string) error {
 	for _, x := range stages {
 		files = append(files, x.Filename)
 	}
-	holders, response, err := LoadBuckets(files)
+	holders, response, err := LoadBuckets(files, configDir)
 	if err != nil {
 		t.Fatalf("failed loading bucket : %s", err)
 	}