github.com/voedger/voedger@v0.0.0-20240520144910-273e84102129/pkg/vit/mock/impl.go (about)

     1  /*
     2   * Copyright (c) 2022-present unTill Pro, Ltd.
     3   */
     4  
     5  package mock
     6  
     7  import (
     8  	"github.com/stretchr/testify/mock"
     9  
    10  	"github.com/voedger/voedger/pkg/appdef"
    11  	"github.com/voedger/voedger/pkg/istructs"
    12  )
    13  
    14  type CUDRow struct {
    15  	istructs.ICUDRow
    16  	mock.Mock
    17  }
    18  
    19  func (r *CUDRow) AsString(name string) string { return r.Called(name).String(0) }
    20  func (r *CUDRow) AsBool(name string) bool     { return r.Called(name).Bool(0) }
    21  func (r *CUDRow) QName() appdef.QName         { return r.Called().Get(0).(appdef.QName) }
    22  func (r *CUDRow) ID() istructs.RecordID       { return r.Called().Get(0).(istructs.RecordID) }
    23  
    24  type PLogEvent struct {
    25  	istructs.IPLogEvent
    26  	mock.Mock
    27  }
    28  
    29  func (e *PLogEvent) ArgumentObject() istructs.IObject { return e.Called().Get(0).(istructs.IObject) }
    30  func (e *PLogEvent) CUDs(cb func(istructs.ICUDRow))   { e.Called(cb) }
    31  func (e *PLogEvent) Workspace() istructs.WSID         { return e.Called().Get(0).(istructs.WSID) }
    32  
    33  type State struct {
    34  	istructs.IState
    35  	mock.Mock
    36  }
    37  
    38  func (s *State) KeyBuilder(storage, entity appdef.QName) (builder istructs.IStateKeyBuilder, err error) {
    39  	args := s.Called(storage, entity)
    40  	if intf := args.Get(0); intf != nil {
    41  		builder = intf.(istructs.IStateKeyBuilder)
    42  	}
    43  	err = args.Error(1)
    44  	return
    45  }
    46  func (s *State) MustExist(key istructs.IStateKeyBuilder) (value istructs.IStateValue, err error) {
    47  	args := s.Called(key)
    48  	if intf := args.Get(0); intf != nil {
    49  		value = intf.(istructs.IStateValue)
    50  	}
    51  	err = args.Error(1)
    52  	return
    53  }
    54  
    55  type Intents struct {
    56  	istructs.IIntents
    57  	mock.Mock
    58  }
    59  
    60  func (i *Intents) NewValue(key istructs.IStateKeyBuilder) (builder istructs.IStateValueBuilder, err error) {
    61  	args := i.Called(key)
    62  	if intf := args.Get(0); intf != nil {
    63  		builder = intf.(istructs.IStateValueBuilder)
    64  	}
    65  	err = args.Error(1)
    66  	return
    67  }
    68  
    69  type StateKeyBuilder struct {
    70  	istructs.IStateKeyBuilder
    71  	mock.Mock
    72  }
    73  
    74  func (b *StateKeyBuilder) PutInt32(name string, value int32)        { b.Called(name, value) }
    75  func (b *StateKeyBuilder) PutInt64(name string, value int64)        { b.Called(name, value) }
    76  func (b *StateKeyBuilder) PutString(name, value string)             { b.Called(name, value) }
    77  func (b *StateKeyBuilder) PutQName(name string, value appdef.QName) { b.Called(name, value) }
    78  
    79  type StateValueBuilder struct {
    80  	istructs.IStateValueBuilder
    81  	mock.Mock
    82  }
    83  
    84  type StateValue struct {
    85  	istructs.IStateValue
    86  	mock.Mock
    87  }
    88  
    89  func (v *StateValue) AsInt32(name string) int32   { return v.Called(name).Get(0).(int32) }
    90  func (v *StateValue) AsInt64(name string) int64   { return v.Called(name).Get(0).(int64) }
    91  func (v *StateValue) AsString(name string) string { return v.Called(name).String(0) }
    92  
    93  type AppStructsProvider struct {
    94  	mock.Mock
    95  }
    96  
    97  func (p *AppStructsProvider) AppStructs(aqn istructs.AppQName) (structs istructs.IAppStructs, err error) {
    98  	args := p.Called(aqn)
    99  	if intf := args.Get(0); intf != nil {
   100  		structs = intf.(istructs.IAppStructs)
   101  	}
   102  	err = args.Error(1)
   103  	return
   104  }
   105  
   106  type AppStructs struct {
   107  	istructs.IAppStructs
   108  	mock.Mock
   109  }
   110  
   111  func (s *AppStructs) Records() istructs.IRecords { return s.Called().Get(0).(istructs.IRecords) }
   112  
   113  type Records struct {
   114  	istructs.IRecords
   115  	mock.Mock
   116  }
   117  
   118  func (r *Records) GetSingleton(workspace istructs.WSID, qName appdef.QName) (record istructs.IRecord, err error) {
   119  	args := r.Called(workspace, qName)
   120  	if intf := args.Get(0); intf != nil {
   121  		record = intf.(istructs.IRecord)
   122  	}
   123  	err = args.Error(1)
   124  	return
   125  }
   126  
   127  type Object struct {
   128  	istructs.IObject
   129  	mock.Mock
   130  }
   131  
   132  func (o *Object) AsInt32(name string) int32                            { return o.Called(name).Get(0).(int32) }
   133  func (o *Object) AsInt64(name string) int64                            { return o.Called(name).Get(0).(int64) }
   134  func (o *Object) AsString(name string) string                          { return o.Called(name).String(0) }
   135  func (o *Object) Children(container string, cb func(istructs.IObject)) { o.Called(container, cb) }
   136  
   137  type Record struct {
   138  	istructs.IRecord
   139  	mock.Mock
   140  }
   141  
   142  func (r *Record) AsString(name string) string { return r.Called(name).String(0) }
   143  func (r *Record) AsInt64(name string) int64   { return r.Called(name).Get(0).(int64) }
   144  func (r *Record) AsRecordID(name string) istructs.RecordID {
   145  	return r.Called(name).Get(0).(istructs.RecordID)
   146  }