diff options
Diffstat (limited to 'integrationtests/dmap_test.go')
| -rw-r--r-- | integrationtests/dmap_test.go | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/integrationtests/dmap_test.go b/integrationtests/dmap_test.go index bfca039..dc508e2 100644 --- a/integrationtests/dmap_test.go +++ b/integrationtests/dmap_test.go @@ -11,6 +11,8 @@ func TestDMap(t *testing.T) { stdoutFile := "dmap.stdout.tmp" csvFile := "dmap.csv.tmp" expectedCsvFile := "dmap.csv.expected" + queryFile := fmt.Sprintf("%s.query", csvFile) + expectedQueryFile := "dmap.csv.query.expected" query := fmt.Sprintf("from STATS select count($line),last($time),avg($goroutines),min(concurrentConnections),max(lifetimeConnections) group by $hostname outfile %s", csvFile) @@ -18,14 +20,18 @@ func TestDMap(t *testing.T) { t.Error(err) return } - if err := compareFiles(t, csvFile, expectedCsvFile); err != nil { t.Error(err) return } + if err := compareFiles(t, queryFile, expectedQueryFile); err != nil { + t.Error(err) + return + } os.Remove(stdoutFile) os.Remove(csvFile) + os.Remove(queryFile) } func TestDMap2(t *testing.T) { @@ -33,6 +39,8 @@ func TestDMap2(t *testing.T) { stdoutFile := "dmap2.stdout.tmp" csvFile := "dmap2.csv.tmp" expectedCsvFile := "dmap2.csv.expected" + queryFile := fmt.Sprintf("%s.query", csvFile) + expectedQueryFile := "dmap2.csv.query.expected" query := fmt.Sprintf("from STATS select count($time),$time,max($goroutines),avg($goroutines),min($goroutines) group by $time order by count($time) outfile %s", csvFile) @@ -40,12 +48,16 @@ func TestDMap2(t *testing.T) { t.Error(err) return } - if err := compareFilesContents(t, csvFile, expectedCsvFile); err != nil { t.Error(err) return } + if err := compareFiles(t, queryFile, expectedQueryFile); err != nil { + t.Error(err) + return + } os.Remove(stdoutFile) os.Remove(csvFile) + os.Remove(queryFile) } |