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
259
api/adapter/subscriber_test.go
Normal file
259
api/adapter/subscriber_test.go
Normal file
@@ -0,0 +1,259 @@
|
||||
package adapter_test
|
||||
|
||||
import (
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"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 TestNewSubscriberWithPulsarClient(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)
|
||||
assert.NotNil(t, sub)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe(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()
|
||||
msgChan, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_WithContextValues(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()
|
||||
|
||||
msgChan, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan)
|
||||
}
|
||||
|
||||
func TestSubscriber_Close(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)
|
||||
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestSubscriber_Close_AfterSubscribe(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, "test-topic")
|
||||
require.NoError(t, err)
|
||||
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_MultipleTopics(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()
|
||||
msgChan1, err := sub.Subscribe(ctx, "topic-1")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan1)
|
||||
|
||||
msgChan2, err := sub.Subscribe(ctx, "topic-2")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan2)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_ReuseConsumer(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()
|
||||
msgChan1, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
|
||||
msgChan2, err := sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
|
||||
assert.NotNil(t, msgChan1)
|
||||
assert.NotNil(t, msgChan2)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_ContextCancellation(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, cancel := context.WithCancel(context.Background())
|
||||
_, err = sub.Subscribe(ctx, "test-topic")
|
||||
require.NoError(t, err)
|
||||
|
||||
// Cancel context
|
||||
cancel()
|
||||
|
||||
// Wait a bit for goroutine to process cancellation
|
||||
time.Sleep(100 * time.Millisecond)
|
||||
|
||||
// Close subscriber
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_EmptyTopic(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()
|
||||
msgChan, err := sub.Subscribe(ctx, "")
|
||||
require.NoError(t, err)
|
||||
assert.NotNil(t, msgChan)
|
||||
}
|
||||
|
||||
func TestSubscriber_Close_MultipleTimes(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)
|
||||
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
|
||||
// Close again should be safe
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestSubscriber_Subscribe_AfterClose(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)
|
||||
|
||||
err = sub.Close()
|
||||
require.NoError(t, err)
|
||||
|
||||
ctx := context.Background()
|
||||
_, err = sub.Subscribe(ctx, "test-topic")
|
||||
// Behavior depends on implementation - may succeed or fail
|
||||
_ = err
|
||||
}
|
||||
|
||||
func TestNewSubscriber_InvalidURL(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
config := adapter.SubscriberConfig{
|
||||
URL: "invalid-url",
|
||||
SubscriberName: "test-sub",
|
||||
SubscriberType: pulsar.Shared,
|
||||
}
|
||||
log := logger.NewNopLogger()
|
||||
|
||||
_, err := adapter.NewSubscriber(config, log)
|
||||
require.Error(t, err)
|
||||
assert.Contains(t, err.Error(), "cannot connect")
|
||||
}
|
||||
Reference in New Issue
Block a user