mirror of https://github.com/databricks/cli.git
33 lines
606 B
Go
33 lines
606 B
Go
|
package log
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"golang.org/x/exp/slog"
|
||
|
|
)
|
||
|
|
|
||
|
|
func testReplaceA(groups []string, a slog.Attr) slog.Attr {
|
||
|
|
if a.Key == "foo" {
|
||
|
|
return slog.Int("foo", int(a.Value.Int64())+1)
|
||
|
|
}
|
||
|
|
return a
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestReplaceAttrGroup(t *testing.T) {
|
||
|
|
var foo, bar, out slog.Attr
|
||
|
|
|
||
|
|
fn := ReplaceAttrFunctions{
|
||
|
|
testReplaceA,
|
||
|
|
testReplaceA,
|
||
|
|
}
|
||
|
|
|
||
|
|
foo = slog.Int("foo", 0)
|
||
|
|
out = fn.ReplaceAttr([]string{}, foo)
|
||
|
|
assert.EqualValues(t, 2, out.Value.Int64())
|
||
|
|
|
||
|
|
bar = slog.Int("bar", 0)
|
||
|
|
out = fn.ReplaceAttr([]string{}, bar)
|
||
|
|
assert.EqualValues(t, 0, out.Value.Int64())
|
||
|
|
}
|