package querylog import ( "errors" "io" "os" "testing" "time" "github.com/stretchr/testify/assert" ) func TestQLogReaderEmpty(t *testing.T) { r, err := NewQLogReader([]string{}) assert.Nil(t, err) assert.NotNil(t, r) defer r.Close() // seek to the start err = r.SeekStart() assert.Nil(t, err) line, err := r.ReadNext() assert.Empty(t, line) assert.Equal(t, io.EOF, err) } func TestQLogReaderOneFile(t *testing.T) { // let's do one small file count := 10 filesCount := 1 testDir := prepareTestDir() defer func() { _ = os.RemoveAll(testDir) }() testFiles := prepareTestFiles(testDir, filesCount, count) r, err := NewQLogReader(testFiles) assert.Nil(t, err) assert.NotNil(t, r) defer r.Close() // seek to the start err = r.SeekStart() assert.Nil(t, err) // read everything read := 0 var line string for err == nil { line, err = r.ReadNext() if err == nil { assert.True(t, len(line) > 0) read++ } } assert.Equal(t, count*filesCount, read) assert.Equal(t, io.EOF, err) } func TestQLogReaderMultipleFiles(t *testing.T) { // should be large enough count := 10000 filesCount := 5 testDir := prepareTestDir() defer func() { _ = os.RemoveAll(testDir) }() testFiles := prepareTestFiles(testDir, filesCount, count) r, err := NewQLogReader(testFiles) assert.Nil(t, err) assert.NotNil(t, r) defer r.Close() // seek to the start err = r.SeekStart() assert.Nil(t, err) // read everything read := 0 var line string for err == nil { line, err = r.ReadNext() if err == nil { assert.True(t, len(line) > 0) read++ } } assert.Equal(t, count*filesCount, read) assert.Equal(t, io.EOF, err) } func TestQLogReader_Seek(t *testing.T) { count := 10000 filesCount := 2 testDir := prepareTestDir() t.Cleanup(func() { _ = os.RemoveAll(testDir) }) testFiles := prepareTestFiles(testDir, filesCount, count) r, err := NewQLogReader(testFiles) assert.Nil(t, err) assert.NotNil(t, r) t.Cleanup(func() { _ = r.Close() }) testCases := []struct { name string time string want error }{{ name: "not_too_old", time: "2020-02-19T04:04:56.920973+03:00", want: nil, }, { name: "old", time: "2020-02-19T01:28:16.920973+03:00", want: nil, }, { name: "first", time: "2020-02-19T04:09:55.920973+03:00", want: nil, }, { name: "last", time: "2020-02-19T01:23:16.920973+03:00", want: nil, }, { name: "non-existent_long_ago", time: "2000-02-19T01:23:16.920973+03:00", want: ErrTSTooEarly, }, { name: "non-existent_far_ahead", time: "2100-02-19T01:23:16.920973+03:00", want: nil, }, { name: "non-existent_but_could", time: "2020-02-18T22:36:37.000000+03:00", want: ErrTSNotFound, }} for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { timestamp, err := time.Parse(time.RFC3339Nano, tc.time) assert.Nil(t, err) err = r.SeekTS(timestamp.UnixNano()) assert.True(t, errors.Is(err, tc.want), err) }) } } func TestQLogReader_ReadNext(t *testing.T) { count := 10 filesCount := 1 testDir := prepareTestDir() t.Cleanup(func() { _ = os.RemoveAll(testDir) }) testFiles := prepareTestFiles(testDir, filesCount, count) r, err := NewQLogReader(testFiles) assert.Nil(t, err) assert.NotNil(t, r) t.Cleanup(func() { _ = r.Close() }) testCases := []struct { name string start int want error }{{ name: "ok", start: 0, want: nil, }, { name: "too_big", start: count + 1, want: io.EOF, }} for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { err := r.SeekStart() assert.Nil(t, err, err) for i := 1; i < tc.start; i++ { _, err := r.ReadNext() assert.Nil(t, err) } _, err = r.ReadNext() assert.Equal(t, tc.want, err) }) } } // TODO(e.burkov): Remove the tests below. Make tests above more compelling. func TestQLogReaderSeek(t *testing.T) { // more or less big file count := 10000 filesCount := 2 testDir := prepareTestDir() defer func() { _ = os.RemoveAll(testDir) }() testFiles := prepareTestFiles(testDir, filesCount, count) r, err := NewQLogReader(testFiles) assert.Nil(t, err) assert.NotNil(t, r) defer r.Close() // CASE 1: NOT TOO OLD LINE testSeekLineQLogReader(t, r, 300) // CASE 2: OLD LINE testSeekLineQLogReader(t, r, count-300) // CASE 3: FIRST LINE testSeekLineQLogReader(t, r, 0) // CASE 4: LAST LINE testSeekLineQLogReader(t, r, count) // CASE 5: Seek non-existent (too low) err = r.SeekTS(123) assert.NotNil(t, err) // CASE 6: Seek non-existent (too high) ts, _ := time.Parse(time.RFC3339, "2100-01-02T15:04:05Z07:00") err = r.SeekTS(ts.UnixNano()) assert.NotNil(t, err) } func testSeekLineQLogReader(t *testing.T, r *QLogReader, lineNumber int) { line, err := getQLogReaderLine(r, lineNumber) assert.Nil(t, err) ts := readQLogTimestamp(line) assert.NotEqualValues(t, 0, ts) // try seeking to that line now err = r.SeekTS(ts) assert.Nil(t, err) testLine, err := r.ReadNext() assert.Nil(t, err) assert.Equal(t, line, testLine) } func getQLogReaderLine(r *QLogReader, lineNumber int) (string, error) { err := r.SeekStart() if err != nil { return "", err } for i := 1; i < lineNumber; i++ { _, err := r.ReadNext() if err != nil { return "", err } } return r.ReadNext() }