44 "fmt"
55 "testing"
66
7- "github.com/fatih/set"
87 "github.com/stretchr/testify/assert"
98
109 "github.com/elastic/beats/libbeat/common"
@@ -17,32 +16,32 @@ const inputIdx = 3
1716const retValIdx = 4
1817
1918func TestFlattenCommonMapStr (t * testing.T ) {
20- emptyBlacklist := set . New ()
21- blacklist := set . New ("a.bMap" , "f" )
22- expectedAll := set . New ("a" , "a.bStr" , "a.bMap" , "a.bMap.cMap" , "a.bMap.cMap.d" , "a.bMap.cStr" , "a.bAnotherMap" , "a.bAnotherMap.e" , "f" )
23- expectedWoBlacklisted := set . New ("a" , "a.bStr" , "a.bAnotherMap" , "a.bAnotherMap.e" )
24- expectedAllPrefixed := set . New ("pre" , "pre.a" , "pre.a.bStr" , "pre.a.bMap" , "pre.a.bMap.cMap" , "pre.a.bMap.cMap.d" , "pre.a.bMap.cStr" , "pre.a.bAnotherMap" , "pre.a.bAnotherMap.e" , "pre.f" )
25- expectedWithFilledInput := set . New ("prefilled" , "a" , "a.bStr" , "a.bAnotherMap" , "a.bAnotherMap.e" )
19+ emptyBlacklist := NewSet ()
20+ blacklist := NewSet ("a.bMap" , "f" )
21+ expectedAll := NewSet ("a" , "a.bStr" , "a.bMap" , "a.bMap.cMap" , "a.bMap.cMap.d" , "a.bMap.cStr" , "a.bAnotherMap" , "a.bAnotherMap.e" , "f" )
22+ expectedWoBlacklisted := NewSet ("a" , "a.bStr" , "a.bAnotherMap" , "a.bAnotherMap.e" )
23+ expectedAllPrefixed := NewSet ("pre" , "pre.a" , "pre.a.bStr" , "pre.a.bMap" , "pre.a.bMap.cMap" , "pre.a.bMap.cMap.d" , "pre.a.bMap.cStr" , "pre.a.bAnotherMap" , "pre.a.bAnotherMap.e" , "pre.f" )
24+ expectedWithFilledInput := NewSet ("prefilled" , "a" , "a.bStr" , "a.bAnotherMap" , "a.bAnotherMap.e" )
2625 data := [][]interface {}{
27- []interface {}{common.MapStr {}, "whatever" , emptyBlacklist , set . New (), set . New ("whatever" )},
28- []interface {}{common.MapStr {}, "" , blacklist , set . New (), set . New ()},
29- []interface {}{commonMapStr (), "" , emptyBlacklist , set . New (), expectedAll },
30- []interface {}{commonMapStr (), "" , blacklist , set . New (), expectedWoBlacklisted },
31- []interface {}{commonMapStr (), "pre" , emptyBlacklist , set . New (), expectedAllPrefixed },
32- []interface {}{commonMapStr (), "" , blacklist , set . New ("prefilled" ), expectedWithFilledInput },
26+ []interface {}{common.MapStr {}, "whatever" , emptyBlacklist , NewSet (), NewSet ("whatever" )},
27+ []interface {}{common.MapStr {}, "" , blacklist , NewSet (), NewSet ()},
28+ []interface {}{commonMapStr (), "" , emptyBlacklist , NewSet (), expectedAll },
29+ []interface {}{commonMapStr (), "" , blacklist , NewSet (), expectedWoBlacklisted },
30+ []interface {}{commonMapStr (), "pre" , emptyBlacklist , NewSet (), expectedAllPrefixed },
31+ []interface {}{commonMapStr (), "" , blacklist , NewSet ("prefilled" ), expectedWithFilledInput },
3332 }
3433 for idx , dataRow := range data {
3534 m := dataRow [mapDataIdx ].(common.MapStr )
3635 prefix := dataRow [prefixIdx ].(string )
37- blacklist := dataRow [blacklistedIdx ].(* set. Set )
38- flattened := dataRow [inputIdx ].(* set. Set )
36+ blacklist := dataRow [blacklistedIdx ].(* Set )
37+ flattened := dataRow [inputIdx ].(* Set )
3938
4039 flattenMapStr (m , prefix , blacklist , flattened )
41- expected := dataRow [retValIdx ].(* set. Set )
42- diff := set . SymmetricDifference (flattened , expected ).( * set. Set )
40+ expected := dataRow [retValIdx ].(* Set )
41+ diff := SymmDifference (flattened , expected )
4342
4443 errMsg := fmt .Sprintf ("Failed for idx %v, diff: %v" , idx , diff )
45- assert .Equal (t , 0 , diff .Size (), errMsg )
44+ assert .Equal (t , 0 , diff .Len (), errMsg )
4645 }
4746}
4847
@@ -70,8 +69,8 @@ func TestLoadFields(t *testing.T) {
7069
7170 fields , err := loadFields ("./_meta/fields.yml" )
7271 assert .Nil (t , err )
73- expected := set . New ("transaction" , "transaction.id" , "transaction.context" , "exception" , "exception.http" , "exception.http.url" , "exception.http.meta" , "exception.stacktrace" )
74- flattened := set . New ()
72+ expected := NewSet ("transaction" , "transaction.id" , "transaction.context" , "exception" , "exception.http" , "exception.http.url" , "exception.http.meta" , "exception.stacktrace" )
73+ flattened := NewSet ()
7574 flattenFieldNames (fields , "" , addAllFields , flattened )
7675 assert .Equal (t , expected , flattened )
7776}
@@ -80,14 +79,14 @@ func TestFlattenFieldNames(t *testing.T) {
8079
8180 fields , _ := loadFields ("./_meta/fields.yml" )
8281
83- expected := set . New ("transaction" , "transaction.id" , "transaction.context" , "exception" , "exception.http" , "exception.http.url" , "exception.http.meta" , "exception.stacktrace" )
82+ expected := NewSet ("transaction" , "transaction.id" , "transaction.context" , "exception" , "exception.http" , "exception.http.url" , "exception.http.meta" , "exception.stacktrace" )
8483
85- flattened := set . New ()
84+ flattened := NewSet ()
8685 flattenFieldNames (fields , "" , addAllFields , flattened )
8786 assert .Equal (t , expected , flattened )
8887
89- flattened = set . New ()
88+ flattened = NewSet ()
9089 flattenFieldNames (fields , "" , addOnlyDisabledFields , flattened )
91- expected = set . New ("transaction.context" , "exception.stacktrace" )
90+ expected = NewSet ("transaction.context" , "exception.stacktrace" )
9291 assert .Equal (t , expected , flattened )
9392}
0 commit comments