github.com/ydb-platform/ydb-go-sdk/v3@v3.89.2/internal/query/session_fixtures_test.go (about)

     1  package query
     2  
     3  import (
     4  	"fmt"
     5  
     6  	"github.com/rekby/fixenv"
     7  	"go.uber.org/mock/gomock"
     8  )
     9  
    10  func SessionOverGrpcMock(e fixenv.Env) *Session {
    11  	f := func() (*fixenv.GenericResult[*Session], error) {
    12  		s := newTestSession(fmt.Sprintf("test-session-id-%v", e.T().Name()))
    13  		s.client = QueryGrpcMock(e)
    14  
    15  		return fixenv.NewGenericResult(s), nil
    16  	}
    17  
    18  	return fixenv.CacheResult(e, f)
    19  }
    20  
    21  func QueryGrpcMock(e fixenv.Env) *MockQueryServiceClient {
    22  	f := func() (*fixenv.GenericResult[*MockQueryServiceClient], error) {
    23  		m := NewMockQueryServiceClient(MockController(e))
    24  
    25  		return fixenv.NewGenericResult(m), nil
    26  	}
    27  
    28  	return fixenv.CacheResult(e, f)
    29  }
    30  
    31  func MockController(e fixenv.Env) *gomock.Controller {
    32  	f := func() (*fixenv.GenericResult[*gomock.Controller], error) {
    33  		mc := gomock.NewController(e.T().(gomock.TestReporter))
    34  
    35  		return fixenv.NewGenericResult(mc), nil
    36  	}
    37  
    38  	return fixenv.CacheResult(e, f)
    39  }