mirror of
https://gitee.com/milvus-io/milvus.git
synced 2025-12-06 17:18:35 +08:00
0a465bb5b7
issue: #41544 - add lock interceptor into wal. - use recovery and shardmanager to replace the original implementation of segment assignment. - remove redundant implementation and unittest. - remove redundant proto definition. - use 2 streamingnode in e2e. --------- Signed-off-by: chyezh <chyezh@outlook.com>
34 lines
949 B
Go
34 lines
949 B
Go
package adaptor
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/cockroachdb/errors"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/milvus-io/milvus/internal/streamingnode/server/wal"
|
|
"github.com/milvus-io/milvus/pkg/v2/mocks/streaming/mock_walimpls"
|
|
"github.com/milvus-io/milvus/pkg/v2/streaming/walimpls"
|
|
"github.com/milvus-io/milvus/pkg/v2/util/paramtable"
|
|
)
|
|
|
|
func TestMain(m *testing.M) {
|
|
paramtable.Init()
|
|
m.Run()
|
|
}
|
|
|
|
func TestOpenerAdaptorFailure(t *testing.T) {
|
|
basicOpener := mock_walimpls.NewMockOpenerImpls(t)
|
|
errExpected := errors.New("test")
|
|
basicOpener.EXPECT().Open(mock.Anything, mock.Anything).RunAndReturn(func(ctx context.Context, boo *walimpls.OpenOption) (walimpls.WALImpls, error) {
|
|
return nil, errExpected
|
|
})
|
|
|
|
opener := adaptImplsToOpener(basicOpener, nil)
|
|
l, err := opener.Open(context.Background(), &wal.OpenOption{})
|
|
assert.ErrorIs(t, err, errExpected)
|
|
assert.Nil(t, l)
|
|
}
|