mirror of
https://github.com/logos-messaging/sds-go-bindings.git
synced 2026-01-02 14:13:08 +00:00
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package sds
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCreateAndCleanup(t *testing.T) {
|
|
|
|
rm1, err := NewReliabilityManager("my-channel-id-1")
|
|
require.NoError(t, err)
|
|
|
|
err = rm1.Cleanup()
|
|
require.NoError(t, err)
|
|
|
|
}
|
|
|
|
func TestReset(t *testing.T) {
|
|
|
|
rm, err := NewReliabilityManager("my-channel-id")
|
|
require.NoError(t, err)
|
|
|
|
err = rm.Reset()
|
|
require.NoError(t, err)
|
|
|
|
err = rm.Cleanup()
|
|
require.NoError(t, err)
|
|
|
|
}
|
|
|
|
// Test wrapping and unwrapping a simple message
|
|
func TestWrapUnwrap(t *testing.T) {
|
|
channelID := "test-wrap-unwrap"
|
|
rm, err := NewReliabilityManager(channelID)
|
|
require.NoError(t, err)
|
|
defer rm.Cleanup()
|
|
|
|
originalPayload := []byte("hello reliability")
|
|
messageID := MessageID("msg-wrap-1")
|
|
|
|
wrappedMsg, err := rm.WrapOutgoingMessage(originalPayload, messageID)
|
|
require.NoError(t, err)
|
|
|
|
require.Greater(t, len(wrappedMsg), 0, "Expected non-empty wrapped message")
|
|
|
|
// Simulate receiving the wrapped message
|
|
unwrappedMessage, err := rm.UnwrapReceivedMessage(wrappedMsg)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, string(*unwrappedMessage.Message), string(originalPayload), "Expected unwrapped and original payloads to be equal")
|
|
require.Equal(t, len(*unwrappedMessage.MissingDeps), 0, "Expexted to be no missing dependencies")
|
|
}
|