github.com/aporeto-inc/trireme-lib@v10.358.0+incompatible/mockgen.sh (about) 1 #! /bin/bash -e 2 3 GO111MODULE=on go get github.com/golang/mock/mockgen@latest 4 go get github.com/golang/mock/mockgen/model 5 go get -u golang.org/x/tools/cmd/goimports 6 7 goimport_sanitize () { 8 tmp=$(mktemp) 9 goimports "$1" > "$tmp" 10 sed $'s/^func /\/\/ nolint\\\nfunc /g' < "$tmp" | sed $'s/^type /\/\/ nolint\\\ntype /g' > "$1" 11 rm -f "$tmp" 12 } 13 14 echo "Cgnetcls Mocks" 15 mkdir -p utils/cgnetcls/mockcgnetcls 16 mockgen -source utils/cgnetcls/interfaces.go -destination utils/cgnetcls/mockcgnetcls/mockcgnetcls.go -package mockcgnetcls 17 goimport_sanitize utils/cgnetcls/mockcgnetcls/mockcgnetcls.go 18 19 echo "Controller/internal/supervisor/Provider Mocks" 20 mkdir -p controller/internal/supervisor/mocksupervisor 21 mockgen -source controller/internal/supervisor/interfaces.go -destination controller/internal/supervisor/mocksupervisor/mocksupervisor.go -package mocksupervisor 22 goimport_sanitize controller/internal/supervisor/mocksupervisor/mocksupervisor.go 23 24 echo "Enforcer Mocks" 25 mkdir -p controller/internal/enforcer/mockenforcer 26 mockgen -source controller/internal/enforcer/enforcer.go -destination controller/internal/enforcer/mockenforcer/mockenforcer.go -package mockenforcer 27 goimport_sanitize controller/internal/enforcer/mockenforcer/mockenforcer.go 28 29 echo "DNSProxy Mocks" 30 mkdir -p controller/internal/enforcer/dnsproxy/mockdnsproxy 31 mockgen -source controller/internal/enforcer/dnsproxy/dnsproxy.go -destination controller/internal/enforcer/dnsproxy/mockdnsproxy/mockdnsproxy.go -package mockdnsproxy 32 goimport_sanitize controller/internal/enforcer/dnsproxy/mockdnsproxy/mockdnsproxy.go 33 34 35 echo "Controller/Processmon Mocks" 36 mkdir -p controller/internal/processmon/mockprocessmon 37 mockgen -source controller/internal/processmon/interfaces.go -destination controller/internal/processmon/mockprocessmon/mockprocessmon.go -package mockprocessmon 38 goimport_sanitize controller/internal/processmon/mockprocessmon/mockprocessmon.go 39 40 echo "controller/pkg/remoteenforcer Mocks" 41 mkdir -p controller/pkg/remoteenforcer/mockremoteenforcer 42 mockgen -source controller/pkg/remoteenforcer/interfaces.go -destination controller/pkg/remoteenforcer/mockremoteenforcer/mockremoteenforcer.go -package mockremoteenforcer 43 goimport_sanitize controller/pkg/remoteenforcer/mockremoteenforcer/mockremoteenforcer.go 44 45 echo "controller/pkg/remoteenforcer/client Mocks" 46 mkdir -p controller/pkg/remoteenforcer/internal/client/mockclient 47 mockgen -source controller/pkg/remoteenforcer/internal/client/interfaces.go -destination controller/pkg/remoteenforcer/internal/client/mockclient/mockclient.go -package mockclient 48 goimport_sanitize controller/pkg/remoteenforcer/internal/client/mockclient/mockclient.go 49 50 echo "controller/pkg/remoteenforcer/TokenIssuer Mocks" 51 mkdir -p controller/pkg/remoteenforcer/internal/tokenissuer/mocktokenclient 52 mockgen -source controller/pkg/remoteenforcer/internal/tokenissuer/tokenissuer.go -destination controller/pkg/remoteenforcer/internal/tokenissuer/mocktokenclient/mocktokenclient.go -package mocktokenclient 53 goimport_sanitize controller/pkg/remoteenforcer/internal/tokenissuer/mocktokenclient/mocktokenclient.go 54 55 echo "controller/pkg/remoteenforcer/StatsCollector Mocks" 56 mkdir -p controller/pkg/remoteenforcer/internal/statscollector/mockstatscollector 57 mockgen \ 58 -source controller/pkg/remoteenforcer/internal/statscollector/interfaces.go \ 59 -destination controller/pkg/remoteenforcer/internal/statscollector/mockstatscollector/mockstatscollector.go \ 60 -package mockstatscollector \ 61 -aux_files collector=collector/interfaces.go \ 62 -imports statscollector=go.aporeto.io/enforcerd/trireme-lib/controller/pkg/remoteenforcer/internal/statscollector 63 goimport_sanitize controller/pkg/remoteenforcer/internal/statscollector/mockstatscollector/mockstatscollector.go 64 65 echo "controller/pkg/usertokens Mocks" 66 mkdir -p controller/pkg/usertokens/mockusertokens 67 mockgen -source controller/pkg/usertokens/usertokens.go -destination controller/pkg/usertokens/mockusertokens/mockusertokens.go -package mockusertokens 68 goimport_sanitize controller/pkg/usertokens/mockusertokens/mockusertokens.go 69 70 echo "controller/pkg/flowtracking Mocks" 71 mkdir -p controller/pkg/flowtracking/mockflowclient 72 mockgen -source controller/pkg/flowtracking/interfaces.go -destination controller/pkg/flowtracking/mockflowclient/mockflowclient.go -package mockflowclient 73 goimport_sanitize controller/pkg/flowtracking/mockflowclient/mockflowclient.go 74 75 echo "Collector Mocks" 76 mkdir -p collector/mockcollector 77 mockgen -source collector/interfaces.go -destination collector/mockcollector/mockcollector.go -package mockcollector 78 goimport_sanitize collector/mockcollector/mockcollector.go 79 80 echo "Monitor Mocks" 81 mkdir -p monitor/mockmonitor 82 mockgen -source monitor/interfaces.go -destination monitor/mockmonitor/mockmonitor.go -package mockmonitor 83 goimport_sanitize monitor/mockmonitor/mockmonitor.go 84 85 echo "Monitor remoteapi client mocks" 86 mkdir -p monitor/remoteapi/client/mockclient 87 mockgen -source monitor/remoteapi/client/interfaces.go -destination monitor/remoteapi/client/mockclient/mockclient.go -package mockclient 88 goimport_sanitize monitor/remoteapi/client/mockclient/mockclient.go 89 90 echo "Monitor/processor Mocks" 91 mkdir -p monitor/processor/mockprocessor 92 mockgen -source monitor/processor/interfaces.go -destination monitor/processor/mockprocessor/mockprocessor.go -aux_files collector=collector/interfaces.go -package mockprocessor 93 goimport_sanitize monitor/processor/mockprocessor/mockprocessor.go 94 95 echo "controller/internal/enforcer/nfqdatapath/tokenaccessor Mocks" 96 mkdir -p controller/internal/enforcer/nfqdatapath/tokenaccessor/mocktokenaccessor 97 mockgen -source controller/internal/enforcer/nfqdatapath/tokenaccessor/interfaces.go -destination controller/internal/enforcer/nfqdatapath/tokenaccessor/mocktokenaccessor/mocktokenaccessor.go -package mocktokenaccessor 98 goimport_sanitize controller/internal/enforcer/nfqdatapath/tokenaccessor/mocktokenaccessor/mocktokenaccessor.go 99 100 echo "controller/internal/enforcer/utils/ephemeralkeys Mocks" 101 mkdir -p controller/internal/enforcer/utils/ephemeralkeys/mockephemeralkeys 102 mockgen -source controller/internal/enforcer/utils/ephemeralkeys/interfaces.go -destination controller/internal/enforcer/utils/ephemeralkeys/mockephemeralkeys/mockephemeralkeys.go -package mockephemeralkeys 103 goimport_sanitize controller/internal/enforcer/utils/ephemeralkeys/mockephemeralkeys/mockephemeralkeys.go 104 105 echo "controller/pkg/tokens Mocks" 106 mkdir -p controller/pkg/tokens/mocktokens 107 mockgen -source controller/pkg/tokens/tokens.go -destination controller/pkg/tokens/mocktokens/mocktokens.go -package mocktokens 108 goimport_sanitize controller/pkg/tokens/mocktokens/mocktokens.go 109 110 echo "RPC Wrapper Mocks" 111 mkdir -p controller/internal/enforcer/utils/rpcwrapper/mockrpcwrapper 112 mockgen -source controller/internal/enforcer/utils/rpcwrapper/interfaces.go -destination controller/internal/enforcer/utils/rpcwrapper/mockrpcwrapper/mockrpcwrapper.go -package mockrpcwrapper 113 goimport_sanitize controller/internal/enforcer/utils/rpcwrapper/mockrpcwrapper/mockrpcwrapper.go 114 115 echo "Policy Interfaces Mock" 116 mkdir -p policy/mockpolicy 117 mockgen -source policy/interfaces.go -destination policy/mockpolicy/mockpolicy.go -package mockpolicy 118 goimport_sanitize policy/mockpolicy/mockpolicy.go 119 120 echo "Trireme Controller Mock" 121 mkdir -p controller/mockcontroller 122 mockgen -source controller/interfaces.go -destination controller/mockcontroller/mocktrireme.go -package mockcontroller -aux_files constants=controller/constants/constants.go events=common/events.go policy=policy/interfaces.go processor=monitor/processor/interfaces.go supervisor=controller/internal/supervisor/interfaces.go 123 goimport_sanitize controller/mockcontroller/mocktrireme.go 124 125 echo "Pod Monitor Mocks (manager, client and zap core)" 126 # NOTE: this uses interface mode because these are all 3rd party dependencies 127 mockgen -package podmonitor -destination monitor/internal/pod/mockzapcore_test.go go.uber.org/zap/zapcore Core 128 goimport_sanitize monitor/internal/pod/mockzapcore_test.go 129 mockgen -package podmonitor -destination monitor/internal/pod/mockclient_test.go sigs.k8s.io/controller-runtime/pkg/client Client 130 goimport_sanitize monitor/internal/pod/mockclient_test.go 131 mockgen -package podmonitor -destination monitor/internal/pod/mockcache_test.go sigs.k8s.io/controller-runtime/pkg/cache Cache 132 goimport_sanitize monitor/internal/pod/mockcache_test.go 133 mockgen -package podmonitor -destination monitor/internal/pod/mockinformer_test.go sigs.k8s.io/controller-runtime/pkg/cache Informer 134 goimport_sanitize monitor/internal/pod/mockinformer_test.go 135 # mockgen -package podmonitor -destination monitor/internal/pod/mockinformer_test.go k8s.io/client-go/tools/cache SharedIndexInformer 136 # goimport_sanitize monitor/internal/pod/mockinformer_test.go 137 mockgen -package podmonitor -destination monitor/internal/pod/mockmanager_test.go sigs.k8s.io/controller-runtime/pkg/manager Manager 138 goimport_sanitize monitor/internal/pod/mockmanager_test.go 139 140 echo >&2 "OK"