refactor: 重构trustlog-sdk目录结构到trustlog/go-trustlog
- 将所有trustlog-sdk文件移动到trustlog/go-trustlog/目录 - 更新README中所有import路径从trustlog-sdk改为go-trustlog - 更新cookiecutter配置文件中的项目名称 - 更新根目录.lefthook.yml以引用新位置的配置 - 添加go.sum文件到版本控制 - 删除过时的示例文件 这次重构与trustlog-server保持一致的目录结构， 为未来支持多语言SDK（Python、Java等）预留空间。
This commit is contained in:
ryan
216
api/adapter/subscriber_advanced_test.go
Normal file
216
api/adapter/subscriber_advanced_test.go
Normal file
@@ -0,0 +1,216 @@
|
||||
package adapter_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/ThreeDotsLabs/watermill/message"
|
||||
"github.com/apache/pulsar-client-go/pulsar"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
|
||||
"go.yandata.net/iod/iod/trustlog-sdk/api/adapter"
|
||||
"go.yandata.net/iod/iod/trustlog-sdk/api/adapter/mocks"
|
||||
"go.yandata.net/iod/iod/trustlog-sdk/api/logger"
|
||||
)
|
||||
|
||||
func TestSubscriber_Subscribe_WithAllContextValues(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
config := adapter.SubscriberConfig{
|
||||
SubscriberName: "test-sub",
|
||||
SubscriberType: pulsar.Shared,
|
||||
}
|
||||
|
||||
sub, err := adapter.NewSubscriberWithPulsarClient(mockClient, config, log)
|
||||
require.NoError(t, err)
|
||||
defer sub.Close()
|
||||
|
||||
ctx := context.WithValue(context.Background(), adapter.SubNameKey, "custom-sub")
|
||||
ctx = context.WithValue(ctx, adapter.IndexKey, 2)
|
||||
ctx = context.WithValue(ctx, adapter.ReceiverQueueSizeKey, 1500)
|
||||
|
||||
msgChan, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_ReuseExistingConsumer(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
config := adapter.SubscriberConfig{
|
||||
SubscriberName: "test-sub",
|
||||
SubscriberType: pulsar.Shared,
|
||||
}
|
||||
|
||||
sub, err := adapter.NewSubscriberWithPulsarClient(mockClient, config, log)
|
||||
require.NoError(t, err)
|
||||
defer sub.Close()
|
||||
|
||||
ctx := context.Background()
|
||||
|
||||
// Subscribe first time
|
||||
msgChan1, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan1)
|
||||
|
||||
// Wait a bit
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
|
||||
// Subscribe again with same topic - should reuse consumer
|
||||
msgChan2, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan2)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_DifferentIndices(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
config := adapter.SubscriberConfig{
|
||||
SubscriberName: "test-sub",
|
||||
SubscriberType: pulsar.Shared,
|
||||
}
|
||||
|
||||
sub, err := adapter.NewSubscriberWithPulsarClient(mockClient, config, log)
|
||||
require.NoError(t, err)
|
||||
defer sub.Close()
|
||||
|
||||
ctx1 := context.WithValue(context.Background(), adapter.IndexKey, 0)
|
||||
msgChan1, err := sub.Subscribe(ctx1, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan1)
|
||||
|
||||
ctx2 := context.WithValue(context.Background(), adapter.IndexKey, 1)
|
||||
msgChan2, err := sub.Subscribe(ctx2, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan2)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_WithoutSubscriberName(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
config := adapter.SubscriberConfig{
|
||||
SubscriberType: pulsar.Shared,
|
||||
}
|
||||
|
||||
sub, err := adapter.NewSubscriberWithPulsarClient(mockClient, config, log)
|
||||
require.NoError(t, err)
|
||||
defer sub.Close()
|
||||
|
||||
ctx := context.Background()
|
||||
msgChan, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan)
|
||||
}
|
||||
|
||||
func TestSubscriber_Close_WithMultipleSubscriptions(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
config := adapter.SubscriberConfig{
|
||||
SubscriberName: "test-sub",
|
||||
SubscriberType: pulsar.Shared,
|
||||
}
|
||||
|
||||
sub, err := adapter.NewSubscriberWithPulsarClient(mockClient, config, log)
|
||||
require.NoError(t, err)
|
||||
|
||||
ctx := context.Background()
|
||||
_, err = sub.Subscribe(ctx, "topic-1")
|
||||
require.NoError(t, err)
|
||||
|
||||
_, err = sub.Subscribe(ctx, "topic-2")
|
||||
require.NoError(t, err)
|
||||
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestPublisher_Publish_EmptyMessages(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
|
||||
pub, err := adapter.NewPublisherWithPulsarClient(mockClient, log)
|
||||
require.NoError(t, err)
|
||||
defer pub.Close()
|
||||
|
||||
// Publish with no messages - should succeed
|
||||
err = pub.Publish("test-topic")
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestPublisher_Publish_MultipleMessagesSameTopic(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
|
||||
pub, err := adapter.NewPublisherWithPulsarClient(mockClient, log)
|
||||
require.NoError(t, err)
|
||||
defer pub.Close()
|
||||
|
||||
msg1 := message.NewMessage("uuid-1", []byte("payload-1"))
|
||||
msg2 := message.NewMessage("uuid-2", []byte("payload-2"))
|
||||
msg3 := message.NewMessage("uuid-3", []byte("payload-3"))
|
||||
|
||||
err = pub.Publish("test-topic", msg1, msg2, msg3)
|
||||
require.NoError(t, err)
|
||||
|
||||
producer := mockClient.GetProducer("test-topic")
|
||||
require.NotNil(t, producer)
|
||||
messages := producer.GetMessages()
|
||||
require.Len(t, messages, 3)
|
||||
}
|
||||
|
||||
func TestPublisher_Close_WithMultipleProducers(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
|
||||
pub, err := adapter.NewPublisherWithPulsarClient(mockClient, log)
|
||||
require.NoError(t, err)
|
||||
|
||||
msg1 := message.NewMessage("uuid-1", []byte("payload-1"))
|
||||
msg2 := message.NewMessage("uuid-2", []byte("payload-2"))
|
||||
|
||||
err = pub.Publish("topic-1", msg1)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = pub.Publish("topic-2", msg2)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = pub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestPublisher_Close_MultipleTimes(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
mockClient := mocks.NewMockPulsarClient()
|
||||
log := logger.NewNopLogger()
|
||||
|
||||
pub, err := adapter.NewPublisherWithPulsarClient(mockClient, log)
|
||||
require.NoError(t, err)
|
||||
|
||||
err = pub.Close()
|
||||
require.NoError(t, err)
|
||||
|
||||
// Close again should be safe
|
||||
err = pub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
Reference in New Issue
Block a user