Function writeToLogWriterAndVerify

x/log_writer_test.go:119–158

(t *testing.T, lw *LogWriter, path string)

}

func writeToLogWriterAndVerify(t *testing.T, lw *LogWriter, path string) {
	msg := []byte("abcd")
	msg = bytes.Repeat(msg, 256)
	msg[1023] = '\n'
	for range 10 {
		go func() {
			for range 1000 {
				n, err := lw.Write(msg)
				require.NoError(t, err)
				require.Equal(t, n, len(msg), "write length is not equal")
			}
		}()
	}
	time.Sleep(time.Second * 10)
	require.NoError(t, lw.Close())
	files, err := os.ReadDir("./log_test")
	require.NoError(t, err)

	lineCount := 0
	for _, f := range files {
		file, _ := os.Open(filepath.Join(filepath.Dir(path), f.Name()))

		var fileScanner *bufio.Scanner
		if strings.HasSuffix(file.Name(), ".gz") {
			gz, err := gzip.NewReader(file)
			require.NoError(t, err)
			all, err := io.ReadAll(gz)
			require.NoError(t, err)
			fileScanner = bufio.NewScanner(bytes.NewReader(all))
			gz.Close()
		} else {
			fileScanner = bufio.NewScanner(file)
		}
		for fileScanner.Scan() {
			lineCount = lineCount + 1
		}
	}

	require.Equal(t, lineCount, 10000)
}