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 }