mirror of
https://github.com/hanzoai/tasks.git
synced 2026-10-30 14:56:19 +00:00
Rodrigo ZhouandGitHub
e6eea9390b
## What changed? Add `SearchAttributesMap.ToProto` function. Refactored `VisibilityValue` to account for the supported types in Visibility. ## Why? Added a way to encode the CHASM search attribute to proto. ## How did you test it? - [x] built - [ ] run locally and tested manually - [x] covered by existing tests - [x] added new unit test(s) - [ ] added new functional test(s) ## Potential risks
121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
package chasm
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"go.temporal.io/server/common/payload"
|
|
)
|
|
|
|
func TestVisibilityValue(t *testing.T) {
|
|
t.Run("Int64", func(t *testing.T) {
|
|
v := VisibilityValueInt64(9876543210)
|
|
p := v.MustEncode()
|
|
require.NotNil(t, p)
|
|
|
|
var out int64
|
|
err := payload.Decode(p, &out)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int64(9876543210), out)
|
|
|
|
require.True(t, v.Equal(VisibilityValueInt64(9876543210)))
|
|
require.False(t, v.Equal(VisibilityValueInt64(9876543211)))
|
|
})
|
|
|
|
t.Run("String", func(t *testing.T) {
|
|
v := VisibilityValueKeyword("hello, 世界")
|
|
p := v.MustEncode()
|
|
require.NotNil(t, p)
|
|
|
|
var out string
|
|
err := payload.Decode(p, &out)
|
|
require.NoError(t, err)
|
|
require.Equal(t, "hello, 世界", out)
|
|
|
|
require.True(t, v.Equal(VisibilityValueKeyword("hello, 世界")))
|
|
require.False(t, v.Equal(VisibilityValueKeyword("hello")))
|
|
require.False(t, v.Equal(VisibilityValueBool(true)))
|
|
})
|
|
|
|
t.Run("Bool", func(t *testing.T) {
|
|
v := VisibilityValueBool(true)
|
|
p := v.MustEncode()
|
|
require.NotNil(t, p)
|
|
|
|
var out bool
|
|
err := payload.Decode(p, &out)
|
|
require.NoError(t, err)
|
|
require.Equal(t, true, out)
|
|
|
|
require.True(t, v.Equal(VisibilityValueBool(true)))
|
|
require.False(t, v.Equal(VisibilityValueBool(false)))
|
|
require.False(t, v.Equal(VisibilityValueKeyword("true")))
|
|
})
|
|
|
|
t.Run("Float64", func(t *testing.T) {
|
|
v := VisibilityValueFloat64(3.14159)
|
|
p := v.MustEncode()
|
|
require.NotNil(t, p)
|
|
|
|
var out float64
|
|
err := payload.Decode(p, &out)
|
|
require.NoError(t, err)
|
|
require.InDelta(t, 3.14159, out, 1e-9)
|
|
|
|
require.True(t, v.Equal(VisibilityValueFloat64(3.14159)))
|
|
require.False(t, v.Equal(VisibilityValueFloat64(2.71828)))
|
|
})
|
|
|
|
t.Run("StringSlice", func(t *testing.T) {
|
|
v := VisibilityValueStringSlice([]string{"a", "b", "c"})
|
|
p := v.MustEncode()
|
|
require.NotNil(t, p)
|
|
|
|
var out []string
|
|
err := payload.Decode(p, &out)
|
|
require.NoError(t, err)
|
|
require.Equal(t, []string{"a", "b", "c"}, out)
|
|
|
|
require.True(t, v.Equal(VisibilityValueStringSlice([]string{"a", "b", "c"})))
|
|
require.False(t, v.Equal(VisibilityValueStringSlice([]string{"a", "c", "b"})))
|
|
require.False(t, v.Equal(VisibilityValueStringSlice([]string{"a", "b"})))
|
|
require.False(t, v.Equal(VisibilityValueKeyword("[a b c]")))
|
|
})
|
|
|
|
// Time
|
|
t.Run("Time", func(t *testing.T) {
|
|
// Use a fixed UTC time for deterministic comparison
|
|
base := time.Date(2025, 9, 28, 12, 34, 56, 789000000, time.UTC)
|
|
v := VisibilityValueTime(base)
|
|
p := v.MustEncode()
|
|
require.NotNil(t, p)
|
|
|
|
var out time.Time
|
|
err := payload.Decode(p, &out)
|
|
require.NoError(t, err)
|
|
require.True(t, base.Equal(out))
|
|
|
|
require.True(t, v.Equal(VisibilityValueTime(base)))
|
|
require.False(t, v.Equal(VisibilityValueTime(base.Add(time.Second))))
|
|
require.False(t, v.Equal(VisibilityValueKeyword(base.String())))
|
|
})
|
|
}
|
|
|
|
func TestIsVisibilityValueEqual(t *testing.T) {
|
|
// nil vs nil
|
|
require.True(t, isVisibilityValueEqual(nil, nil))
|
|
|
|
// one nil
|
|
require.False(t, isVisibilityValueEqual(VisibilityValueInt64(1), nil))
|
|
require.False(t, isVisibilityValueEqual(nil, VisibilityValueInt64(1)))
|
|
|
|
// equal values
|
|
require.True(t, isVisibilityValueEqual(VisibilityValueKeyword("x"), VisibilityValueKeyword("x")))
|
|
require.True(t, isVisibilityValueEqual(VisibilityValueInt64(5), VisibilityValueInt64(5)))
|
|
|
|
// not equal values
|
|
require.False(t, isVisibilityValueEqual(VisibilityValueInt64(5), VisibilityValueInt64(6)))
|
|
require.False(t, isVisibilityValueEqual(VisibilityValueInt64(5), VisibilityValueFloat64(5)))
|
|
}
|