118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package wmi
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/go-kit/log"
|
|
)
|
|
|
|
type fakeWmiClass struct {
|
|
Name string
|
|
SomeProperty int
|
|
}
|
|
|
|
var (
|
|
mapQueryAll = func(src interface{}, class string, where string) string {
|
|
return QueryAll(src, log.NewNopLogger())
|
|
}
|
|
mapQueryAllWhere = func(src interface{}, class string, where string) string {
|
|
return QueryAllWhere(src, where, log.NewNopLogger())
|
|
}
|
|
mapQueryAllForClass = func(src interface{}, class string, where string) string {
|
|
return QueryAllForClass(src, class, log.NewNopLogger())
|
|
}
|
|
mapQueryAllForClassWhere = func(src interface{}, class string, where string) string {
|
|
return QueryAllForClassWhere(src, class, where, log.NewNopLogger())
|
|
}
|
|
)
|
|
|
|
type queryFunc func(src interface{}, class string, where string) string
|
|
|
|
func TestCreateQuery(t *testing.T) {
|
|
cases := []struct {
|
|
desc string
|
|
dst interface{}
|
|
class string
|
|
where string
|
|
queryFunc queryFunc
|
|
expected string
|
|
}{
|
|
{
|
|
desc: "queryAll on single instance",
|
|
dst: fakeWmiClass{},
|
|
queryFunc: mapQueryAll,
|
|
expected: "SELECT * FROM fakeWmiClass",
|
|
},
|
|
{
|
|
desc: "queryAll on slice",
|
|
dst: []fakeWmiClass{},
|
|
queryFunc: mapQueryAll,
|
|
expected: "SELECT * FROM fakeWmiClass",
|
|
},
|
|
{
|
|
desc: "queryAllWhere on single instance",
|
|
dst: fakeWmiClass{},
|
|
where: "foo = bar",
|
|
queryFunc: mapQueryAllWhere,
|
|
expected: "SELECT * FROM fakeWmiClass WHERE foo = bar",
|
|
},
|
|
{
|
|
desc: "queryAllWhere on slice",
|
|
dst: []fakeWmiClass{},
|
|
where: "foo = bar",
|
|
queryFunc: mapQueryAllWhere,
|
|
expected: "SELECT * FROM fakeWmiClass WHERE foo = bar",
|
|
},
|
|
{
|
|
desc: "queryAllWhere on single instance with empty where",
|
|
dst: fakeWmiClass{},
|
|
queryFunc: mapQueryAllWhere,
|
|
expected: "SELECT * FROM fakeWmiClass",
|
|
},
|
|
{
|
|
desc: "queryAllForClass on single instance",
|
|
dst: fakeWmiClass{},
|
|
class: "someClass",
|
|
queryFunc: mapQueryAllForClass,
|
|
expected: "SELECT * FROM someClass",
|
|
},
|
|
{
|
|
desc: "queryAllForClass on slice",
|
|
dst: []fakeWmiClass{},
|
|
class: "someClass",
|
|
queryFunc: mapQueryAllForClass,
|
|
expected: "SELECT * FROM someClass",
|
|
},
|
|
{
|
|
desc: "queryAllForClassWhere on single instance",
|
|
dst: fakeWmiClass{},
|
|
class: "someClass",
|
|
where: "foo = bar",
|
|
queryFunc: mapQueryAllForClassWhere,
|
|
expected: "SELECT * FROM someClass WHERE foo = bar",
|
|
},
|
|
{
|
|
desc: "queryAllForClassWhere on slice",
|
|
dst: []fakeWmiClass{},
|
|
class: "someClass",
|
|
where: "foo = bar",
|
|
queryFunc: mapQueryAllForClassWhere,
|
|
expected: "SELECT * FROM someClass WHERE foo = bar",
|
|
},
|
|
{
|
|
desc: "queryAllForClassWhere on single instance with empty where",
|
|
dst: fakeWmiClass{},
|
|
class: "someClass",
|
|
queryFunc: mapQueryAllForClassWhere,
|
|
expected: "SELECT * FROM someClass",
|
|
},
|
|
}
|
|
for _, c := range cases {
|
|
t.Run(c.desc, func(t *testing.T) {
|
|
if q := c.queryFunc(c.dst, c.class, c.where); q != c.expected {
|
|
t.Errorf("Case %q failed: Expected %q, got %q", c.desc, c.expected, q)
|
|
}
|
|
})
|
|
}
|
|
}
|