Function TestSharedState_PreforkSafeWithSharedStorage

shared_state_test.go:165–212

(t *testing.T)

}

func TestSharedState_PreforkSafeWithSharedStorage(t *testing.T) {
	t.Parallel()

	store := newSharedStateMemoryStorage(t)
	workerA := New(Config{AppName: "prefork-app", SharedStorage: store})
	workerB := New(Config{AppName: "prefork-app", SharedStorage: store})

	workerA.State().Set("process-only", "from-worker-a")
	_, ok := workerB.State().Get("process-only")
	require.False(t, ok)

	payload := Map{"worker": "a", "version": 3}
	err := workerA.SharedState().SetJSON("cluster-key", payload, time.Minute)
	require.NoError(t, err)

	err = workerA.SharedState().Set("raw-cluster-key", []byte("raw-value"), time.Minute)
	require.NoError(t, err)

	var out map[string]any
	rawJSON, found, err := workerB.SharedState().GetJSON("cluster-key", &out)
	require.NoError(t, err)
	require.True(t, found)
	require.NotNil(t, rawJSON)
	require.Equal(t, "a", out["worker"])
	require.EqualValues(t, 3, out["version"])

	raw, found, err := workerB.SharedState().Get("raw-cluster-key")
	require.NoError(t, err)
	require.True(t, found)
	require.Equal(t, []byte("raw-value"), raw)

	has, err := workerB.SharedState().Has("cluster-key")
	require.NoError(t, err)
	require.True(t, has)

	err = workerB.SharedState().Delete("cluster-key")
	require.NoError(t, err)

	has, err = workerA.SharedState().Has("cluster-key")
	require.NoError(t, err)
	require.False(t, has)

	require.NoError(t, workerA.SharedState().Delete("raw-cluster-key"))
	raw, found, err = workerB.SharedState().Get("raw-cluster-key")
	require.NoError(t, err)
	require.Nil(t, raw)
	require.False(t, found)
}

func TestSharedState_ExplicitSerializationError(t *testing.T) {
	t.Parallel()