console: split graph.go into query.go and graph.go

We will introduce more query types.
This commit is contained in:
Vincent Bernat
2022-05-21 19:53:08 +02:00
parent 0879da9512
commit 375a0fe88f
5 changed files with 184 additions and 170 deletions

42
console/query_test.go Normal file
View File

@@ -0,0 +1,42 @@
package console
import (
"testing"
"akvorado/common/helpers"
)
func TestQueryColumnSQLSelect(t *testing.T) {
cases := []struct {
Input queryColumn
Expected string
}{
{
Input: queryColumnSrcAddr,
Expected: `IPv6NumToString(SrcAddr)`,
}, {
Input: queryColumnDstAS,
Expected: `concat(toString(DstAS), ': ', dictGetOrDefault('asns', 'name', DstAS, '???'))`,
}, {
Input: queryColumnProto,
Expected: `dictGetOrDefault('protocols', 'name', Proto, '???')`,
}, {
Input: queryColumnEType,
Expected: `if(EType = 0x800, 'IPv4', if(EType = 0x86dd, 'IPv6', '???'))`,
}, {
Input: queryColumnOutIfSpeed,
Expected: `toString(OutIfSpeed)`,
}, {
Input: queryColumnExporterName,
Expected: `ExporterName`,
},
}
for _, tc := range cases {
t.Run(tc.Input.String(), func(t *testing.T) {
got := tc.Input.toSQLSelect()
if diff := helpers.Diff(got, tc.Expected); diff != "" {
t.Errorf("toSQLWhere (-got, +want):\n%s", diff)
}
})
}
}