CL-Softwares
/
telegraf
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test(outputs.sql): simplify clickhouse test (#12189)

This commit is contained in:
Joshua Powers 2022-11-07 09:06:30 -07:00 committed by GitHub
parent 49c1c28461
commit b3a460f7e9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 27 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
plugins/outputs/sql/sql_test.go
View File

@ -386,52 +386,32 @@ func TestClickHouseIntegration(t *testing.T) {
require.NoError(t, p.Connect())
require.NoError(t, p.Write(testMetrics))
// wait for last test metric to get written
require.Eventually(t, func() bool {
var out io.Reader
_, out, err = container.Exec([]string{
"bash",
"-c",
"clickhouse-client" +
" --user=" + username +
" --database=" + dbname +
" --format=TabSeparatedRaw" +
" --multiquery --query=" +
"\"SELECT * FROM \\\"metric three\\\";" +
"SHOW CREATE TABLE \\\"metric three\\\"\"",
})
require.NoError(t, err)
bytes, err := io.ReadAll(out)
require.NoError(t, err)
return strings.Contains(string(bytes), "`string two` String")
}, 5*time.Second, 10*time.Millisecond)
// dump the database
var rc int
for _, testMetric := range testMetrics {
rc, _, err = container.Exec([]string{
"bash",
"-c",
"clickhouse-client" +
" --user=" + username +
" --database=" + dbname +
" --format=TabSeparatedRaw" +
" --multiquery --query=" +
"\"SELECT * FROM \\\"" + testMetric.Name() + "\\\";" +
"SHOW CREATE TABLE \\\"" + testMetric.Name() + "\\\"\"" +
" >> /out/dump 2>&1",
})
require.NoError(t, err)
require.Equal(t, 0, rc)
cases := []struct {
table string
expected string
}{
{"metric_one", "`float64_one` Float64"},
{"metric_two", "`string_one` String"},
{"metric three", "`string two` String"},
}
for _, tc := range cases {
require.Eventually(t, func() bool {
var out io.Reader
_, out, err = container.Exec([]string{
"bash",
"-c",
"clickhouse-client" +
" --user=" + username +
" --database=" + dbname +
" --format=TabSeparatedRaw" +
" --multiquery --query=" +
"\"SELECT * FROM \\\"" + tc.table + "\\\";" +
"SHOW CREATE TABLE \\\"" + tc.table + "\\\"\"",
})
require.NoError(t, err)
bytes, err := io.ReadAll(out)
require.NoError(t, err)
return strings.Contains(string(bytes), tc.expected)
}, 5*time.Second, 10*time.Millisecond)
}
dumpfile := filepath.Join(outDir, "dump")
require.FileExists(t, dumpfile)
//compare the dump to what we expected
expected, err := os.ReadFile("testdata/clickhouse/expected.txt")
require.NoError(t, err)
actual, err := os.ReadFile(dumpfile)
require.NoError(t, err)
require.Equal(t, string(expected), string(actual))
}