github.com/getgauge/gauge@v1.6.9/api/lang/mockClient.go (about)

     1  /*----------------------------------------------------------------
     2   *  Copyright (c) ThoughtWorks, Inc.
     3   *  Licensed under the Apache License, Version 2.0
     4   *  See LICENSE in the project root for license information.
     5   *----------------------------------------------------------------*/
     6  
     7  package lang
     8  
     9  import (
    10  	"context"
    11  
    12  	gm "github.com/getgauge/gauge-proto/go/gauge_messages"
    13  	"google.golang.org/grpc"
    14  )
    15  
    16  type mockClient struct {
    17  	responses map[gm.Message_MessageType]interface{}
    18  	err       error
    19  }
    20  
    21  func (r *mockClient) GetStepNames(ctx context.Context, in *gm.StepNamesRequest, opts ...grpc.CallOption) (*gm.StepNamesResponse, error) {
    22  	return r.responses[gm.Message_StepNamesResponse].(*gm.StepNamesResponse), r.err
    23  }
    24  func (r *mockClient) CacheFile(ctx context.Context, in *gm.CacheFileRequest, opts ...grpc.CallOption) (*gm.Empty, error) {
    25  	return &gm.Empty{}, r.err
    26  }
    27  func (r *mockClient) GetStepPositions(ctx context.Context, in *gm.StepPositionsRequest, opts ...grpc.CallOption) (*gm.StepPositionsResponse, error) {
    28  	return r.responses[gm.Message_StepPositionsResponse].(*gm.StepPositionsResponse), r.err
    29  }
    30  func (r *mockClient) GetImplementationFiles(ctx context.Context, in *gm.Empty, opts ...grpc.CallOption) (*gm.ImplementationFileListResponse, error) {
    31  	return r.responses[gm.Message_ImplementationFileListResponse].(*gm.ImplementationFileListResponse), r.err
    32  }
    33  func (r *mockClient) ImplementStub(ctx context.Context, in *gm.StubImplementationCodeRequest, opts ...grpc.CallOption) (*gm.FileDiff, error) {
    34  	return r.responses[gm.Message_FileDiff].(*gm.FileDiff), r.err
    35  }
    36  
    37  func (r *mockClient) ValidateStep(ctx context.Context, in *gm.StepValidateRequest, opts ...grpc.CallOption) (*gm.StepValidateResponse, error) {
    38  	return r.responses[gm.Message_StepValidateResponse].(*gm.StepValidateResponse), r.err
    39  }
    40  func (r *mockClient) Refactor(ctx context.Context, in *gm.RefactorRequest, opts ...grpc.CallOption) (*gm.RefactorResponse, error) {
    41  	return r.responses[gm.Message_RefactorResponse].(*gm.RefactorResponse), r.err
    42  }
    43  func (r *mockClient) GetStepName(ctx context.Context, in *gm.StepNameRequest, opts ...grpc.CallOption) (*gm.StepNameResponse, error) {
    44  	return r.responses[gm.Message_StepNameResponse].(*gm.StepNameResponse), r.err
    45  }
    46  
    47  func (r *mockClient) GetGlobPatterns(ctx context.Context, in *gm.Empty, opts ...grpc.CallOption) (*gm.ImplementationFileGlobPatternResponse, error) {
    48  	return r.responses[gm.Message_ImplementationFileGlobPatternResponse].(*gm.ImplementationFileGlobPatternResponse), r.err
    49  }
    50  
    51  func (r *mockClient) KillProcess(ctx context.Context, in *gm.KillProcessRequest, opts ...grpc.CallOption) (*gm.Empty, error) {
    52  	return nil, nil
    53  }