From 3344d2f41b86d2828b3252117e88a941a677a509 Mon Sep 17 00:00:00 2001
From: 9547 <nivefive9547@gmail.com>
Date: Tue, 3 Nov 2020 06:49:56 +0800
Subject: [PATCH] cluster/audit: add audit unit test

---
 pkg/cluster/audit/audit_test.go | 65 +++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 pkg/cluster/audit/audit_test.go

diff --git a/pkg/cluster/audit/audit_test.go b/pkg/cluster/audit/audit_test.go
new file mode 100644
index 0000000000..254a4460ec
--- /dev/null
+++ b/pkg/cluster/audit/audit_test.go
@@ -0,0 +1,65 @@
+// Copyright 2020 PingCAP, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package audit
+
+import (
+	"os"
+	"path"
+	"path/filepath"
+	"runtime"
+
+	. "github.com/pingcap/check"
+	"golang.org/x/sync/errgroup"
+)
+
+var _ = Suite(&testAuditSuite{})
+
+type testAuditSuite struct{}
+
+func currentDir() string {
+	_, file, _, _ := runtime.Caller(0)
+	return filepath.Dir(file)
+}
+
+func auditDir() string {
+	return path.Join(currentDir(), "testdata", "audit")
+}
+
+func (s *testAuditSuite) SetUpSuite(c *C) {
+	_ = os.RemoveAll(auditDir())
+	_ = os.MkdirAll(auditDir(), 0777)
+}
+
+func (s *testAuditSuite) TearDownSuite(c *C) {
+	_ = os.RemoveAll(auditDir())
+}
+
+func (s *testAuditSuite) TestOutputAuditLog(c *C) {
+	dir := auditDir()
+	var g errgroup.Group
+	for i := 0; i < 20; i++ {
+		g.Go(func() error { return OutputAuditLog(dir, []byte("audit log")) })
+	}
+	err := g.Wait()
+	c.Assert(err, IsNil)
+
+	var paths []string
+	err = filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
+		// simply filter the not relate files.
+		paths = append(paths, path)
+		return nil
+	})
+	c.Assert(err, IsNil)
+	c.Assert(len(paths), Equals, 20)
+}