2025-04-10 14:11:29 +03:00
|
|
|
package sds
|
|
|
|
|
|
|
|
|
|
import (
|
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
)
|
|
|
|
|
|
2025-04-15 17:12:39 +03:00
|
|
|
// Test basic creation, cleanup, and reset
|
|
|
|
|
func TestLifecycle(t *testing.T) {
|
|
|
|
|
channelID := "test-lifecycle"
|
|
|
|
|
rm, err := NewReliabilityManager(channelID)
|
2025-04-10 14:11:29 +03:00
|
|
|
require.NoError(t, err)
|
2025-04-15 17:12:39 +03:00
|
|
|
require.NotNil(t, rm, "Expected ReliabilityManager to be not nil")
|
2025-04-10 14:11:29 +03:00
|
|
|
|
2025-04-15 17:12:39 +03:00
|
|
|
defer rm.Cleanup() // Ensure cleanup even on test failure
|
2025-04-10 14:33:17 +03:00
|
|
|
|
|
|
|
|
err = rm.Reset()
|
|
|
|
|
require.NoError(t, err)
|
|
|
|
|
}
|
2025-04-15 13:13:56 +03:00
|
|
|
|
2025-04-15 17:07:47 +03:00
|
|
|
// Test wrapping and unwrapping a simple message
|
|
|
|
|
func TestWrapUnwrap(t *testing.T) {
|
|
|
|
|
channelID := "test-wrap-unwrap"
|
|
|
|
|
rm, err := NewReliabilityManager(channelID)
|
2025-04-15 13:13:56 +03:00
|
|
|
require.NoError(t, err)
|
|
|
|
|
defer rm.Cleanup()
|
|
|
|
|
|
2025-04-15 17:07:47 +03:00
|
|
|
originalPayload := []byte("hello reliability")
|
|
|
|
|
messageID := MessageID("msg-wrap-1")
|
2025-04-15 13:13:56 +03:00
|
|
|
|
2025-04-15 17:07:47 +03:00
|
|
|
wrappedMsg, err := rm.WrapOutgoingMessage(originalPayload, messageID)
|
2025-04-15 13:13:56 +03:00
|
|
|
require.NoError(t, err)
|
|
|
|
|
|
2025-04-15 17:07:47 +03:00
|
|
|
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)
|
2025-04-15 13:13:56 +03:00
|
|
|
|
2025-04-15 17:07:47 +03:00
|
|
|
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")
|
2025-04-15 13:13:56 +03:00
|
|
|
}
|