mirror of
https://github.com/refraction-networking/uquic.git
synced 2025-04-03 04:07:35 +03:00
56 lines
1.5 KiB
Go
56 lines
1.5 KiB
Go
package qlog
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/refraction-networking/uquic/internal/protocol"
|
|
"github.com/refraction-networking/uquic/logging"
|
|
)
|
|
|
|
var _ = Describe("qlog dir tests", Serial, func() {
|
|
var originalQlogDirValue string
|
|
var tempTestDirPath string
|
|
ctx := context.Background()
|
|
perspective := logging.PerspectiveClient
|
|
connID, _ := protocol.GenerateConnectionIDForInitial()
|
|
|
|
BeforeEach(func() {
|
|
originalQlogDirValue = os.Getenv("QLOGDIR")
|
|
var err error
|
|
tempTestDirPath, err = os.MkdirTemp("", "temp_test_dir")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
})
|
|
|
|
AfterEach(func() {
|
|
err := os.Setenv("QLOGDIR", originalQlogDirValue)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
err = os.RemoveAll(tempTestDirPath)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
})
|
|
|
|
It("environment variable is set", func() {
|
|
qlogDir := path.Join(tempTestDirPath, "qlogs")
|
|
err := os.Setenv("QLOGDIR", qlogDir)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
tracer := DefaultTracer(ctx, perspective, connID)
|
|
Expect(tracer).ToNot(BeNil())
|
|
tracer.Close()
|
|
_, err = os.Stat(qlogDir)
|
|
qlogDirCreated := !os.IsNotExist(err)
|
|
Expect(qlogDirCreated).To(BeTrue())
|
|
childs, err := os.ReadDir(qlogDir)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(len(childs)).To(Equal(1))
|
|
})
|
|
|
|
It("environment variable is not set", func() {
|
|
err := os.Setenv("QLOGDIR", "")
|
|
Expect(err).ToNot(HaveOccurred())
|
|
tracer := DefaultTracer(ctx, perspective, connID)
|
|
Expect(tracer).To(BeNil())
|
|
})
|
|
})
|