From 5377dd4d7faa75adb81584b66430593f03d3067e Mon Sep 17 00:00:00 2001 From: Sammy Kerata Oina <44265300+SammyOina@users.noreply.github.com> Date: Mon, 1 Sep 2025 15:28:11 +0300 Subject: [PATCH] NOISSUE - Prepare cocos for v0.8.0 (#512) * Refactor mock interfaces to use 'any' instead of 'interface{}' for improved type safety and readability across multiple files in the manager and pkg directories. Signed-off-by: Sammy Oina * Update Go version to 1.25.x in CI workflows and remove obsolete Go package files Signed-off-by: Sammy Oina * Add mock implementations for various components in the attestation and SDK packages - Created mock for MeasurementProvider in pkg/attestation/cmdconfig/mocks/mocks_test.go - Created mock for Provider in pkg/attestation/mocks/mocks_test.go - Created mock for Client in pkg/clients/grpc/mocks/mocks_test.go - Created mock for SDK in pkg/sdk/mocks/mocks_test.go These mocks are generated using mockery and are intended for unit testing purposes. Signed-off-by: Sammy Oina * Remove autogenerated mock files and update mock usage in tests - Deleted mocks for gRPC clients in pkg/clients/grpc/mocks/mocks_test.go and pkg/sdk/mocks/mocks_test.go. - Updated test files in pkg/progressbar/progress_test.go to use the new mock structure without type parameters for gRPC client interfaces. - Refactored mock generation in pkg/sdk/mocks/sdk.go to streamline the mock creation process and ensure consistency across mock methods. Signed-off-by: Sammy Oina * Update protobuf generated files for events and manager - Bump protoc-gen-go version from v1.36.5 to v1.36.8 in events.pb.go and manager.pb.go. - Refactor raw descriptor definitions in events.pb.go and manager.pb.go to use string concatenation for better readability and maintainability. - Ensure compatibility with the latest protobuf specifications and improve code generation consistency. Signed-off-by: Sammy Oina * Update test commands to use GOTOOLCHAIN for consistent Go version handling Signed-off-by: Sammy Oina * Fix GOTOOLCHAIN usage in test command for consistency Signed-off-by: Sammy Oina --------- Signed-off-by: Sammy Oina --- .github/workflows/checkproto.yaml | 4 +- .github/workflows/hal.yml | 4 +- .github/workflows/main.yaml | 12 +- .mockery.yml | 141 ++++++ Makefile | 2 +- agent/agent.pb.go | 110 ++--- agent/algorithm/mocks/algorithm.go | 66 +-- agent/api/grpc/endpoint.go | 12 +- agent/api/grpc/interceptors.go | 4 +- agent/api/grpc/interceptors_test.go | 4 +- agent/api/grpc/server.go | 36 +- agent/auth/auth.go | 6 +- .../auth/mocks/{auth.go => authenticator.go} | 76 +-- agent/cvms/api/grpc/storage/mocks/storage.go | 126 ++--- agent/cvms/cvms.pb.go | 261 ++++------- .../mocks/{server.go => agentserver.go} | 79 ++-- agent/events/events.pb.go | 68 ++- agent/events/mocks/{events.go => service.go} | 66 ++- agent/mocks/agent_grpc_algo.go | 434 ----------------- agent/mocks/agent_grpc_data.go | 434 ----------------- agent/mocks/agent_grpc_ima.go | 388 --------------- agent/mocks/agentservice_algoclient.go | 442 ++++++++++++++++++ agent/mocks/agentservice_dataclient.go | 442 ++++++++++++++++++ .../agentservice_imameasurementsclient.go | 391 ++++++++++++++++ agent/mocks/{agent.go => service.go} | 330 +++++++------ .../mocks/{state.go => statemachine.go} | 149 +++--- cli/attestation.go | 6 +- cli/attestation_snp_test.go | 6 +- cli/attestation_test.go | 2 +- cli/datasets.go | 2 +- cli/keys.go | 2 +- cli/keys_test.go | 2 +- go.mod | 2 +- hal/linux/README.md | 2 +- hal/linux/package/agent/agent.mk | 2 +- internal/server/mocks/server.go | 66 +-- manager/attestation_policy_test.go | 6 +- manager/manager.pb.go | 112 ++--- ...vice_client.go => managerserviceclient.go} | 214 +++++---- manager/mocks/service.go | 208 +++++---- manager/qemu/mocks/persistence.go | 108 +++-- manager/qemu/vm.go | 4 +- manager/vm/mocks/provider.go | 14 +- .../{state_machine.go => statemachine.go} | 74 +-- manager/vm/mocks/vm.go | 165 +++---- manager/vm/vm.go | 4 +- mockery.yml | 145 ------ pkg/attestation/azure/snp.go | 4 +- pkg/attestation/azure/snp_test.go | 10 +- ...ent_provider.go => measurementprovider.go} | 85 ++-- .../mocks/{attestation.go => provider.go} | 177 ++++--- pkg/attestation/mocks/verifier.go | 156 ++++--- pkg/attestation/vtpm/vtpm.go | 4 +- pkg/attestation/vtpm/vtpm_test.go | 14 +- pkg/clients/grpc/mocks/client.go | 86 ++-- pkg/progressbar/progress_test.go | 6 +- pkg/progressbar/progressbar.go | 22 +- pkg/sdk/mocks/sdk.go | 287 ++++++++---- 58 files changed, 3197 insertions(+), 2887 deletions(-) create mode 100644 .mockery.yml rename agent/auth/mocks/{auth.go => authenticator.go} (61%) rename agent/cvms/server/mocks/{server.go => agentserver.go} (67%) rename agent/events/mocks/{events.go => service.go} (66%) delete mode 100644 agent/mocks/agent_grpc_algo.go delete mode 100644 agent/mocks/agent_grpc_data.go delete mode 100644 agent/mocks/agent_grpc_ima.go create mode 100644 agent/mocks/agentservice_algoclient.go create mode 100644 agent/mocks/agentservice_dataclient.go create mode 100644 agent/mocks/agentservice_imameasurementsclient.go rename agent/mocks/{agent.go => service.go} (54%) rename agent/statemachine/mocks/{state.go => statemachine.go} (68%) rename manager/mocks/{manager_service_client.go => managerserviceclient.go} (55%) rename manager/vm/mocks/{state_machine.go => statemachine.go} (70%) delete mode 100644 mockery.yml rename pkg/attestation/cmdconfig/mocks/{measurement_provider.go => measurementprovider.go} (67%) rename pkg/attestation/mocks/{attestation.go => provider.go} (56%) diff --git a/.github/workflows/checkproto.yaml b/.github/workflows/checkproto.yaml index c7c88902..bf71a5f3 100644 --- a/.github/workflows/checkproto.yaml +++ b/.github/workflows/checkproto.yaml @@ -30,12 +30,12 @@ jobs: - name: Install Go uses: actions/setup-go@v5 with: - go-version: 1.24.x + go-version: 1.25.x - name: Set up protoc run: | PROTOC_VERSION=29.0 - PROTOC_GEN_VERSION=v1.36.5 + PROTOC_GEN_VERSION=v1.36.8 PROTOC_GRPC_VERSION=v1.5.1 # Download and install protoc diff --git a/.github/workflows/hal.yml b/.github/workflows/hal.yml index 1b416643..ba5212f1 100644 --- a/.github/workflows/hal.yml +++ b/.github/workflows/hal.yml @@ -42,7 +42,7 @@ jobs: - name: Install Go uses: actions/setup-go@v5 with: - go-version: 1.24.x + go-version: 1.25.x cache-dependency-path: "go.sum" - name: Checkout cocos @@ -56,7 +56,7 @@ jobs: with: repository: "buildroot/buildroot" path: buildroot - ref: 2025.05-rc1 + ref: 2025.08-rc3 - name: Build hal run: | diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml index 2b9b5aa8..887123cb 100644 --- a/.github/workflows/main.yaml +++ b/.github/workflows/main.yaml @@ -18,12 +18,12 @@ jobs: - name: Install Go uses: actions/setup-go@v5 with: - go-version: 1.24.x + go-version: 1.25.x - name: golangci-lint - uses: golangci/golangci-lint-action@v7 + uses: golangci/golangci-lint-action@v8 with: - version: v2.0.2 + version: v2.4.0 - name: Build run: make @@ -45,7 +45,7 @@ jobs: - name: Install Go uses: actions/setup-go@v5 with: - go-version: 1.24.x + go-version: 1.25.x - name: Create coverage directory run: mkdir -p coverage @@ -53,9 +53,9 @@ jobs: - name: Run tests for ${{ matrix.module }} run: | if [[ "${{ matrix.module }}" == "manager" ]]; then - sudo go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/... + sudo GOTOOLCHAIN=go1.25.0+auto go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/... else - go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/... + GOTOOLCHAIN=go1.25.0+auto go test -v --race -covermode=atomic -coverprofile coverage/${{ matrix.module }}.out ./${{ matrix.module }}/... fi - name: Upload coverage artifact diff --git a/.mockery.yml b/.mockery.yml new file mode 100644 index 00000000..4126202f --- /dev/null +++ b/.mockery.yml @@ -0,0 +1,141 @@ +pkgname: mocks +template: testify +template-data: + boilerplate-file: ./boilerplate.txt + unroll-variadic: true +packages: + github.com/ultravioletrs/cocos/agent: + interfaces: + AgentService_AlgoClient: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + AgentService_DataClient: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + AgentService_IMAMeasurementsClient: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + Service: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/agent/algorithm: + interfaces: + Algorithm: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/agent/auth: + interfaces: + Authenticator: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/agent/cvms/api/grpc/storage: + interfaces: + Storage: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/agent/cvms/server: + interfaces: + AgentServer: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/agent/events: + interfaces: + Service: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/agent/statemachine: + interfaces: + StateMachine: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/internal/server: + interfaces: + Server: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/manager: + interfaces: + ManagerServiceClient: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + Service: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/manager/qemu: + interfaces: + Persistence: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/manager/vm: + interfaces: + StateMachine: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + VM: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/pkg/attestation: + interfaces: + Provider: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + Verifier: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/pkg/attestation/cmdconfig: + interfaces: + MeasurementProvider: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/pkg/clients/grpc: + interfaces: + Client: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" + github.com/ultravioletrs/cocos/pkg/sdk: + interfaces: + SDK: + config: + dir: '{{.InterfaceDir}}/mocks' + structname: '{{.InterfaceName}}' + filename: "{{.InterfaceName | lower}}.go" diff --git a/Makefile b/Makefile index d4797795..ec8cc373 100644 --- a/Makefile +++ b/Makefile @@ -42,7 +42,7 @@ protoc: protoc -I. --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative agent/cvms/cvms.proto mocks: - mockery --config ./mockery.yml + mockery --config ./.mockery.yml install: $(SERVICES) $(ATTESTATION_POLICY) install -d $(INSTALL_DIR) diff --git a/agent/agent.pb.go b/agent/agent.pb.go index a70a1392..6b822f8f 100644 --- a/agent/agent.pb.go +++ b/agent/agent.pb.go @@ -3,7 +3,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.36.5 +// protoc-gen-go v1.36.8 // protoc v5.29.0 // source: agent/agent.proto @@ -570,76 +570,44 @@ func (x *AttestationTokenResponse) GetFile() []byte { var File_agent_agent_proto protoreflect.FileDescriptor -var file_agent_agent_proto_rawDesc = string([]byte{ - 0x0a, 0x11, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2f, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x70, 0x72, - 0x6f, 0x74, 0x6f, 0x12, 0x05, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x22, 0x4f, 0x0a, 0x0b, 0x41, 0x6c, - 0x67, 0x6f, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x1c, 0x0a, 0x09, 0x61, 0x6c, 0x67, - 0x6f, 0x72, 0x69, 0x74, 0x68, 0x6d, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x09, 0x61, 0x6c, - 0x67, 0x6f, 0x72, 0x69, 0x74, 0x68, 0x6d, 0x12, 0x22, 0x0a, 0x0c, 0x72, 0x65, 0x71, 0x75, 0x69, - 0x72, 0x65, 0x6d, 0x65, 0x6e, 0x74, 0x73, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x0c, 0x72, - 0x65, 0x71, 0x75, 0x69, 0x72, 0x65, 0x6d, 0x65, 0x6e, 0x74, 0x73, 0x22, 0x0e, 0x0a, 0x0c, 0x41, - 0x6c, 0x67, 0x6f, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x43, 0x0a, 0x0b, 0x44, - 0x61, 0x74, 0x61, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x18, 0x0a, 0x07, 0x64, 0x61, - 0x74, 0x61, 0x73, 0x65, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x64, 0x61, 0x74, - 0x61, 0x73, 0x65, 0x74, 0x12, 0x1a, 0x0a, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x6e, 0x61, 0x6d, 0x65, - 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x6e, 0x61, 0x6d, 0x65, - 0x22, 0x0e, 0x0a, 0x0c, 0x44, 0x61, 0x74, 0x61, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, - 0x22, 0x0f, 0x0a, 0x0d, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, - 0x74, 0x22, 0x24, 0x0a, 0x0e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x52, 0x65, 0x73, 0x70, 0x6f, - 0x6e, 0x73, 0x65, 0x12, 0x12, 0x0a, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, - 0x0c, 0x52, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x22, 0x62, 0x0a, 0x12, 0x41, 0x74, 0x74, 0x65, 0x73, - 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x12, 0x1a, 0x0a, - 0x08, 0x74, 0x65, 0x65, 0x4e, 0x6f, 0x6e, 0x63, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, - 0x08, 0x74, 0x65, 0x65, 0x4e, 0x6f, 0x6e, 0x63, 0x65, 0x12, 0x1c, 0x0a, 0x09, 0x76, 0x74, 0x70, - 0x6d, 0x4e, 0x6f, 0x6e, 0x63, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x09, 0x76, 0x74, - 0x70, 0x6d, 0x4e, 0x6f, 0x6e, 0x63, 0x65, 0x12, 0x12, 0x0a, 0x04, 0x74, 0x79, 0x70, 0x65, 0x18, - 0x03, 0x20, 0x01, 0x28, 0x05, 0x52, 0x04, 0x74, 0x79, 0x70, 0x65, 0x22, 0x29, 0x0a, 0x13, 0x41, - 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, - 0x73, 0x65, 0x12, 0x12, 0x0a, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, - 0x52, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x22, 0x18, 0x0a, 0x16, 0x49, 0x4d, 0x41, 0x4d, 0x65, 0x61, - 0x73, 0x75, 0x72, 0x65, 0x6d, 0x65, 0x6e, 0x74, 0x73, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, - 0x22, 0x43, 0x0a, 0x17, 0x49, 0x4d, 0x41, 0x4d, 0x65, 0x61, 0x73, 0x75, 0x72, 0x65, 0x6d, 0x65, - 0x6e, 0x74, 0x73, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x12, 0x0a, 0x04, 0x66, - 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x12, - 0x14, 0x0a, 0x05, 0x70, 0x63, 0x72, 0x31, 0x30, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x05, - 0x70, 0x63, 0x72, 0x31, 0x30, 0x22, 0x4d, 0x0a, 0x17, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, - 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, - 0x12, 0x1e, 0x0a, 0x0a, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x4e, 0x6f, 0x6e, 0x63, 0x65, 0x18, 0x01, - 0x20, 0x01, 0x28, 0x0c, 0x52, 0x0a, 0x74, 0x6f, 0x6b, 0x65, 0x6e, 0x4e, 0x6f, 0x6e, 0x63, 0x65, - 0x12, 0x12, 0x0a, 0x04, 0x74, 0x79, 0x70, 0x65, 0x18, 0x03, 0x20, 0x01, 0x28, 0x05, 0x52, 0x04, - 0x74, 0x79, 0x70, 0x65, 0x22, 0x2e, 0x0a, 0x18, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, - 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, - 0x12, 0x12, 0x0a, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, - 0x66, 0x69, 0x6c, 0x65, 0x32, 0xaf, 0x03, 0x0a, 0x0c, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x65, - 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x33, 0x0a, 0x04, 0x41, 0x6c, 0x67, 0x6f, 0x12, 0x12, 0x2e, - 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x41, 0x6c, 0x67, 0x6f, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, - 0x74, 0x1a, 0x13, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x41, 0x6c, 0x67, 0x6f, 0x52, 0x65, - 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x28, 0x01, 0x12, 0x33, 0x0a, 0x04, 0x44, 0x61, - 0x74, 0x61, 0x12, 0x12, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x44, 0x61, 0x74, 0x61, 0x52, - 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x13, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x44, - 0x61, 0x74, 0x61, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x28, 0x01, 0x12, - 0x39, 0x0a, 0x06, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x12, 0x14, 0x2e, 0x61, 0x67, 0x65, 0x6e, - 0x74, 0x2e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x52, 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, - 0x15, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x52, 0x65, - 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x30, 0x01, 0x12, 0x48, 0x0a, 0x0b, 0x41, 0x74, - 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x19, 0x2e, 0x61, 0x67, 0x65, 0x6e, - 0x74, 0x2e, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x71, - 0x75, 0x65, 0x73, 0x74, 0x1a, 0x1a, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x41, 0x74, 0x74, - 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, - 0x22, 0x00, 0x30, 0x01, 0x12, 0x54, 0x0a, 0x0f, 0x49, 0x4d, 0x41, 0x4d, 0x65, 0x61, 0x73, 0x75, - 0x72, 0x65, 0x6d, 0x65, 0x6e, 0x74, 0x73, 0x12, 0x1d, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, - 0x49, 0x4d, 0x41, 0x4d, 0x65, 0x61, 0x73, 0x75, 0x72, 0x65, 0x6d, 0x65, 0x6e, 0x74, 0x73, 0x52, - 0x65, 0x71, 0x75, 0x65, 0x73, 0x74, 0x1a, 0x1e, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x49, - 0x4d, 0x41, 0x4d, 0x65, 0x61, 0x73, 0x75, 0x72, 0x65, 0x6d, 0x65, 0x6e, 0x74, 0x73, 0x52, 0x65, - 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x30, 0x01, 0x12, 0x5a, 0x0a, 0x15, 0x41, 0x7a, - 0x75, 0x72, 0x65, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, - 0x6b, 0x65, 0x6e, 0x12, 0x1e, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x41, 0x74, 0x74, 0x65, - 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x52, 0x65, 0x71, 0x75, - 0x65, 0x73, 0x74, 0x1a, 0x1f, 0x2e, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2e, 0x41, 0x74, 0x74, 0x65, - 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x52, 0x65, 0x73, 0x70, - 0x6f, 0x6e, 0x73, 0x65, 0x22, 0x00, 0x42, 0x09, 0x5a, 0x07, 0x2e, 0x2f, 0x61, 0x67, 0x65, 0x6e, - 0x74, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, -}) +const file_agent_agent_proto_rawDesc = "" + + "\n" + + "\x11agent/agent.proto\x12\x05agent\"O\n" + + "\vAlgoRequest\x12\x1c\n" + + "\talgorithm\x18\x01 \x01(\fR\talgorithm\x12\"\n" + + "\frequirements\x18\x02 \x01(\fR\frequirements\"\x0e\n" + + "\fAlgoResponse\"C\n" + + "\vDataRequest\x12\x18\n" + + "\adataset\x18\x01 \x01(\fR\adataset\x12\x1a\n" + + "\bfilename\x18\x02 \x01(\tR\bfilename\"\x0e\n" + + "\fDataResponse\"\x0f\n" + + "\rResultRequest\"$\n" + + "\x0eResultResponse\x12\x12\n" + + "\x04file\x18\x01 \x01(\fR\x04file\"b\n" + + "\x12AttestationRequest\x12\x1a\n" + + "\bteeNonce\x18\x01 \x01(\fR\bteeNonce\x12\x1c\n" + + "\tvtpmNonce\x18\x02 \x01(\fR\tvtpmNonce\x12\x12\n" + + "\x04type\x18\x03 \x01(\x05R\x04type\")\n" + + "\x13AttestationResponse\x12\x12\n" + + "\x04file\x18\x01 \x01(\fR\x04file\"\x18\n" + + "\x16IMAMeasurementsRequest\"C\n" + + "\x17IMAMeasurementsResponse\x12\x12\n" + + "\x04file\x18\x01 \x01(\fR\x04file\x12\x14\n" + + "\x05pcr10\x18\x02 \x01(\fR\x05pcr10\"M\n" + + "\x17AttestationTokenRequest\x12\x1e\n" + + "\n" + + "tokenNonce\x18\x01 \x01(\fR\n" + + "tokenNonce\x12\x12\n" + + "\x04type\x18\x03 \x01(\x05R\x04type\".\n" + + "\x18AttestationTokenResponse\x12\x12\n" + + "\x04file\x18\x01 \x01(\fR\x04file2\xaf\x03\n" + + "\fAgentService\x123\n" + + "\x04Algo\x12\x12.agent.AlgoRequest\x1a\x13.agent.AlgoResponse\"\x00(\x01\x123\n" + + "\x04Data\x12\x12.agent.DataRequest\x1a\x13.agent.DataResponse\"\x00(\x01\x129\n" + + "\x06Result\x12\x14.agent.ResultRequest\x1a\x15.agent.ResultResponse\"\x000\x01\x12H\n" + + "\vAttestation\x12\x19.agent.AttestationRequest\x1a\x1a.agent.AttestationResponse\"\x000\x01\x12T\n" + + "\x0fIMAMeasurements\x12\x1d.agent.IMAMeasurementsRequest\x1a\x1e.agent.IMAMeasurementsResponse\"\x000\x01\x12Z\n" + + "\x15AzureAttestationToken\x12\x1e.agent.AttestationTokenRequest\x1a\x1f.agent.AttestationTokenResponse\"\x00B\tZ\a./agentb\x06proto3" var ( file_agent_agent_proto_rawDescOnce sync.Once diff --git a/agent/algorithm/mocks/algorithm.go b/agent/algorithm/mocks/algorithm.go index d9a3e5e9..a2947a58 100644 --- a/agent/algorithm/mocks/algorithm.go +++ b/agent/algorithm/mocks/algorithm.go @@ -1,11 +1,29 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks -import mock "github.com/stretchr/testify/mock" +import ( + mock "github.com/stretchr/testify/mock" +) + +// NewAlgorithm creates a new instance of Algorithm. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewAlgorithm(t interface { + mock.TestingT + Cleanup(func()) +}) *Algorithm { + mock := &Algorithm{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} // Algorithm is an autogenerated mock type for the Algorithm type type Algorithm struct { @@ -20,21 +38,20 @@ func (_m *Algorithm) EXPECT() *Algorithm_Expecter { return &Algorithm_Expecter{mock: &_m.Mock} } -// Run provides a mock function with no fields -func (_m *Algorithm) Run() error { - ret := _m.Called() +// Run provides a mock function for the type Algorithm +func (_mock *Algorithm) Run() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Run") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -55,8 +72,8 @@ func (_c *Algorithm_Run_Call) Run(run func()) *Algorithm_Run_Call { return _c } -func (_c *Algorithm_Run_Call) Return(_a0 error) *Algorithm_Run_Call { - _c.Call.Return(_a0) +func (_c *Algorithm_Run_Call) Return(err error) *Algorithm_Run_Call { + _c.Call.Return(err) return _c } @@ -65,21 +82,20 @@ func (_c *Algorithm_Run_Call) RunAndReturn(run func() error) *Algorithm_Run_Call return _c } -// Stop provides a mock function with no fields -func (_m *Algorithm) Stop() error { - ret := _m.Called() +// Stop provides a mock function for the type Algorithm +func (_mock *Algorithm) Stop() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Stop") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -100,8 +116,8 @@ func (_c *Algorithm_Stop_Call) Run(run func()) *Algorithm_Stop_Call { return _c } -func (_c *Algorithm_Stop_Call) Return(_a0 error) *Algorithm_Stop_Call { - _c.Call.Return(_a0) +func (_c *Algorithm_Stop_Call) Return(err error) *Algorithm_Stop_Call { + _c.Call.Return(err) return _c } @@ -109,17 +125,3 @@ func (_c *Algorithm_Stop_Call) RunAndReturn(run func() error) *Algorithm_Stop_Ca _c.Call.Return(run) return _c } - -// NewAlgorithm creates a new instance of Algorithm. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewAlgorithm(t interface { - mock.TestingT - Cleanup(func()) -}) *Algorithm { - mock := &Algorithm{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/api/grpc/endpoint.go b/agent/api/grpc/endpoint.go index 75c8e4f4..a654a8e4 100644 --- a/agent/api/grpc/endpoint.go +++ b/agent/api/grpc/endpoint.go @@ -11,7 +11,7 @@ import ( ) func algoEndpoint(svc agent.Service) endpoint.Endpoint { - return func(ctx context.Context, request interface{}) (interface{}, error) { + return func(ctx context.Context, request any) (any, error) { req := request.(algoReq) if err := req.validate(); err != nil { @@ -30,7 +30,7 @@ func algoEndpoint(svc agent.Service) endpoint.Endpoint { } func dataEndpoint(svc agent.Service) endpoint.Endpoint { - return func(ctx context.Context, request interface{}) (interface{}, error) { + return func(ctx context.Context, request any) (any, error) { req := request.(dataReq) if err := req.validate(); err != nil { @@ -49,7 +49,7 @@ func dataEndpoint(svc agent.Service) endpoint.Endpoint { } func resultEndpoint(svc agent.Service) endpoint.Endpoint { - return func(ctx context.Context, request interface{}) (interface{}, error) { + return func(ctx context.Context, request any) (any, error) { req := request.(resultReq) if err := req.validate(); err != nil { @@ -65,7 +65,7 @@ func resultEndpoint(svc agent.Service) endpoint.Endpoint { } func attestationEndpoint(svc agent.Service) endpoint.Endpoint { - return func(ctx context.Context, request interface{}) (interface{}, error) { + return func(ctx context.Context, request any) (any, error) { req := request.(attestationReq) if err := req.validate(); err != nil { @@ -81,7 +81,7 @@ func attestationEndpoint(svc agent.Service) endpoint.Endpoint { } func imaMeasurementsEndpoint(svc agent.Service) endpoint.Endpoint { - return func(ctx context.Context, request interface{}) (interface{}, error) { + return func(ctx context.Context, request any) (any, error) { req := request.(imaMeasurementsReq) if err := req.validate(); err != nil { @@ -97,7 +97,7 @@ func imaMeasurementsEndpoint(svc agent.Service) endpoint.Endpoint { } func azureAttestationTokenEndpoint(svc agent.Service) endpoint.Endpoint { - return func(ctx context.Context, request interface{}) (interface{}, error) { + return func(ctx context.Context, request any) (any, error) { req := request.(azureAttestationTokenReq) if err := req.validate(); err != nil { return fetchAttestationTokenRes{}, err diff --git a/agent/api/grpc/interceptors.go b/agent/api/grpc/interceptors.go index eb4a246d..381221f9 100644 --- a/agent/api/grpc/interceptors.go +++ b/agent/api/grpc/interceptors.go @@ -31,7 +31,7 @@ func NewAuthInterceptor(authSvc auth.Authenticator) (grpc.UnaryServerInterceptor } func (s *authInterceptor) AuthStreamInterceptor() grpc.StreamServerInterceptor { - return func(srv interface{}, stream grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error { + return func(srv any, stream grpc.ServerStream, info *grpc.StreamServerInfo, handler grpc.StreamHandler) error { switch info.FullMethod { case agent.AgentService_Algo_FullMethodName: if _, err := s.auth.AuthenticateUser(stream.Context(), auth.AlgorithmProviderRole); err != nil { @@ -59,7 +59,7 @@ func (s *authInterceptor) AuthStreamInterceptor() grpc.StreamServerInterceptor { } func (s *authInterceptor) AuthUnaryInterceptor() grpc.UnaryServerInterceptor { - return func(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (interface{}, error) { + return func(ctx context.Context, req any, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (any, error) { switch info.FullMethod { case agent.AgentService_Result_FullMethodName: ctx, err := s.auth.AuthenticateUser(ctx, auth.ConsumerRole) diff --git a/agent/api/grpc/interceptors_test.go b/agent/api/grpc/interceptors_test.go index 97092e8f..cbb65e9e 100644 --- a/agent/api/grpc/interceptors_test.go +++ b/agent/api/grpc/interceptors_test.go @@ -58,7 +58,7 @@ func TestAuthUnaryInterceptor(t *testing.T) { } unaryInt, _ := NewAuthInterceptor(authmock) - _, err := unaryInt(context.Background(), nil, &grpc.UnaryServerInfo{FullMethod: tt.method}, func(ctx context.Context, req interface{}) (interface{}, error) { + _, err := unaryInt(context.Background(), nil, &grpc.UnaryServerInfo{FullMethod: tt.method}, func(ctx context.Context, req any) (any, error) { return nil, nil }) @@ -129,7 +129,7 @@ func TestAuthStreamInterceptor(t *testing.T) { } _, streamInt := NewAuthInterceptor(authmock) - err := streamInt(nil, &mockServerStream{ctx: metadata.NewIncomingContext(context.Background(), metadata.Pairs())}, &grpc.StreamServerInfo{FullMethod: tt.method}, func(srv interface{}, stream grpc.ServerStream) error { + err := streamInt(nil, &mockServerStream{ctx: metadata.NewIncomingContext(context.Background(), metadata.Pairs())}, &grpc.StreamServerInfo{FullMethod: tt.method}, func(srv any, stream grpc.ServerStream) error { return nil }) diff --git a/agent/api/grpc/server.go b/agent/api/grpc/server.go index 6cab5ffc..9159d1fb 100644 --- a/agent/api/grpc/server.go +++ b/agent/api/grpc/server.go @@ -96,7 +96,7 @@ func NewServer(svc agent.Service) agent.AgentServiceServer { } } -func decodeAlgoRequest(_ context.Context, grpcReq interface{}) (interface{}, error) { +func decodeAlgoRequest(_ context.Context, grpcReq any) (any, error) { req := grpcReq.(*agent.AlgoRequest) return algoReq{ Algorithm: req.Algorithm, @@ -104,11 +104,11 @@ func decodeAlgoRequest(_ context.Context, grpcReq interface{}) (interface{}, err }, nil } -func encodeAlgoResponse(_ context.Context, response interface{}) (interface{}, error) { +func encodeAlgoResponse(_ context.Context, response any) (any, error) { return &agent.AlgoResponse{}, nil } -func decodeDataRequest(_ context.Context, grpcReq interface{}) (interface{}, error) { +func decodeDataRequest(_ context.Context, grpcReq any) (any, error) { req := grpcReq.(*agent.DataRequest) return dataReq{ Dataset: req.Dataset, @@ -116,22 +116,22 @@ func decodeDataRequest(_ context.Context, grpcReq interface{}) (interface{}, err }, nil } -func encodeDataResponse(_ context.Context, response interface{}) (interface{}, error) { +func encodeDataResponse(_ context.Context, response any) (any, error) { return &agent.DataResponse{}, nil } -func decodeResultRequest(_ context.Context, grpcReq interface{}) (interface{}, error) { +func decodeResultRequest(_ context.Context, grpcReq any) (any, error) { return resultReq{}, nil } -func encodeResultResponse(_ context.Context, response interface{}) (interface{}, error) { +func encodeResultResponse(_ context.Context, response any) (any, error) { res := response.(resultRes) return &agent.ResultResponse{ File: res.File, }, nil } -func validateNonce(nonce []byte, maxLen int, target interface{}) error { +func validateNonce(nonce []byte, maxLen int, target any) error { if len(nonce) > maxLen { switch maxLen { case quoteprovider.Nonce: @@ -154,7 +154,7 @@ func validateNonce(nonce []byte, maxLen int, target interface{}) error { return nil } -func decodeAttestationRequest(_ context.Context, grpcReq interface{}) (interface{}, error) { +func decodeAttestationRequest(_ context.Context, grpcReq any) (any, error) { req := grpcReq.(*agent.AttestationRequest) var reportData [quoteprovider.Nonce]byte var nonce [vtpm.Nonce]byte @@ -174,14 +174,14 @@ func decodeAttestationRequest(_ context.Context, grpcReq interface{}) (interface }, nil } -func encodeAttestationResponse(_ context.Context, response interface{}) (interface{}, error) { +func encodeAttestationResponse(_ context.Context, response any) (any, error) { res := response.(attestationRes) return &agent.AttestationResponse{ File: res.File, }, nil } -func decodeAttestationTokenRequest(_ context.Context, grpcReq interface{}) (interface{}, error) { +func decodeAttestationTokenRequest(_ context.Context, grpcReq any) (any, error) { req := grpcReq.(*agent.AttestationTokenRequest) var nonce [vtpm.Nonce]byte @@ -193,18 +193,18 @@ func decodeAttestationTokenRequest(_ context.Context, grpcReq interface{}) (inte }, nil } -func encodeAttestationTokenResponse(_ context.Context, response interface{}) (interface{}, error) { +func encodeAttestationTokenResponse(_ context.Context, response any) (any, error) { res := response.(fetchAttestationTokenRes) return &agent.AttestationTokenResponse{ File: res.File, }, nil } -func decodeIMAMeasurementsRequest(_ context.Context, grpcReq interface{}) (interface{}, error) { +func decodeIMAMeasurementsRequest(_ context.Context, grpcReq any) (any, error) { return imaMeasurementsReq{}, nil } -func encodeIMAMeasurementsResponse(_ context.Context, response interface{}) (interface{}, error) { +func encodeIMAMeasurementsResponse(_ context.Context, response any) (any, error) { res := response.(imaMeasurementsRes) return &agent.IMAMeasurementsResponse{ File: res.File, @@ -215,10 +215,10 @@ func encodeIMAMeasurementsResponse(_ context.Context, response interface{}) (int func (s *grpcServer) streamingHandler( ctx context.Context, handlerName string, - req interface{}, - stream interface{}, + req any, + stream any, sendFn func([]byte) error, - getFileData func(interface{}) []byte, + getFileData func(any) []byte, ) error { handler, ok := s.handlers[handlerName] if !ok { @@ -350,7 +350,7 @@ func (s *grpcServer) Result(req *agent.ResultRequest, stream agent.AgentService_ func(data []byte) error { return stream.Send(&agent.ResultResponse{File: data}) }, - func(res interface{}) []byte { + func(res any) []byte { return res.(*agent.ResultResponse).File }, ) @@ -365,7 +365,7 @@ func (s *grpcServer) Attestation(req *agent.AttestationRequest, stream agent.Age func(data []byte) error { return stream.Send(&agent.AttestationResponse{File: data}) }, - func(res interface{}) []byte { + func(res any) []byte { return res.(*agent.AttestationResponse).File }, ) diff --git a/agent/auth/auth.go b/agent/auth/auth.go index bfcb4ea3..3d2288b0 100644 --- a/agent/auth/auth.go +++ b/agent/auth/auth.go @@ -41,9 +41,9 @@ type Authenticator interface { } type service struct { - resultConsumers []interface{} - datasetProviders []interface{} - algorithmProvider interface{} + resultConsumers []any + datasetProviders []any + algorithmProvider any } func New(manifest agent.Computation) (Authenticator, error) { diff --git a/agent/auth/mocks/auth.go b/agent/auth/mocks/authenticator.go similarity index 61% rename from agent/auth/mocks/auth.go rename to agent/auth/mocks/authenticator.go index 6d19af6f..a13ae7b7 100644 --- a/agent/auth/mocks/auth.go +++ b/agent/auth/mocks/authenticator.go @@ -1,18 +1,33 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - context "context" - - auth "github.com/ultravioletrs/cocos/agent/auth" + "context" mock "github.com/stretchr/testify/mock" + "github.com/ultravioletrs/cocos/agent/auth" ) +// NewAuthenticator creates a new instance of Authenticator. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewAuthenticator(t interface { + mock.TestingT + Cleanup(func()) +}) *Authenticator { + mock := &Authenticator{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Authenticator is an autogenerated mock type for the Authenticator type type Authenticator struct { mock.Mock @@ -26,9 +41,9 @@ func (_m *Authenticator) EXPECT() *Authenticator_Expecter { return &Authenticator_Expecter{mock: &_m.Mock} } -// AuthenticateUser provides a mock function with given fields: ctx, role -func (_m *Authenticator) AuthenticateUser(ctx context.Context, role auth.UserRole) (context.Context, error) { - ret := _m.Called(ctx, role) +// AuthenticateUser provides a mock function for the type Authenticator +func (_mock *Authenticator) AuthenticateUser(ctx context.Context, role auth.UserRole) (context.Context, error) { + ret := _mock.Called(ctx, role) if len(ret) == 0 { panic("no return value specified for AuthenticateUser") @@ -36,23 +51,21 @@ func (_m *Authenticator) AuthenticateUser(ctx context.Context, role auth.UserRol var r0 context.Context var r1 error - if rf, ok := ret.Get(0).(func(context.Context, auth.UserRole) (context.Context, error)); ok { - return rf(ctx, role) + if returnFunc, ok := ret.Get(0).(func(context.Context, auth.UserRole) (context.Context, error)); ok { + return returnFunc(ctx, role) } - if rf, ok := ret.Get(0).(func(context.Context, auth.UserRole) context.Context); ok { - r0 = rf(ctx, role) + if returnFunc, ok := ret.Get(0).(func(context.Context, auth.UserRole) context.Context); ok { + r0 = returnFunc(ctx, role) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(context.Context) } } - - if rf, ok := ret.Get(1).(func(context.Context, auth.UserRole) error); ok { - r1 = rf(ctx, role) + if returnFunc, ok := ret.Get(1).(func(context.Context, auth.UserRole) error); ok { + r1 = returnFunc(ctx, role) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -70,31 +83,28 @@ func (_e *Authenticator_Expecter) AuthenticateUser(ctx interface{}, role interfa func (_c *Authenticator_AuthenticateUser_Call) Run(run func(ctx context.Context, role auth.UserRole)) *Authenticator_AuthenticateUser_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(auth.UserRole)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 auth.UserRole + if args[1] != nil { + arg1 = args[1].(auth.UserRole) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Authenticator_AuthenticateUser_Call) Return(_a0 context.Context, _a1 error) *Authenticator_AuthenticateUser_Call { - _c.Call.Return(_a0, _a1) +func (_c *Authenticator_AuthenticateUser_Call) Return(context1 context.Context, err error) *Authenticator_AuthenticateUser_Call { + _c.Call.Return(context1, err) return _c } -func (_c *Authenticator_AuthenticateUser_Call) RunAndReturn(run func(context.Context, auth.UserRole) (context.Context, error)) *Authenticator_AuthenticateUser_Call { +func (_c *Authenticator_AuthenticateUser_Call) RunAndReturn(run func(ctx context.Context, role auth.UserRole) (context.Context, error)) *Authenticator_AuthenticateUser_Call { _c.Call.Return(run) return _c } - -// NewAuthenticator creates a new instance of Authenticator. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewAuthenticator(t interface { - mock.TestingT - Cleanup(func()) -}) *Authenticator { - mock := &Authenticator{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/cvms/api/grpc/storage/mocks/storage.go b/agent/cvms/api/grpc/storage/mocks/storage.go index 2e471948..077929d1 100644 --- a/agent/cvms/api/grpc/storage/mocks/storage.go +++ b/agent/cvms/api/grpc/storage/mocks/storage.go @@ -1,17 +1,32 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( mock "github.com/stretchr/testify/mock" - cvms "github.com/ultravioletrs/cocos/agent/cvms" - - storage "github.com/ultravioletrs/cocos/agent/cvms/api/grpc/storage" + "github.com/ultravioletrs/cocos/agent/cvms" + "github.com/ultravioletrs/cocos/agent/cvms/api/grpc/storage" ) +// NewStorage creates a new instance of Storage. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewStorage(t interface { + mock.TestingT + Cleanup(func()) +}) *Storage { + mock := &Storage{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Storage is an autogenerated mock type for the Storage type type Storage struct { mock.Mock @@ -25,21 +40,20 @@ func (_m *Storage) EXPECT() *Storage_Expecter { return &Storage_Expecter{mock: &_m.Mock} } -// Add provides a mock function with given fields: msg -func (_m *Storage) Add(msg *cvms.ClientStreamMessage) error { - ret := _m.Called(msg) +// Add provides a mock function for the type Storage +func (_mock *Storage) Add(msg *cvms.ClientStreamMessage) error { + ret := _mock.Called(msg) if len(ret) == 0 { panic("no return value specified for Add") } var r0 error - if rf, ok := ret.Get(0).(func(*cvms.ClientStreamMessage) error); ok { - r0 = rf(msg) + if returnFunc, ok := ret.Get(0).(func(*cvms.ClientStreamMessage) error); ok { + r0 = returnFunc(msg) } else { r0 = ret.Error(0) } - return r0 } @@ -56,36 +70,41 @@ func (_e *Storage_Expecter) Add(msg interface{}) *Storage_Add_Call { func (_c *Storage_Add_Call) Run(run func(msg *cvms.ClientStreamMessage)) *Storage_Add_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(*cvms.ClientStreamMessage)) + var arg0 *cvms.ClientStreamMessage + if args[0] != nil { + arg0 = args[0].(*cvms.ClientStreamMessage) + } + run( + arg0, + ) }) return _c } -func (_c *Storage_Add_Call) Return(_a0 error) *Storage_Add_Call { - _c.Call.Return(_a0) +func (_c *Storage_Add_Call) Return(err error) *Storage_Add_Call { + _c.Call.Return(err) return _c } -func (_c *Storage_Add_Call) RunAndReturn(run func(*cvms.ClientStreamMessage) error) *Storage_Add_Call { +func (_c *Storage_Add_Call) RunAndReturn(run func(msg *cvms.ClientStreamMessage) error) *Storage_Add_Call { _c.Call.Return(run) return _c } -// Clear provides a mock function with no fields -func (_m *Storage) Clear() error { - ret := _m.Called() +// Clear provides a mock function for the type Storage +func (_mock *Storage) Clear() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Clear") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -106,8 +125,8 @@ func (_c *Storage_Clear_Call) Run(run func()) *Storage_Clear_Call { return _c } -func (_c *Storage_Clear_Call) Return(_a0 error) *Storage_Clear_Call { - _c.Call.Return(_a0) +func (_c *Storage_Clear_Call) Return(err error) *Storage_Clear_Call { + _c.Call.Return(err) return _c } @@ -116,9 +135,9 @@ func (_c *Storage_Clear_Call) RunAndReturn(run func() error) *Storage_Clear_Call return _c } -// Load provides a mock function with no fields -func (_m *Storage) Load() ([]storage.Message, error) { - ret := _m.Called() +// Load provides a mock function for the type Storage +func (_mock *Storage) Load() ([]storage.Message, error) { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Load") @@ -126,23 +145,21 @@ func (_m *Storage) Load() ([]storage.Message, error) { var r0 []storage.Message var r1 error - if rf, ok := ret.Get(0).(func() ([]storage.Message, error)); ok { - return rf() + if returnFunc, ok := ret.Get(0).(func() ([]storage.Message, error)); ok { + return returnFunc() } - if rf, ok := ret.Get(0).(func() []storage.Message); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() []storage.Message); ok { + r0 = returnFunc() } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]storage.Message) } } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() } else { r1 = ret.Error(1) } - return r0, r1 } @@ -163,8 +180,8 @@ func (_c *Storage_Load_Call) Run(run func()) *Storage_Load_Call { return _c } -func (_c *Storage_Load_Call) Return(_a0 []storage.Message, _a1 error) *Storage_Load_Call { - _c.Call.Return(_a0, _a1) +func (_c *Storage_Load_Call) Return(messages []storage.Message, err error) *Storage_Load_Call { + _c.Call.Return(messages, err) return _c } @@ -173,21 +190,20 @@ func (_c *Storage_Load_Call) RunAndReturn(run func() ([]storage.Message, error)) return _c } -// Save provides a mock function with given fields: messages -func (_m *Storage) Save(messages []storage.Message) error { - ret := _m.Called(messages) +// Save provides a mock function for the type Storage +func (_mock *Storage) Save(messages []storage.Message) error { + ret := _mock.Called(messages) if len(ret) == 0 { panic("no return value specified for Save") } var r0 error - if rf, ok := ret.Get(0).(func([]storage.Message) error); ok { - r0 = rf(messages) + if returnFunc, ok := ret.Get(0).(func([]storage.Message) error); ok { + r0 = returnFunc(messages) } else { r0 = ret.Error(0) } - return r0 } @@ -204,31 +220,23 @@ func (_e *Storage_Expecter) Save(messages interface{}) *Storage_Save_Call { func (_c *Storage_Save_Call) Run(run func(messages []storage.Message)) *Storage_Save_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]storage.Message)) + var arg0 []storage.Message + if args[0] != nil { + arg0 = args[0].([]storage.Message) + } + run( + arg0, + ) }) return _c } -func (_c *Storage_Save_Call) Return(_a0 error) *Storage_Save_Call { - _c.Call.Return(_a0) +func (_c *Storage_Save_Call) Return(err error) *Storage_Save_Call { + _c.Call.Return(err) return _c } -func (_c *Storage_Save_Call) RunAndReturn(run func([]storage.Message) error) *Storage_Save_Call { +func (_c *Storage_Save_Call) RunAndReturn(run func(messages []storage.Message) error) *Storage_Save_Call { _c.Call.Return(run) return _c } - -// NewStorage creates a new instance of Storage. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewStorage(t interface { - mock.TestingT - Cleanup(func()) -}) *Storage { - mock := &Storage{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/cvms/cvms.pb.go b/agent/cvms/cvms.pb.go index 5f5e1437..f73c695a 100644 --- a/agent/cvms/cvms.pb.go +++ b/agent/cvms/cvms.pb.go @@ -3,7 +3,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.36.5 +// protoc-gen-go v1.36.8 // protoc v5.29.0 // source: agent/cvms/cvms.proto @@ -1263,177 +1263,94 @@ func (x *AzureAttestationToken) GetCertSerialNumber() string { var File_agent_cvms_cvms_proto protoreflect.FileDescriptor -var file_agent_cvms_cvms_proto_rawDesc = string([]byte{ - 0x0a, 0x15, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2f, 0x63, 0x76, 0x6d, 0x73, 0x2f, 0x63, 0x76, 0x6d, - 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x04, 0x63, 0x76, 0x6d, 0x73, 0x1a, 0x1f, 0x67, - 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2f, 0x74, - 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0x1f, - 0x0a, 0x0d, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x12, - 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, 0x22, - 0x35, 0x0a, 0x0d, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, - 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, - 0x12, 0x14, 0x0a, 0x05, 0x73, 0x74, 0x61, 0x74, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, - 0x05, 0x73, 0x74, 0x61, 0x74, 0x65, 0x22, 0x38, 0x0a, 0x0f, 0x53, 0x74, 0x6f, 0x70, 0x43, 0x6f, - 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, - 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, - 0x22, 0x5a, 0x0a, 0x17, 0x53, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, - 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x25, 0x0a, 0x0e, 0x63, - 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x01, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, - 0x49, 0x64, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x18, 0x02, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x22, 0x4a, 0x0a, 0x0b, - 0x52, 0x75, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x25, 0x0a, 0x0e, 0x63, - 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x01, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, - 0x49, 0x64, 0x12, 0x14, 0x0a, 0x05, 0x65, 0x72, 0x72, 0x6f, 0x72, 0x18, 0x02, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x05, 0x65, 0x72, 0x72, 0x6f, 0x72, 0x22, 0xde, 0x01, 0x0a, 0x0a, 0x41, 0x67, 0x65, - 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x12, 0x1d, 0x0a, 0x0a, 0x65, 0x76, 0x65, 0x6e, 0x74, - 0x5f, 0x74, 0x79, 0x70, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x65, 0x76, 0x65, - 0x6e, 0x74, 0x54, 0x79, 0x70, 0x65, 0x12, 0x38, 0x0a, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, - 0x61, 0x6d, 0x70, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, - 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, - 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, - 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, - 0x69, 0x64, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, - 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x18, 0x0a, 0x07, 0x64, 0x65, 0x74, 0x61, 0x69, - 0x6c, 0x73, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x64, 0x65, 0x74, 0x61, 0x69, 0x6c, - 0x73, 0x12, 0x1e, 0x0a, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, 0x72, 0x18, - 0x05, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, - 0x72, 0x12, 0x16, 0x0a, 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x18, 0x06, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x22, 0x9b, 0x01, 0x0a, 0x08, 0x41, 0x67, - 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, - 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, - 0x12, 0x25, 0x0a, 0x0e, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, - 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, - 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x49, 0x64, 0x12, 0x14, 0x0a, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, - 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x12, 0x38, 0x0a, - 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0b, - 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, - 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x09, 0x74, 0x69, - 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x22, 0xe8, 0x03, 0x0a, 0x13, 0x43, 0x6c, 0x69, 0x65, - 0x6e, 0x74, 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, - 0x2d, 0x0a, 0x09, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x6c, 0x6f, 0x67, 0x18, 0x01, 0x20, 0x01, - 0x28, 0x0b, 0x32, 0x0e, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x4c, - 0x6f, 0x67, 0x48, 0x00, 0x52, 0x08, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x12, 0x33, - 0x0a, 0x0b, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x18, 0x02, 0x20, - 0x01, 0x28, 0x0b, 0x32, 0x10, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, - 0x45, 0x76, 0x65, 0x6e, 0x74, 0x48, 0x00, 0x52, 0x0a, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, - 0x65, 0x6e, 0x74, 0x12, 0x2c, 0x0a, 0x07, 0x72, 0x75, 0x6e, 0x5f, 0x72, 0x65, 0x73, 0x18, 0x03, - 0x20, 0x01, 0x28, 0x0b, 0x32, 0x11, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x52, 0x75, 0x6e, 0x52, - 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x48, 0x00, 0x52, 0x06, 0x72, 0x75, 0x6e, 0x52, 0x65, - 0x73, 0x12, 0x4f, 0x0a, 0x12, 0x73, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, - 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1d, 0x2e, - 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x53, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, - 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x48, 0x00, 0x52, 0x12, - 0x73, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, - 0x65, 0x73, 0x12, 0x3b, 0x0a, 0x0d, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, - 0x52, 0x65, 0x73, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x13, 0x2e, 0x63, 0x76, 0x6d, 0x73, - 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x48, 0x00, - 0x52, 0x0d, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x12, - 0x51, 0x0a, 0x15, 0x76, 0x54, 0x50, 0x4d, 0x61, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, - 0x6f, 0x6e, 0x52, 0x65, 0x70, 0x6f, 0x72, 0x74, 0x18, 0x06, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x19, - 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, - 0x6e, 0x52, 0x65, 0x73, 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x48, 0x00, 0x52, 0x15, 0x76, 0x54, 0x50, - 0x4d, 0x61, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x70, 0x6f, - 0x72, 0x74, 0x12, 0x53, 0x0a, 0x15, 0x61, 0x7a, 0x75, 0x72, 0x65, 0x41, 0x74, 0x74, 0x65, 0x73, - 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x18, 0x07, 0x20, 0x01, 0x28, - 0x0b, 0x32, 0x1b, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x61, 0x7a, 0x75, 0x72, 0x65, 0x41, 0x74, - 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x48, 0x00, - 0x52, 0x15, 0x61, 0x7a, 0x75, 0x72, 0x65, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, - 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x42, 0x09, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, - 0x67, 0x65, 0x22, 0xca, 0x02, 0x0a, 0x13, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, 0x53, 0x74, 0x72, - 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, 0x38, 0x0a, 0x0c, 0x72, 0x75, - 0x6e, 0x52, 0x65, 0x71, 0x43, 0x68, 0x75, 0x6e, 0x6b, 0x73, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0b, - 0x32, 0x12, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x43, 0x68, - 0x75, 0x6e, 0x6b, 0x73, 0x48, 0x00, 0x52, 0x0c, 0x72, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x43, 0x68, - 0x75, 0x6e, 0x6b, 0x73, 0x12, 0x31, 0x0a, 0x06, 0x72, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x18, 0x02, - 0x20, 0x01, 0x28, 0x0b, 0x32, 0x17, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x43, 0x6f, 0x6d, 0x70, - 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x48, 0x00, 0x52, - 0x06, 0x72, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x12, 0x41, 0x0a, 0x0f, 0x73, 0x74, 0x6f, 0x70, 0x43, - 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0b, - 0x32, 0x15, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x53, 0x74, 0x6f, 0x70, 0x43, 0x6f, 0x6d, 0x70, - 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x48, 0x00, 0x52, 0x0f, 0x73, 0x74, 0x6f, 0x70, 0x43, - 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x3b, 0x0a, 0x0d, 0x61, 0x67, - 0x65, 0x6e, 0x74, 0x53, 0x74, 0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x18, 0x04, 0x20, 0x01, 0x28, - 0x0b, 0x32, 0x13, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x53, 0x74, - 0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x48, 0x00, 0x52, 0x0d, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x53, - 0x74, 0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x12, 0x3b, 0x0a, 0x0d, 0x64, 0x69, 0x73, 0x63, 0x6f, - 0x6e, 0x6e, 0x65, 0x63, 0x74, 0x52, 0x65, 0x71, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x13, - 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x44, 0x69, 0x73, 0x63, 0x6f, 0x6e, 0x6e, 0x65, 0x63, 0x74, - 0x52, 0x65, 0x71, 0x48, 0x00, 0x52, 0x0d, 0x64, 0x69, 0x73, 0x63, 0x6f, 0x6e, 0x6e, 0x65, 0x63, - 0x74, 0x52, 0x65, 0x71, 0x42, 0x09, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x22, - 0x1f, 0x0a, 0x0d, 0x44, 0x69, 0x73, 0x63, 0x6f, 0x6e, 0x6e, 0x65, 0x63, 0x74, 0x52, 0x65, 0x71, - 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, - 0x22, 0x4b, 0x0a, 0x0c, 0x52, 0x75, 0x6e, 0x52, 0x65, 0x71, 0x43, 0x68, 0x75, 0x6e, 0x6b, 0x73, - 0x12, 0x12, 0x0a, 0x04, 0x64, 0x61, 0x74, 0x61, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, - 0x64, 0x61, 0x74, 0x61, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, - 0x52, 0x02, 0x69, 0x64, 0x12, 0x17, 0x0a, 0x07, 0x69, 0x73, 0x5f, 0x6c, 0x61, 0x73, 0x74, 0x18, - 0x03, 0x20, 0x01, 0x28, 0x08, 0x52, 0x06, 0x69, 0x73, 0x4c, 0x61, 0x73, 0x74, 0x22, 0xaa, 0x02, - 0x0a, 0x11, 0x43, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x75, 0x6e, - 0x52, 0x65, 0x71, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, - 0x02, 0x69, 0x64, 0x12, 0x12, 0x0a, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x04, 0x6e, 0x61, 0x6d, 0x65, 0x12, 0x20, 0x0a, 0x0b, 0x64, 0x65, 0x73, 0x63, 0x72, - 0x69, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0b, 0x64, 0x65, - 0x73, 0x63, 0x72, 0x69, 0x70, 0x74, 0x69, 0x6f, 0x6e, 0x12, 0x29, 0x0a, 0x08, 0x64, 0x61, 0x74, - 0x61, 0x73, 0x65, 0x74, 0x73, 0x18, 0x04, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x0d, 0x2e, 0x63, 0x76, - 0x6d, 0x73, 0x2e, 0x44, 0x61, 0x74, 0x61, 0x73, 0x65, 0x74, 0x52, 0x08, 0x64, 0x61, 0x74, 0x61, - 0x73, 0x65, 0x74, 0x73, 0x12, 0x2d, 0x0a, 0x09, 0x61, 0x6c, 0x67, 0x6f, 0x72, 0x69, 0x74, 0x68, - 0x6d, 0x18, 0x05, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x0f, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x41, - 0x6c, 0x67, 0x6f, 0x72, 0x69, 0x74, 0x68, 0x6d, 0x52, 0x09, 0x61, 0x6c, 0x67, 0x6f, 0x72, 0x69, - 0x74, 0x68, 0x6d, 0x12, 0x3f, 0x0a, 0x10, 0x72, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x5f, 0x63, 0x6f, - 0x6e, 0x73, 0x75, 0x6d, 0x65, 0x72, 0x73, 0x18, 0x06, 0x20, 0x03, 0x28, 0x0b, 0x32, 0x14, 0x2e, - 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x52, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x43, 0x6f, 0x6e, 0x73, 0x75, - 0x6d, 0x65, 0x72, 0x52, 0x0f, 0x72, 0x65, 0x73, 0x75, 0x6c, 0x74, 0x43, 0x6f, 0x6e, 0x73, 0x75, - 0x6d, 0x65, 0x72, 0x73, 0x12, 0x34, 0x0a, 0x0c, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x6f, - 0x6e, 0x66, 0x69, 0x67, 0x18, 0x07, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x11, 0x2e, 0x63, 0x76, 0x6d, - 0x73, 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x52, 0x0b, 0x61, - 0x67, 0x65, 0x6e, 0x74, 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x22, 0x2a, 0x0a, 0x0e, 0x52, 0x65, - 0x73, 0x75, 0x6c, 0x74, 0x43, 0x6f, 0x6e, 0x73, 0x75, 0x6d, 0x65, 0x72, 0x12, 0x18, 0x0a, 0x07, - 0x75, 0x73, 0x65, 0x72, 0x4b, 0x65, 0x79, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x75, - 0x73, 0x65, 0x72, 0x4b, 0x65, 0x79, 0x22, 0x53, 0x0a, 0x07, 0x44, 0x61, 0x74, 0x61, 0x73, 0x65, - 0x74, 0x12, 0x12, 0x0a, 0x04, 0x68, 0x61, 0x73, 0x68, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, - 0x04, 0x68, 0x61, 0x73, 0x68, 0x12, 0x18, 0x0a, 0x07, 0x75, 0x73, 0x65, 0x72, 0x4b, 0x65, 0x79, - 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x75, 0x73, 0x65, 0x72, 0x4b, 0x65, 0x79, 0x12, - 0x1a, 0x0a, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x6e, 0x61, 0x6d, 0x65, 0x18, 0x03, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x08, 0x66, 0x69, 0x6c, 0x65, 0x6e, 0x61, 0x6d, 0x65, 0x22, 0x39, 0x0a, 0x09, 0x41, - 0x6c, 0x67, 0x6f, 0x72, 0x69, 0x74, 0x68, 0x6d, 0x12, 0x12, 0x0a, 0x04, 0x68, 0x61, 0x73, 0x68, - 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x68, 0x61, 0x73, 0x68, 0x12, 0x18, 0x0a, 0x07, - 0x75, 0x73, 0x65, 0x72, 0x4b, 0x65, 0x79, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x75, - 0x73, 0x65, 0x72, 0x4b, 0x65, 0x79, 0x22, 0xe5, 0x01, 0x0a, 0x0b, 0x41, 0x67, 0x65, 0x6e, 0x74, - 0x43, 0x6f, 0x6e, 0x66, 0x69, 0x67, 0x12, 0x12, 0x0a, 0x04, 0x70, 0x6f, 0x72, 0x74, 0x18, 0x01, - 0x20, 0x01, 0x28, 0x09, 0x52, 0x04, 0x70, 0x6f, 0x72, 0x74, 0x12, 0x1b, 0x0a, 0x09, 0x63, 0x65, - 0x72, 0x74, 0x5f, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x08, 0x63, - 0x65, 0x72, 0x74, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x19, 0x0a, 0x08, 0x6b, 0x65, 0x79, 0x5f, 0x66, - 0x69, 0x6c, 0x65, 0x18, 0x03, 0x20, 0x01, 0x28, 0x09, 0x52, 0x07, 0x6b, 0x65, 0x79, 0x46, 0x69, - 0x6c, 0x65, 0x12, 0x24, 0x0a, 0x0e, 0x63, 0x6c, 0x69, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x61, 0x5f, - 0x66, 0x69, 0x6c, 0x65, 0x18, 0x04, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0c, 0x63, 0x6c, 0x69, 0x65, - 0x6e, 0x74, 0x43, 0x61, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x24, 0x0a, 0x0e, 0x73, 0x65, 0x72, 0x76, - 0x65, 0x72, 0x5f, 0x63, 0x61, 0x5f, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x05, 0x20, 0x01, 0x28, 0x09, - 0x52, 0x0c, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x43, 0x61, 0x46, 0x69, 0x6c, 0x65, 0x12, 0x1b, - 0x0a, 0x09, 0x6c, 0x6f, 0x67, 0x5f, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x18, 0x06, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x08, 0x6c, 0x6f, 0x67, 0x4c, 0x65, 0x76, 0x65, 0x6c, 0x12, 0x21, 0x0a, 0x0c, 0x61, - 0x74, 0x74, 0x65, 0x73, 0x74, 0x65, 0x64, 0x5f, 0x74, 0x6c, 0x73, 0x18, 0x07, 0x20, 0x01, 0x28, - 0x08, 0x52, 0x0b, 0x61, 0x74, 0x74, 0x65, 0x73, 0x74, 0x65, 0x64, 0x54, 0x6c, 0x73, 0x22, 0x55, - 0x0a, 0x13, 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x52, 0x65, 0x73, - 0x70, 0x6f, 0x6e, 0x73, 0x65, 0x12, 0x12, 0x0a, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, - 0x01, 0x28, 0x0c, 0x52, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x12, 0x2a, 0x0a, 0x10, 0x63, 0x65, 0x72, - 0x74, 0x53, 0x65, 0x72, 0x69, 0x61, 0x6c, 0x4e, 0x75, 0x6d, 0x62, 0x65, 0x72, 0x18, 0x02, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x10, 0x63, 0x65, 0x72, 0x74, 0x53, 0x65, 0x72, 0x69, 0x61, 0x6c, 0x4e, - 0x75, 0x6d, 0x62, 0x65, 0x72, 0x22, 0x57, 0x0a, 0x15, 0x61, 0x7a, 0x75, 0x72, 0x65, 0x41, 0x74, - 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x54, 0x6f, 0x6b, 0x65, 0x6e, 0x12, 0x12, - 0x0a, 0x04, 0x66, 0x69, 0x6c, 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x04, 0x66, 0x69, - 0x6c, 0x65, 0x12, 0x2a, 0x0a, 0x10, 0x63, 0x65, 0x72, 0x74, 0x53, 0x65, 0x72, 0x69, 0x61, 0x6c, - 0x4e, 0x75, 0x6d, 0x62, 0x65, 0x72, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x10, 0x63, 0x65, - 0x72, 0x74, 0x53, 0x65, 0x72, 0x69, 0x61, 0x6c, 0x4e, 0x75, 0x6d, 0x62, 0x65, 0x72, 0x32, 0x50, - 0x0a, 0x07, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x45, 0x0a, 0x07, 0x50, 0x72, 0x6f, - 0x63, 0x65, 0x73, 0x73, 0x12, 0x19, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x43, 0x6c, 0x69, 0x65, - 0x6e, 0x74, 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x1a, - 0x19, 0x2e, 0x63, 0x76, 0x6d, 0x73, 0x2e, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, 0x53, 0x74, 0x72, - 0x65, 0x61, 0x6d, 0x4d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x22, 0x00, 0x28, 0x01, 0x30, 0x01, - 0x42, 0x08, 0x5a, 0x06, 0x2e, 0x2f, 0x63, 0x76, 0x6d, 0x73, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, - 0x6f, 0x33, -}) +const file_agent_cvms_cvms_proto_rawDesc = "" + + "\n" + + "\x15agent/cvms/cvms.proto\x12\x04cvms\x1a\x1fgoogle/protobuf/timestamp.proto\"\x1f\n" + + "\rAgentStateReq\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id\"5\n" + + "\rAgentStateRes\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id\x12\x14\n" + + "\x05state\x18\x02 \x01(\tR\x05state\"8\n" + + "\x0fStopComputation\x12%\n" + + "\x0ecomputation_id\x18\x01 \x01(\tR\rcomputationId\"Z\n" + + "\x17StopComputationResponse\x12%\n" + + "\x0ecomputation_id\x18\x01 \x01(\tR\rcomputationId\x12\x18\n" + + "\amessage\x18\x02 \x01(\tR\amessage\"J\n" + + "\vRunResponse\x12%\n" + + "\x0ecomputation_id\x18\x01 \x01(\tR\rcomputationId\x12\x14\n" + + "\x05error\x18\x02 \x01(\tR\x05error\"\xde\x01\n" + + "\n" + + "AgentEvent\x12\x1d\n" + + "\n" + + "event_type\x18\x01 \x01(\tR\teventType\x128\n" + + "\ttimestamp\x18\x02 \x01(\v2\x1a.google.protobuf.TimestampR\ttimestamp\x12%\n" + + "\x0ecomputation_id\x18\x03 \x01(\tR\rcomputationId\x12\x18\n" + + "\adetails\x18\x04 \x01(\fR\adetails\x12\x1e\n" + + "\n" + + "originator\x18\x05 \x01(\tR\n" + + "originator\x12\x16\n" + + "\x06status\x18\x06 \x01(\tR\x06status\"\x9b\x01\n" + + "\bAgentLog\x12\x18\n" + + "\amessage\x18\x01 \x01(\tR\amessage\x12%\n" + + "\x0ecomputation_id\x18\x02 \x01(\tR\rcomputationId\x12\x14\n" + + "\x05level\x18\x03 \x01(\tR\x05level\x128\n" + + "\ttimestamp\x18\x04 \x01(\v2\x1a.google.protobuf.TimestampR\ttimestamp\"\xe8\x03\n" + + "\x13ClientStreamMessage\x12-\n" + + "\tagent_log\x18\x01 \x01(\v2\x0e.cvms.AgentLogH\x00R\bagentLog\x123\n" + + "\vagent_event\x18\x02 \x01(\v2\x10.cvms.AgentEventH\x00R\n" + + "agentEvent\x12,\n" + + "\arun_res\x18\x03 \x01(\v2\x11.cvms.RunResponseH\x00R\x06runRes\x12O\n" + + "\x12stopComputationRes\x18\x04 \x01(\v2\x1d.cvms.StopComputationResponseH\x00R\x12stopComputationRes\x12;\n" + + "\ragentStateRes\x18\x05 \x01(\v2\x13.cvms.AgentStateResH\x00R\ragentStateRes\x12Q\n" + + "\x15vTPMattestationReport\x18\x06 \x01(\v2\x19.cvms.AttestationResponseH\x00R\x15vTPMattestationReport\x12S\n" + + "\x15azureAttestationToken\x18\a \x01(\v2\x1b.cvms.azureAttestationTokenH\x00R\x15azureAttestationTokenB\t\n" + + "\amessage\"\xca\x02\n" + + "\x13ServerStreamMessage\x128\n" + + "\frunReqChunks\x18\x01 \x01(\v2\x12.cvms.RunReqChunksH\x00R\frunReqChunks\x121\n" + + "\x06runReq\x18\x02 \x01(\v2\x17.cvms.ComputationRunReqH\x00R\x06runReq\x12A\n" + + "\x0fstopComputation\x18\x03 \x01(\v2\x15.cvms.StopComputationH\x00R\x0fstopComputation\x12;\n" + + "\ragentStateReq\x18\x04 \x01(\v2\x13.cvms.AgentStateReqH\x00R\ragentStateReq\x12;\n" + + "\rdisconnectReq\x18\x05 \x01(\v2\x13.cvms.DisconnectReqH\x00R\rdisconnectReqB\t\n" + + "\amessage\"\x1f\n" + + "\rDisconnectReq\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id\"K\n" + + "\fRunReqChunks\x12\x12\n" + + "\x04data\x18\x01 \x01(\fR\x04data\x12\x0e\n" + + "\x02id\x18\x02 \x01(\tR\x02id\x12\x17\n" + + "\ais_last\x18\x03 \x01(\bR\x06isLast\"\xaa\x02\n" + + "\x11ComputationRunReq\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id\x12\x12\n" + + "\x04name\x18\x02 \x01(\tR\x04name\x12 \n" + + "\vdescription\x18\x03 \x01(\tR\vdescription\x12)\n" + + "\bdatasets\x18\x04 \x03(\v2\r.cvms.DatasetR\bdatasets\x12-\n" + + "\talgorithm\x18\x05 \x01(\v2\x0f.cvms.AlgorithmR\talgorithm\x12?\n" + + "\x10result_consumers\x18\x06 \x03(\v2\x14.cvms.ResultConsumerR\x0fresultConsumers\x124\n" + + "\fagent_config\x18\a \x01(\v2\x11.cvms.AgentConfigR\vagentConfig\"*\n" + + "\x0eResultConsumer\x12\x18\n" + + "\auserKey\x18\x01 \x01(\fR\auserKey\"S\n" + + "\aDataset\x12\x12\n" + + "\x04hash\x18\x01 \x01(\fR\x04hash\x12\x18\n" + + "\auserKey\x18\x02 \x01(\fR\auserKey\x12\x1a\n" + + "\bfilename\x18\x03 \x01(\tR\bfilename\"9\n" + + "\tAlgorithm\x12\x12\n" + + "\x04hash\x18\x01 \x01(\fR\x04hash\x12\x18\n" + + "\auserKey\x18\x02 \x01(\fR\auserKey\"\xe5\x01\n" + + "\vAgentConfig\x12\x12\n" + + "\x04port\x18\x01 \x01(\tR\x04port\x12\x1b\n" + + "\tcert_file\x18\x02 \x01(\tR\bcertFile\x12\x19\n" + + "\bkey_file\x18\x03 \x01(\tR\akeyFile\x12$\n" + + "\x0eclient_ca_file\x18\x04 \x01(\tR\fclientCaFile\x12$\n" + + "\x0eserver_ca_file\x18\x05 \x01(\tR\fserverCaFile\x12\x1b\n" + + "\tlog_level\x18\x06 \x01(\tR\blogLevel\x12!\n" + + "\fattested_tls\x18\a \x01(\bR\vattestedTls\"U\n" + + "\x13AttestationResponse\x12\x12\n" + + "\x04file\x18\x01 \x01(\fR\x04file\x12*\n" + + "\x10certSerialNumber\x18\x02 \x01(\tR\x10certSerialNumber\"W\n" + + "\x15azureAttestationToken\x12\x12\n" + + "\x04file\x18\x01 \x01(\fR\x04file\x12*\n" + + "\x10certSerialNumber\x18\x02 \x01(\tR\x10certSerialNumber2P\n" + + "\aService\x12E\n" + + "\aProcess\x12\x19.cvms.ClientStreamMessage\x1a\x19.cvms.ServerStreamMessage\"\x00(\x010\x01B\bZ\x06./cvmsb\x06proto3" var ( file_agent_cvms_cvms_proto_rawDescOnce sync.Once diff --git a/agent/cvms/server/mocks/server.go b/agent/cvms/server/mocks/agentserver.go similarity index 67% rename from agent/cvms/server/mocks/server.go rename to agent/cvms/server/mocks/agentserver.go index 7a7aab82..787dd94b 100644 --- a/agent/cvms/server/mocks/server.go +++ b/agent/cvms/server/mocks/agentserver.go @@ -1,15 +1,31 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( mock "github.com/stretchr/testify/mock" - agent "github.com/ultravioletrs/cocos/agent" + "github.com/ultravioletrs/cocos/agent" ) +// NewAgentServer creates a new instance of AgentServer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewAgentServer(t interface { + mock.TestingT + Cleanup(func()) +}) *AgentServer { + mock := &AgentServer{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // AgentServer is an autogenerated mock type for the AgentServer type type AgentServer struct { mock.Mock @@ -23,21 +39,20 @@ func (_m *AgentServer) EXPECT() *AgentServer_Expecter { return &AgentServer_Expecter{mock: &_m.Mock} } -// Start provides a mock function with given fields: cfg, cmp -func (_m *AgentServer) Start(cfg agent.AgentConfig, cmp agent.Computation) error { - ret := _m.Called(cfg, cmp) +// Start provides a mock function for the type AgentServer +func (_mock *AgentServer) Start(cfg agent.AgentConfig, cmp agent.Computation) error { + ret := _mock.Called(cfg, cmp) if len(ret) == 0 { panic("no return value specified for Start") } var r0 error - if rf, ok := ret.Get(0).(func(agent.AgentConfig, agent.Computation) error); ok { - r0 = rf(cfg, cmp) + if returnFunc, ok := ret.Get(0).(func(agent.AgentConfig, agent.Computation) error); ok { + r0 = returnFunc(cfg, cmp) } else { r0 = ret.Error(0) } - return r0 } @@ -55,36 +70,46 @@ func (_e *AgentServer_Expecter) Start(cfg interface{}, cmp interface{}) *AgentSe func (_c *AgentServer_Start_Call) Run(run func(cfg agent.AgentConfig, cmp agent.Computation)) *AgentServer_Start_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(agent.AgentConfig), args[1].(agent.Computation)) + var arg0 agent.AgentConfig + if args[0] != nil { + arg0 = args[0].(agent.AgentConfig) + } + var arg1 agent.Computation + if args[1] != nil { + arg1 = args[1].(agent.Computation) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *AgentServer_Start_Call) Return(_a0 error) *AgentServer_Start_Call { - _c.Call.Return(_a0) +func (_c *AgentServer_Start_Call) Return(err error) *AgentServer_Start_Call { + _c.Call.Return(err) return _c } -func (_c *AgentServer_Start_Call) RunAndReturn(run func(agent.AgentConfig, agent.Computation) error) *AgentServer_Start_Call { +func (_c *AgentServer_Start_Call) RunAndReturn(run func(cfg agent.AgentConfig, cmp agent.Computation) error) *AgentServer_Start_Call { _c.Call.Return(run) return _c } -// Stop provides a mock function with no fields -func (_m *AgentServer) Stop() error { - ret := _m.Called() +// Stop provides a mock function for the type AgentServer +func (_mock *AgentServer) Stop() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Stop") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -105,8 +130,8 @@ func (_c *AgentServer_Stop_Call) Run(run func()) *AgentServer_Stop_Call { return _c } -func (_c *AgentServer_Stop_Call) Return(_a0 error) *AgentServer_Stop_Call { - _c.Call.Return(_a0) +func (_c *AgentServer_Stop_Call) Return(err error) *AgentServer_Stop_Call { + _c.Call.Return(err) return _c } @@ -114,17 +139,3 @@ func (_c *AgentServer_Stop_Call) RunAndReturn(run func() error) *AgentServer_Sto _c.Call.Return(run) return _c } - -// NewAgentServer creates a new instance of AgentServer. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewAgentServer(t interface { - mock.TestingT - Cleanup(func()) -}) *AgentServer { - mock := &AgentServer{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/events/events.pb.go b/agent/events/events.pb.go index b2839cfc..566f1715 100644 --- a/agent/events/events.pb.go +++ b/agent/events/events.pb.go @@ -3,7 +3,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.36.5 +// protoc-gen-go v1.36.8 // protoc v5.29.0 // source: agent/events/events.proto @@ -261,46 +261,32 @@ func (*EventsLogs_AgentEvent) isEventsLogs_Message() {} var File_agent_events_events_proto protoreflect.FileDescriptor -var file_agent_events_events_proto_rawDesc = string([]byte{ - 0x0a, 0x19, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x2f, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x73, 0x2f, 0x65, - 0x76, 0x65, 0x6e, 0x74, 0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x06, 0x65, 0x76, 0x65, - 0x6e, 0x74, 0x73, 0x1a, 0x1f, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, - 0x6f, 0x62, 0x75, 0x66, 0x2f, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x2e, 0x70, - 0x72, 0x6f, 0x74, 0x6f, 0x22, 0xde, 0x01, 0x0a, 0x0a, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, - 0x65, 0x6e, 0x74, 0x12, 0x1d, 0x0a, 0x0a, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x5f, 0x74, 0x79, 0x70, - 0x65, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x09, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x54, 0x79, - 0x70, 0x65, 0x12, 0x38, 0x0a, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x18, - 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, - 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, - 0x70, 0x52, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x12, 0x25, 0x0a, 0x0e, - 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x03, - 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, - 0x6e, 0x49, 0x64, 0x12, 0x18, 0x0a, 0x07, 0x64, 0x65, 0x74, 0x61, 0x69, 0x6c, 0x73, 0x18, 0x04, - 0x20, 0x01, 0x28, 0x0c, 0x52, 0x07, 0x64, 0x65, 0x74, 0x61, 0x69, 0x6c, 0x73, 0x12, 0x1e, 0x0a, - 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, 0x72, 0x18, 0x05, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x0a, 0x6f, 0x72, 0x69, 0x67, 0x69, 0x6e, 0x61, 0x74, 0x6f, 0x72, 0x12, 0x16, 0x0a, - 0x06, 0x73, 0x74, 0x61, 0x74, 0x75, 0x73, 0x18, 0x06, 0x20, 0x01, 0x28, 0x09, 0x52, 0x06, 0x73, - 0x74, 0x61, 0x74, 0x75, 0x73, 0x22, 0x9b, 0x01, 0x0a, 0x08, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x4c, - 0x6f, 0x67, 0x12, 0x18, 0x0a, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x18, 0x01, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x07, 0x6d, 0x65, 0x73, 0x73, 0x61, 0x67, 0x65, 0x12, 0x25, 0x0a, 0x0e, - 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x5f, 0x69, 0x64, 0x18, 0x02, - 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x63, 0x6f, 0x6d, 0x70, 0x75, 0x74, 0x61, 0x74, 0x69, 0x6f, - 0x6e, 0x49, 0x64, 0x12, 0x14, 0x0a, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x18, 0x03, 0x20, 0x01, - 0x28, 0x09, 0x52, 0x05, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x12, 0x38, 0x0a, 0x09, 0x74, 0x69, 0x6d, - 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x1a, 0x2e, 0x67, - 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x54, - 0x69, 0x6d, 0x65, 0x73, 0x74, 0x61, 0x6d, 0x70, 0x52, 0x09, 0x74, 0x69, 0x6d, 0x65, 0x73, 0x74, - 0x61, 0x6d, 0x70, 0x22, 0x7f, 0x0a, 0x0a, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x73, 0x4c, 0x6f, 0x67, - 0x73, 0x12, 0x2f, 0x0a, 0x09, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x6c, 0x6f, 0x67, 0x18, 0x01, - 0x20, 0x01, 0x28, 0x0b, 0x32, 0x10, 0x2e, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x73, 0x2e, 0x41, 0x67, - 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x48, 0x00, 0x52, 0x08, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x4c, - 0x6f, 0x67, 0x12, 0x35, 0x0a, 0x0b, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x65, 0x76, 0x65, 0x6e, - 0x74, 0x18, 0x02, 0x20, 0x01, 0x28, 0x0b, 0x32, 0x12, 0x2e, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x73, - 0x2e, 0x41, 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x48, 0x00, 0x52, 0x0a, 0x61, - 0x67, 0x65, 0x6e, 0x74, 0x45, 0x76, 0x65, 0x6e, 0x74, 0x42, 0x09, 0x0a, 0x07, 0x6d, 0x65, 0x73, - 0x73, 0x61, 0x67, 0x65, 0x42, 0x0a, 0x5a, 0x08, 0x2e, 0x2f, 0x65, 0x76, 0x65, 0x6e, 0x74, 0x73, - 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, -}) +const file_agent_events_events_proto_rawDesc = "" + + "\n" + + "\x19agent/events/events.proto\x12\x06events\x1a\x1fgoogle/protobuf/timestamp.proto\"\xde\x01\n" + + "\n" + + "AgentEvent\x12\x1d\n" + + "\n" + + "event_type\x18\x01 \x01(\tR\teventType\x128\n" + + "\ttimestamp\x18\x02 \x01(\v2\x1a.google.protobuf.TimestampR\ttimestamp\x12%\n" + + "\x0ecomputation_id\x18\x03 \x01(\tR\rcomputationId\x12\x18\n" + + "\adetails\x18\x04 \x01(\fR\adetails\x12\x1e\n" + + "\n" + + "originator\x18\x05 \x01(\tR\n" + + "originator\x12\x16\n" + + "\x06status\x18\x06 \x01(\tR\x06status\"\x9b\x01\n" + + "\bAgentLog\x12\x18\n" + + "\amessage\x18\x01 \x01(\tR\amessage\x12%\n" + + "\x0ecomputation_id\x18\x02 \x01(\tR\rcomputationId\x12\x14\n" + + "\x05level\x18\x03 \x01(\tR\x05level\x128\n" + + "\ttimestamp\x18\x04 \x01(\v2\x1a.google.protobuf.TimestampR\ttimestamp\"\x7f\n" + + "\n" + + "EventsLogs\x12/\n" + + "\tagent_log\x18\x01 \x01(\v2\x10.events.AgentLogH\x00R\bagentLog\x125\n" + + "\vagent_event\x18\x02 \x01(\v2\x12.events.AgentEventH\x00R\n" + + "agentEventB\t\n" + + "\amessageB\n" + + "Z\b./eventsb\x06proto3" var ( file_agent_events_events_proto_rawDescOnce sync.Once diff --git a/agent/events/mocks/events.go b/agent/events/mocks/service.go similarity index 66% rename from agent/events/mocks/events.go rename to agent/events/mocks/service.go index aa6afe5e..019c8c75 100644 --- a/agent/events/mocks/events.go +++ b/agent/events/mocks/service.go @@ -1,16 +1,32 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - json "encoding/json" + "encoding/json" mock "github.com/stretchr/testify/mock" ) +// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewService(t interface { + mock.TestingT + Cleanup(func()) +}) *Service { + mock := &Service{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Service is an autogenerated mock type for the Service type type Service struct { mock.Mock @@ -24,9 +40,10 @@ func (_m *Service) EXPECT() *Service_Expecter { return &Service_Expecter{mock: &_m.Mock} } -// SendEvent provides a mock function with given fields: cmpID, event, status, details -func (_m *Service) SendEvent(cmpID string, event string, status string, details json.RawMessage) { - _m.Called(cmpID, event, status, details) +// SendEvent provides a mock function for the type Service +func (_mock *Service) SendEvent(cmpID string, event string, status string, details json.RawMessage) { + _mock.Called(cmpID, event, status, details) + return } // Service_SendEvent_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendEvent' @@ -45,7 +62,28 @@ func (_e *Service_Expecter) SendEvent(cmpID interface{}, event interface{}, stat func (_c *Service_SendEvent_Call) Run(run func(cmpID string, event string, status string, details json.RawMessage)) *Service_SendEvent_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(string), args[1].(string), args[2].(string), args[3].(json.RawMessage)) + var arg0 string + if args[0] != nil { + arg0 = args[0].(string) + } + var arg1 string + if args[1] != nil { + arg1 = args[1].(string) + } + var arg2 string + if args[2] != nil { + arg2 = args[2].(string) + } + var arg3 json.RawMessage + if args[3] != nil { + arg3 = args[3].(json.RawMessage) + } + run( + arg0, + arg1, + arg2, + arg3, + ) }) return _c } @@ -55,21 +93,7 @@ func (_c *Service_SendEvent_Call) Return() *Service_SendEvent_Call { return _c } -func (_c *Service_SendEvent_Call) RunAndReturn(run func(string, string, string, json.RawMessage)) *Service_SendEvent_Call { +func (_c *Service_SendEvent_Call) RunAndReturn(run func(cmpID string, event string, status string, details json.RawMessage)) *Service_SendEvent_Call { _c.Run(run) return _c } - -// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewService(t interface { - mock.TestingT - Cleanup(func()) -}) *Service { - mock := &Service{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/mocks/agent_grpc_algo.go b/agent/mocks/agent_grpc_algo.go deleted file mode 100644 index 98adde2f..00000000 --- a/agent/mocks/agent_grpc_algo.go +++ /dev/null @@ -1,434 +0,0 @@ -// Copyright (c) Ultraviolet -// SPDX-License-Identifier: Apache-2.0 - -// Code generated by mockery v2.53.3. DO NOT EDIT. - -package mocks - -import ( - context "context" - - agent "github.com/ultravioletrs/cocos/agent" - - metadata "google.golang.org/grpc/metadata" - - mock "github.com/stretchr/testify/mock" -) - -// AgentService_AlgoClient is an autogenerated mock type for the AgentService_AlgoClient type -type AgentService_AlgoClient[Req interface{}, Res interface{}] struct { - mock.Mock -} - -type AgentService_AlgoClient_Expecter[Req interface{}, Res interface{}] struct { - mock *mock.Mock -} - -func (_m *AgentService_AlgoClient[Req, Res]) EXPECT() *AgentService_AlgoClient_Expecter[Req, Res] { - return &AgentService_AlgoClient_Expecter[Req, Res]{mock: &_m.Mock} -} - -// CloseAndRecv provides a mock function with no fields -func (_m *AgentService_AlgoClient[Req, Res]) CloseAndRecv() (*agent.AlgoResponse, error) { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for CloseAndRecv") - } - - var r0 *agent.AlgoResponse - var r1 error - if rf, ok := ret.Get(0).(func() (*agent.AlgoResponse, error)); ok { - return rf() - } - if rf, ok := ret.Get(0).(func() *agent.AlgoResponse); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*agent.AlgoResponse) - } - } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// AgentService_AlgoClient_CloseAndRecv_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseAndRecv' -type AgentService_AlgoClient_CloseAndRecv_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// CloseAndRecv is a helper method to define mock.On call -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) CloseAndRecv() *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res] { - return &AgentService_AlgoClient_CloseAndRecv_Call[Req, Res]{Call: _e.mock.On("CloseAndRecv")} -} - -func (_c *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res]) Run(run func()) *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res]) Return(_a0 *agent.AlgoResponse, _a1 error) *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res] { - _c.Call.Return(_a0, _a1) - return _c -} - -func (_c *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res]) RunAndReturn(run func() (*agent.AlgoResponse, error)) *AgentService_AlgoClient_CloseAndRecv_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// CloseSend provides a mock function with no fields -func (_m *AgentService_AlgoClient[Req, Res]) CloseSend() error { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for CloseSend") - } - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_AlgoClient_CloseSend_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseSend' -type AgentService_AlgoClient_CloseSend_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// CloseSend is a helper method to define mock.On call -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) CloseSend() *AgentService_AlgoClient_CloseSend_Call[Req, Res] { - return &AgentService_AlgoClient_CloseSend_Call[Req, Res]{Call: _e.mock.On("CloseSend")} -} - -func (_c *AgentService_AlgoClient_CloseSend_Call[Req, Res]) Run(run func()) *AgentService_AlgoClient_CloseSend_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_AlgoClient_CloseSend_Call[Req, Res]) Return(_a0 error) *AgentService_AlgoClient_CloseSend_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_AlgoClient_CloseSend_Call[Req, Res]) RunAndReturn(run func() error) *AgentService_AlgoClient_CloseSend_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Context provides a mock function with no fields -func (_m *AgentService_AlgoClient[Req, Res]) Context() context.Context { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Context") - } - - var r0 context.Context - if rf, ok := ret.Get(0).(func() context.Context); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(context.Context) - } - } - - return r0 -} - -// AgentService_AlgoClient_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' -type AgentService_AlgoClient_Context_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Context is a helper method to define mock.On call -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) Context() *AgentService_AlgoClient_Context_Call[Req, Res] { - return &AgentService_AlgoClient_Context_Call[Req, Res]{Call: _e.mock.On("Context")} -} - -func (_c *AgentService_AlgoClient_Context_Call[Req, Res]) Run(run func()) *AgentService_AlgoClient_Context_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_AlgoClient_Context_Call[Req, Res]) Return(_a0 context.Context) *AgentService_AlgoClient_Context_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_AlgoClient_Context_Call[Req, Res]) RunAndReturn(run func() context.Context) *AgentService_AlgoClient_Context_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Header provides a mock function with no fields -func (_m *AgentService_AlgoClient[Req, Res]) Header() (metadata.MD, error) { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Header") - } - - var r0 metadata.MD - var r1 error - if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { - return rf() - } - if rf, ok := ret.Get(0).(func() metadata.MD); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(metadata.MD) - } - } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// AgentService_AlgoClient_Header_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Header' -type AgentService_AlgoClient_Header_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Header is a helper method to define mock.On call -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) Header() *AgentService_AlgoClient_Header_Call[Req, Res] { - return &AgentService_AlgoClient_Header_Call[Req, Res]{Call: _e.mock.On("Header")} -} - -func (_c *AgentService_AlgoClient_Header_Call[Req, Res]) Run(run func()) *AgentService_AlgoClient_Header_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_AlgoClient_Header_Call[Req, Res]) Return(_a0 metadata.MD, _a1 error) *AgentService_AlgoClient_Header_Call[Req, Res] { - _c.Call.Return(_a0, _a1) - return _c -} - -func (_c *AgentService_AlgoClient_Header_Call[Req, Res]) RunAndReturn(run func() (metadata.MD, error)) *AgentService_AlgoClient_Header_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// RecvMsg provides a mock function with given fields: m -func (_m *AgentService_AlgoClient[Req, Res]) RecvMsg(m interface{}) error { - ret := _m.Called(m) - - if len(ret) == 0 { - panic("no return value specified for RecvMsg") - } - - var r0 error - if rf, ok := ret.Get(0).(func(interface{}) error); ok { - r0 = rf(m) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_AlgoClient_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' -type AgentService_AlgoClient_RecvMsg_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// RecvMsg is a helper method to define mock.On call -// - m interface{} -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) RecvMsg(m interface{}) *AgentService_AlgoClient_RecvMsg_Call[Req, Res] { - return &AgentService_AlgoClient_RecvMsg_Call[Req, Res]{Call: _e.mock.On("RecvMsg", m)} -} - -func (_c *AgentService_AlgoClient_RecvMsg_Call[Req, Res]) Run(run func(m interface{})) *AgentService_AlgoClient_RecvMsg_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{})) - }) - return _c -} - -func (_c *AgentService_AlgoClient_RecvMsg_Call[Req, Res]) Return(_a0 error) *AgentService_AlgoClient_RecvMsg_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_AlgoClient_RecvMsg_Call[Req, Res]) RunAndReturn(run func(interface{}) error) *AgentService_AlgoClient_RecvMsg_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Send provides a mock function with given fields: _a0 -func (_m *AgentService_AlgoClient[Req, Res]) Send(_a0 *agent.AlgoRequest) error { - ret := _m.Called(_a0) - - if len(ret) == 0 { - panic("no return value specified for Send") - } - - var r0 error - if rf, ok := ret.Get(0).(func(*agent.AlgoRequest) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_AlgoClient_Send_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Send' -type AgentService_AlgoClient_Send_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Send is a helper method to define mock.On call -// - _a0 *agent.AlgoRequest -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) Send(_a0 interface{}) *AgentService_AlgoClient_Send_Call[Req, Res] { - return &AgentService_AlgoClient_Send_Call[Req, Res]{Call: _e.mock.On("Send", _a0)} -} - -func (_c *AgentService_AlgoClient_Send_Call[Req, Res]) Run(run func(_a0 *agent.AlgoRequest)) *AgentService_AlgoClient_Send_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(*agent.AlgoRequest)) - }) - return _c -} - -func (_c *AgentService_AlgoClient_Send_Call[Req, Res]) Return(_a0 error) *AgentService_AlgoClient_Send_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_AlgoClient_Send_Call[Req, Res]) RunAndReturn(run func(*agent.AlgoRequest) error) *AgentService_AlgoClient_Send_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// SendMsg provides a mock function with given fields: m -func (_m *AgentService_AlgoClient[Req, Res]) SendMsg(m interface{}) error { - ret := _m.Called(m) - - if len(ret) == 0 { - panic("no return value specified for SendMsg") - } - - var r0 error - if rf, ok := ret.Get(0).(func(interface{}) error); ok { - r0 = rf(m) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_AlgoClient_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' -type AgentService_AlgoClient_SendMsg_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// SendMsg is a helper method to define mock.On call -// - m interface{} -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) SendMsg(m interface{}) *AgentService_AlgoClient_SendMsg_Call[Req, Res] { - return &AgentService_AlgoClient_SendMsg_Call[Req, Res]{Call: _e.mock.On("SendMsg", m)} -} - -func (_c *AgentService_AlgoClient_SendMsg_Call[Req, Res]) Run(run func(m interface{})) *AgentService_AlgoClient_SendMsg_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{})) - }) - return _c -} - -func (_c *AgentService_AlgoClient_SendMsg_Call[Req, Res]) Return(_a0 error) *AgentService_AlgoClient_SendMsg_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_AlgoClient_SendMsg_Call[Req, Res]) RunAndReturn(run func(interface{}) error) *AgentService_AlgoClient_SendMsg_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Trailer provides a mock function with no fields -func (_m *AgentService_AlgoClient[Req, Res]) Trailer() metadata.MD { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Trailer") - } - - var r0 metadata.MD - if rf, ok := ret.Get(0).(func() metadata.MD); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(metadata.MD) - } - } - - return r0 -} - -// AgentService_AlgoClient_Trailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Trailer' -type AgentService_AlgoClient_Trailer_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Trailer is a helper method to define mock.On call -func (_e *AgentService_AlgoClient_Expecter[Req, Res]) Trailer() *AgentService_AlgoClient_Trailer_Call[Req, Res] { - return &AgentService_AlgoClient_Trailer_Call[Req, Res]{Call: _e.mock.On("Trailer")} -} - -func (_c *AgentService_AlgoClient_Trailer_Call[Req, Res]) Run(run func()) *AgentService_AlgoClient_Trailer_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_AlgoClient_Trailer_Call[Req, Res]) Return(_a0 metadata.MD) *AgentService_AlgoClient_Trailer_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_AlgoClient_Trailer_Call[Req, Res]) RunAndReturn(run func() metadata.MD) *AgentService_AlgoClient_Trailer_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// NewAgentService_AlgoClient creates a new instance of AgentService_AlgoClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewAgentService_AlgoClient[Req interface{}, Res interface{}](t interface { - mock.TestingT - Cleanup(func()) -}) *AgentService_AlgoClient[Req, Res] { - mock := &AgentService_AlgoClient[Req, Res]{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/mocks/agent_grpc_data.go b/agent/mocks/agent_grpc_data.go deleted file mode 100644 index 3a279312..00000000 --- a/agent/mocks/agent_grpc_data.go +++ /dev/null @@ -1,434 +0,0 @@ -// Copyright (c) Ultraviolet -// SPDX-License-Identifier: Apache-2.0 - -// Code generated by mockery v2.53.3. DO NOT EDIT. - -package mocks - -import ( - context "context" - - agent "github.com/ultravioletrs/cocos/agent" - - metadata "google.golang.org/grpc/metadata" - - mock "github.com/stretchr/testify/mock" -) - -// AgentService_DataClient is an autogenerated mock type for the AgentService_DataClient type -type AgentService_DataClient[Req interface{}, Res interface{}] struct { - mock.Mock -} - -type AgentService_DataClient_Expecter[Req interface{}, Res interface{}] struct { - mock *mock.Mock -} - -func (_m *AgentService_DataClient[Req, Res]) EXPECT() *AgentService_DataClient_Expecter[Req, Res] { - return &AgentService_DataClient_Expecter[Req, Res]{mock: &_m.Mock} -} - -// CloseAndRecv provides a mock function with no fields -func (_m *AgentService_DataClient[Req, Res]) CloseAndRecv() (*agent.DataResponse, error) { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for CloseAndRecv") - } - - var r0 *agent.DataResponse - var r1 error - if rf, ok := ret.Get(0).(func() (*agent.DataResponse, error)); ok { - return rf() - } - if rf, ok := ret.Get(0).(func() *agent.DataResponse); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*agent.DataResponse) - } - } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// AgentService_DataClient_CloseAndRecv_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseAndRecv' -type AgentService_DataClient_CloseAndRecv_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// CloseAndRecv is a helper method to define mock.On call -func (_e *AgentService_DataClient_Expecter[Req, Res]) CloseAndRecv() *AgentService_DataClient_CloseAndRecv_Call[Req, Res] { - return &AgentService_DataClient_CloseAndRecv_Call[Req, Res]{Call: _e.mock.On("CloseAndRecv")} -} - -func (_c *AgentService_DataClient_CloseAndRecv_Call[Req, Res]) Run(run func()) *AgentService_DataClient_CloseAndRecv_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_DataClient_CloseAndRecv_Call[Req, Res]) Return(_a0 *agent.DataResponse, _a1 error) *AgentService_DataClient_CloseAndRecv_Call[Req, Res] { - _c.Call.Return(_a0, _a1) - return _c -} - -func (_c *AgentService_DataClient_CloseAndRecv_Call[Req, Res]) RunAndReturn(run func() (*agent.DataResponse, error)) *AgentService_DataClient_CloseAndRecv_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// CloseSend provides a mock function with no fields -func (_m *AgentService_DataClient[Req, Res]) CloseSend() error { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for CloseSend") - } - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_DataClient_CloseSend_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseSend' -type AgentService_DataClient_CloseSend_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// CloseSend is a helper method to define mock.On call -func (_e *AgentService_DataClient_Expecter[Req, Res]) CloseSend() *AgentService_DataClient_CloseSend_Call[Req, Res] { - return &AgentService_DataClient_CloseSend_Call[Req, Res]{Call: _e.mock.On("CloseSend")} -} - -func (_c *AgentService_DataClient_CloseSend_Call[Req, Res]) Run(run func()) *AgentService_DataClient_CloseSend_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_DataClient_CloseSend_Call[Req, Res]) Return(_a0 error) *AgentService_DataClient_CloseSend_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_DataClient_CloseSend_Call[Req, Res]) RunAndReturn(run func() error) *AgentService_DataClient_CloseSend_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Context provides a mock function with no fields -func (_m *AgentService_DataClient[Req, Res]) Context() context.Context { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Context") - } - - var r0 context.Context - if rf, ok := ret.Get(0).(func() context.Context); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(context.Context) - } - } - - return r0 -} - -// AgentService_DataClient_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' -type AgentService_DataClient_Context_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Context is a helper method to define mock.On call -func (_e *AgentService_DataClient_Expecter[Req, Res]) Context() *AgentService_DataClient_Context_Call[Req, Res] { - return &AgentService_DataClient_Context_Call[Req, Res]{Call: _e.mock.On("Context")} -} - -func (_c *AgentService_DataClient_Context_Call[Req, Res]) Run(run func()) *AgentService_DataClient_Context_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_DataClient_Context_Call[Req, Res]) Return(_a0 context.Context) *AgentService_DataClient_Context_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_DataClient_Context_Call[Req, Res]) RunAndReturn(run func() context.Context) *AgentService_DataClient_Context_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Header provides a mock function with no fields -func (_m *AgentService_DataClient[Req, Res]) Header() (metadata.MD, error) { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Header") - } - - var r0 metadata.MD - var r1 error - if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { - return rf() - } - if rf, ok := ret.Get(0).(func() metadata.MD); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(metadata.MD) - } - } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// AgentService_DataClient_Header_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Header' -type AgentService_DataClient_Header_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Header is a helper method to define mock.On call -func (_e *AgentService_DataClient_Expecter[Req, Res]) Header() *AgentService_DataClient_Header_Call[Req, Res] { - return &AgentService_DataClient_Header_Call[Req, Res]{Call: _e.mock.On("Header")} -} - -func (_c *AgentService_DataClient_Header_Call[Req, Res]) Run(run func()) *AgentService_DataClient_Header_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_DataClient_Header_Call[Req, Res]) Return(_a0 metadata.MD, _a1 error) *AgentService_DataClient_Header_Call[Req, Res] { - _c.Call.Return(_a0, _a1) - return _c -} - -func (_c *AgentService_DataClient_Header_Call[Req, Res]) RunAndReturn(run func() (metadata.MD, error)) *AgentService_DataClient_Header_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// RecvMsg provides a mock function with given fields: m -func (_m *AgentService_DataClient[Req, Res]) RecvMsg(m interface{}) error { - ret := _m.Called(m) - - if len(ret) == 0 { - panic("no return value specified for RecvMsg") - } - - var r0 error - if rf, ok := ret.Get(0).(func(interface{}) error); ok { - r0 = rf(m) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_DataClient_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' -type AgentService_DataClient_RecvMsg_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// RecvMsg is a helper method to define mock.On call -// - m interface{} -func (_e *AgentService_DataClient_Expecter[Req, Res]) RecvMsg(m interface{}) *AgentService_DataClient_RecvMsg_Call[Req, Res] { - return &AgentService_DataClient_RecvMsg_Call[Req, Res]{Call: _e.mock.On("RecvMsg", m)} -} - -func (_c *AgentService_DataClient_RecvMsg_Call[Req, Res]) Run(run func(m interface{})) *AgentService_DataClient_RecvMsg_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{})) - }) - return _c -} - -func (_c *AgentService_DataClient_RecvMsg_Call[Req, Res]) Return(_a0 error) *AgentService_DataClient_RecvMsg_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_DataClient_RecvMsg_Call[Req, Res]) RunAndReturn(run func(interface{}) error) *AgentService_DataClient_RecvMsg_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Send provides a mock function with given fields: _a0 -func (_m *AgentService_DataClient[Req, Res]) Send(_a0 *agent.DataRequest) error { - ret := _m.Called(_a0) - - if len(ret) == 0 { - panic("no return value specified for Send") - } - - var r0 error - if rf, ok := ret.Get(0).(func(*agent.DataRequest) error); ok { - r0 = rf(_a0) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_DataClient_Send_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Send' -type AgentService_DataClient_Send_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Send is a helper method to define mock.On call -// - _a0 *agent.DataRequest -func (_e *AgentService_DataClient_Expecter[Req, Res]) Send(_a0 interface{}) *AgentService_DataClient_Send_Call[Req, Res] { - return &AgentService_DataClient_Send_Call[Req, Res]{Call: _e.mock.On("Send", _a0)} -} - -func (_c *AgentService_DataClient_Send_Call[Req, Res]) Run(run func(_a0 *agent.DataRequest)) *AgentService_DataClient_Send_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(*agent.DataRequest)) - }) - return _c -} - -func (_c *AgentService_DataClient_Send_Call[Req, Res]) Return(_a0 error) *AgentService_DataClient_Send_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_DataClient_Send_Call[Req, Res]) RunAndReturn(run func(*agent.DataRequest) error) *AgentService_DataClient_Send_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// SendMsg provides a mock function with given fields: m -func (_m *AgentService_DataClient[Req, Res]) SendMsg(m interface{}) error { - ret := _m.Called(m) - - if len(ret) == 0 { - panic("no return value specified for SendMsg") - } - - var r0 error - if rf, ok := ret.Get(0).(func(interface{}) error); ok { - r0 = rf(m) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_DataClient_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' -type AgentService_DataClient_SendMsg_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// SendMsg is a helper method to define mock.On call -// - m interface{} -func (_e *AgentService_DataClient_Expecter[Req, Res]) SendMsg(m interface{}) *AgentService_DataClient_SendMsg_Call[Req, Res] { - return &AgentService_DataClient_SendMsg_Call[Req, Res]{Call: _e.mock.On("SendMsg", m)} -} - -func (_c *AgentService_DataClient_SendMsg_Call[Req, Res]) Run(run func(m interface{})) *AgentService_DataClient_SendMsg_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{})) - }) - return _c -} - -func (_c *AgentService_DataClient_SendMsg_Call[Req, Res]) Return(_a0 error) *AgentService_DataClient_SendMsg_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_DataClient_SendMsg_Call[Req, Res]) RunAndReturn(run func(interface{}) error) *AgentService_DataClient_SendMsg_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// Trailer provides a mock function with no fields -func (_m *AgentService_DataClient[Req, Res]) Trailer() metadata.MD { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Trailer") - } - - var r0 metadata.MD - if rf, ok := ret.Get(0).(func() metadata.MD); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(metadata.MD) - } - } - - return r0 -} - -// AgentService_DataClient_Trailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Trailer' -type AgentService_DataClient_Trailer_Call[Req interface{}, Res interface{}] struct { - *mock.Call -} - -// Trailer is a helper method to define mock.On call -func (_e *AgentService_DataClient_Expecter[Req, Res]) Trailer() *AgentService_DataClient_Trailer_Call[Req, Res] { - return &AgentService_DataClient_Trailer_Call[Req, Res]{Call: _e.mock.On("Trailer")} -} - -func (_c *AgentService_DataClient_Trailer_Call[Req, Res]) Run(run func()) *AgentService_DataClient_Trailer_Call[Req, Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_DataClient_Trailer_Call[Req, Res]) Return(_a0 metadata.MD) *AgentService_DataClient_Trailer_Call[Req, Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_DataClient_Trailer_Call[Req, Res]) RunAndReturn(run func() metadata.MD) *AgentService_DataClient_Trailer_Call[Req, Res] { - _c.Call.Return(run) - return _c -} - -// NewAgentService_DataClient creates a new instance of AgentService_DataClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewAgentService_DataClient[Req interface{}, Res interface{}](t interface { - mock.TestingT - Cleanup(func()) -}) *AgentService_DataClient[Req, Res] { - mock := &AgentService_DataClient[Req, Res]{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/mocks/agent_grpc_ima.go b/agent/mocks/agent_grpc_ima.go deleted file mode 100644 index f287fd14..00000000 --- a/agent/mocks/agent_grpc_ima.go +++ /dev/null @@ -1,388 +0,0 @@ -// Copyright (c) Ultraviolet -// SPDX-License-Identifier: Apache-2.0 - -// Code generated by mockery v2.53.3. DO NOT EDIT. - -package mocks - -import ( - context "context" - - agent "github.com/ultravioletrs/cocos/agent" - - metadata "google.golang.org/grpc/metadata" - - mock "github.com/stretchr/testify/mock" -) - -// AgentService_IMAMeasurementsClient is an autogenerated mock type for the AgentService_IMAMeasurementsClient type -type AgentService_IMAMeasurementsClient[Res interface{}] struct { - mock.Mock -} - -type AgentService_IMAMeasurementsClient_Expecter[Res interface{}] struct { - mock *mock.Mock -} - -func (_m *AgentService_IMAMeasurementsClient[Res]) EXPECT() *AgentService_IMAMeasurementsClient_Expecter[Res] { - return &AgentService_IMAMeasurementsClient_Expecter[Res]{mock: &_m.Mock} -} - -// CloseSend provides a mock function with no fields -func (_m *AgentService_IMAMeasurementsClient[Res]) CloseSend() error { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for CloseSend") - } - - var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_IMAMeasurementsClient_CloseSend_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseSend' -type AgentService_IMAMeasurementsClient_CloseSend_Call[Res interface{}] struct { - *mock.Call -} - -// CloseSend is a helper method to define mock.On call -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) CloseSend() *AgentService_IMAMeasurementsClient_CloseSend_Call[Res] { - return &AgentService_IMAMeasurementsClient_CloseSend_Call[Res]{Call: _e.mock.On("CloseSend")} -} - -func (_c *AgentService_IMAMeasurementsClient_CloseSend_Call[Res]) Run(run func()) *AgentService_IMAMeasurementsClient_CloseSend_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_CloseSend_Call[Res]) Return(_a0 error) *AgentService_IMAMeasurementsClient_CloseSend_Call[Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_CloseSend_Call[Res]) RunAndReturn(run func() error) *AgentService_IMAMeasurementsClient_CloseSend_Call[Res] { - _c.Call.Return(run) - return _c -} - -// Context provides a mock function with no fields -func (_m *AgentService_IMAMeasurementsClient[Res]) Context() context.Context { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Context") - } - - var r0 context.Context - if rf, ok := ret.Get(0).(func() context.Context); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(context.Context) - } - } - - return r0 -} - -// AgentService_IMAMeasurementsClient_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' -type AgentService_IMAMeasurementsClient_Context_Call[Res interface{}] struct { - *mock.Call -} - -// Context is a helper method to define mock.On call -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) Context() *AgentService_IMAMeasurementsClient_Context_Call[Res] { - return &AgentService_IMAMeasurementsClient_Context_Call[Res]{Call: _e.mock.On("Context")} -} - -func (_c *AgentService_IMAMeasurementsClient_Context_Call[Res]) Run(run func()) *AgentService_IMAMeasurementsClient_Context_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Context_Call[Res]) Return(_a0 context.Context) *AgentService_IMAMeasurementsClient_Context_Call[Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Context_Call[Res]) RunAndReturn(run func() context.Context) *AgentService_IMAMeasurementsClient_Context_Call[Res] { - _c.Call.Return(run) - return _c -} - -// Header provides a mock function with no fields -func (_m *AgentService_IMAMeasurementsClient[Res]) Header() (metadata.MD, error) { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Header") - } - - var r0 metadata.MD - var r1 error - if rf, ok := ret.Get(0).(func() (metadata.MD, error)); ok { - return rf() - } - if rf, ok := ret.Get(0).(func() metadata.MD); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(metadata.MD) - } - } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// AgentService_IMAMeasurementsClient_Header_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Header' -type AgentService_IMAMeasurementsClient_Header_Call[Res interface{}] struct { - *mock.Call -} - -// Header is a helper method to define mock.On call -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) Header() *AgentService_IMAMeasurementsClient_Header_Call[Res] { - return &AgentService_IMAMeasurementsClient_Header_Call[Res]{Call: _e.mock.On("Header")} -} - -func (_c *AgentService_IMAMeasurementsClient_Header_Call[Res]) Run(run func()) *AgentService_IMAMeasurementsClient_Header_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Header_Call[Res]) Return(_a0 metadata.MD, _a1 error) *AgentService_IMAMeasurementsClient_Header_Call[Res] { - _c.Call.Return(_a0, _a1) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Header_Call[Res]) RunAndReturn(run func() (metadata.MD, error)) *AgentService_IMAMeasurementsClient_Header_Call[Res] { - _c.Call.Return(run) - return _c -} - -// Recv provides a mock function with no fields -func (_m *AgentService_IMAMeasurementsClient[Res]) Recv() (*agent.IMAMeasurementsResponse, error) { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Recv") - } - - var r0 *agent.IMAMeasurementsResponse - var r1 error - if rf, ok := ret.Get(0).(func() (*agent.IMAMeasurementsResponse, error)); ok { - return rf() - } - if rf, ok := ret.Get(0).(func() *agent.IMAMeasurementsResponse); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(*agent.IMAMeasurementsResponse) - } - } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() - } else { - r1 = ret.Error(1) - } - - return r0, r1 -} - -// AgentService_IMAMeasurementsClient_Recv_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Recv' -type AgentService_IMAMeasurementsClient_Recv_Call[Res interface{}] struct { - *mock.Call -} - -// Recv is a helper method to define mock.On call -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) Recv() *AgentService_IMAMeasurementsClient_Recv_Call[Res] { - return &AgentService_IMAMeasurementsClient_Recv_Call[Res]{Call: _e.mock.On("Recv")} -} - -func (_c *AgentService_IMAMeasurementsClient_Recv_Call[Res]) Run(run func()) *AgentService_IMAMeasurementsClient_Recv_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Recv_Call[Res]) Return(_a0 *agent.IMAMeasurementsResponse, _a1 error) *AgentService_IMAMeasurementsClient_Recv_Call[Res] { - _c.Call.Return(_a0, _a1) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Recv_Call[Res]) RunAndReturn(run func() (*agent.IMAMeasurementsResponse, error)) *AgentService_IMAMeasurementsClient_Recv_Call[Res] { - _c.Call.Return(run) - return _c -} - -// RecvMsg provides a mock function with given fields: m -func (_m *AgentService_IMAMeasurementsClient[Res]) RecvMsg(m interface{}) error { - ret := _m.Called(m) - - if len(ret) == 0 { - panic("no return value specified for RecvMsg") - } - - var r0 error - if rf, ok := ret.Get(0).(func(interface{}) error); ok { - r0 = rf(m) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_IMAMeasurementsClient_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' -type AgentService_IMAMeasurementsClient_RecvMsg_Call[Res interface{}] struct { - *mock.Call -} - -// RecvMsg is a helper method to define mock.On call -// - m interface{} -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) RecvMsg(m interface{}) *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res] { - return &AgentService_IMAMeasurementsClient_RecvMsg_Call[Res]{Call: _e.mock.On("RecvMsg", m)} -} - -func (_c *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res]) Run(run func(m interface{})) *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{})) - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res]) Return(_a0 error) *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res]) RunAndReturn(run func(interface{}) error) *AgentService_IMAMeasurementsClient_RecvMsg_Call[Res] { - _c.Call.Return(run) - return _c -} - -// SendMsg provides a mock function with given fields: m -func (_m *AgentService_IMAMeasurementsClient[Res]) SendMsg(m interface{}) error { - ret := _m.Called(m) - - if len(ret) == 0 { - panic("no return value specified for SendMsg") - } - - var r0 error - if rf, ok := ret.Get(0).(func(interface{}) error); ok { - r0 = rf(m) - } else { - r0 = ret.Error(0) - } - - return r0 -} - -// AgentService_IMAMeasurementsClient_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' -type AgentService_IMAMeasurementsClient_SendMsg_Call[Res interface{}] struct { - *mock.Call -} - -// SendMsg is a helper method to define mock.On call -// - m interface{} -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) SendMsg(m interface{}) *AgentService_IMAMeasurementsClient_SendMsg_Call[Res] { - return &AgentService_IMAMeasurementsClient_SendMsg_Call[Res]{Call: _e.mock.On("SendMsg", m)} -} - -func (_c *AgentService_IMAMeasurementsClient_SendMsg_Call[Res]) Run(run func(m interface{})) *AgentService_IMAMeasurementsClient_SendMsg_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{})) - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_SendMsg_Call[Res]) Return(_a0 error) *AgentService_IMAMeasurementsClient_SendMsg_Call[Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_SendMsg_Call[Res]) RunAndReturn(run func(interface{}) error) *AgentService_IMAMeasurementsClient_SendMsg_Call[Res] { - _c.Call.Return(run) - return _c -} - -// Trailer provides a mock function with no fields -func (_m *AgentService_IMAMeasurementsClient[Res]) Trailer() metadata.MD { - ret := _m.Called() - - if len(ret) == 0 { - panic("no return value specified for Trailer") - } - - var r0 metadata.MD - if rf, ok := ret.Get(0).(func() metadata.MD); ok { - r0 = rf() - } else { - if ret.Get(0) != nil { - r0 = ret.Get(0).(metadata.MD) - } - } - - return r0 -} - -// AgentService_IMAMeasurementsClient_Trailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Trailer' -type AgentService_IMAMeasurementsClient_Trailer_Call[Res interface{}] struct { - *mock.Call -} - -// Trailer is a helper method to define mock.On call -func (_e *AgentService_IMAMeasurementsClient_Expecter[Res]) Trailer() *AgentService_IMAMeasurementsClient_Trailer_Call[Res] { - return &AgentService_IMAMeasurementsClient_Trailer_Call[Res]{Call: _e.mock.On("Trailer")} -} - -func (_c *AgentService_IMAMeasurementsClient_Trailer_Call[Res]) Run(run func()) *AgentService_IMAMeasurementsClient_Trailer_Call[Res] { - _c.Call.Run(func(args mock.Arguments) { - run() - }) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Trailer_Call[Res]) Return(_a0 metadata.MD) *AgentService_IMAMeasurementsClient_Trailer_Call[Res] { - _c.Call.Return(_a0) - return _c -} - -func (_c *AgentService_IMAMeasurementsClient_Trailer_Call[Res]) RunAndReturn(run func() metadata.MD) *AgentService_IMAMeasurementsClient_Trailer_Call[Res] { - _c.Call.Return(run) - return _c -} - -// NewAgentService_IMAMeasurementsClient creates a new instance of AgentService_IMAMeasurementsClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewAgentService_IMAMeasurementsClient[Res interface{}](t interface { - mock.TestingT - Cleanup(func()) -}) *AgentService_IMAMeasurementsClient[Res] { - mock := &AgentService_IMAMeasurementsClient[Res]{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/mocks/agentservice_algoclient.go b/agent/mocks/agentservice_algoclient.go new file mode 100644 index 00000000..34b4d6f6 --- /dev/null +++ b/agent/mocks/agentservice_algoclient.go @@ -0,0 +1,442 @@ +// Copyright (c) Ultraviolet +// SPDX-License-Identifier: Apache-2.0 + +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify + +package mocks + +import ( + "context" + + mock "github.com/stretchr/testify/mock" + "github.com/ultravioletrs/cocos/agent" + "google.golang.org/grpc/metadata" +) + +// NewAgentService_AlgoClient creates a new instance of AgentService_AlgoClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewAgentService_AlgoClient(t interface { + mock.TestingT + Cleanup(func()) +}) *AgentService_AlgoClient { + mock := &AgentService_AlgoClient{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + +// AgentService_AlgoClient is an autogenerated mock type for the AgentService_AlgoClient type +type AgentService_AlgoClient struct { + mock.Mock +} + +type AgentService_AlgoClient_Expecter struct { + mock *mock.Mock +} + +func (_m *AgentService_AlgoClient) EXPECT() *AgentService_AlgoClient_Expecter { + return &AgentService_AlgoClient_Expecter{mock: &_m.Mock} +} + +// CloseAndRecv provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) CloseAndRecv() (*agent.AlgoResponse, error) { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for CloseAndRecv") + } + + var r0 *agent.AlgoResponse + var r1 error + if returnFunc, ok := ret.Get(0).(func() (*agent.AlgoResponse, error)); ok { + return returnFunc() + } + if returnFunc, ok := ret.Get(0).(func() *agent.AlgoResponse); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*agent.AlgoResponse) + } + } + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() + } else { + r1 = ret.Error(1) + } + return r0, r1 +} + +// AgentService_AlgoClient_CloseAndRecv_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseAndRecv' +type AgentService_AlgoClient_CloseAndRecv_Call struct { + *mock.Call +} + +// CloseAndRecv is a helper method to define mock.On call +func (_e *AgentService_AlgoClient_Expecter) CloseAndRecv() *AgentService_AlgoClient_CloseAndRecv_Call { + return &AgentService_AlgoClient_CloseAndRecv_Call{Call: _e.mock.On("CloseAndRecv")} +} + +func (_c *AgentService_AlgoClient_CloseAndRecv_Call) Run(run func()) *AgentService_AlgoClient_CloseAndRecv_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_AlgoClient_CloseAndRecv_Call) Return(algoResponse *agent.AlgoResponse, err error) *AgentService_AlgoClient_CloseAndRecv_Call { + _c.Call.Return(algoResponse, err) + return _c +} + +func (_c *AgentService_AlgoClient_CloseAndRecv_Call) RunAndReturn(run func() (*agent.AlgoResponse, error)) *AgentService_AlgoClient_CloseAndRecv_Call { + _c.Call.Return(run) + return _c +} + +// CloseSend provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) CloseSend() error { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_AlgoClient_CloseSend_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseSend' +type AgentService_AlgoClient_CloseSend_Call struct { + *mock.Call +} + +// CloseSend is a helper method to define mock.On call +func (_e *AgentService_AlgoClient_Expecter) CloseSend() *AgentService_AlgoClient_CloseSend_Call { + return &AgentService_AlgoClient_CloseSend_Call{Call: _e.mock.On("CloseSend")} +} + +func (_c *AgentService_AlgoClient_CloseSend_Call) Run(run func()) *AgentService_AlgoClient_CloseSend_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_AlgoClient_CloseSend_Call) Return(err error) *AgentService_AlgoClient_CloseSend_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_AlgoClient_CloseSend_Call) RunAndReturn(run func() error) *AgentService_AlgoClient_CloseSend_Call { + _c.Call.Return(run) + return _c +} + +// Context provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) Context() context.Context { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Context") + } + + var r0 context.Context + if returnFunc, ok := ret.Get(0).(func() context.Context); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(context.Context) + } + } + return r0 +} + +// AgentService_AlgoClient_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' +type AgentService_AlgoClient_Context_Call struct { + *mock.Call +} + +// Context is a helper method to define mock.On call +func (_e *AgentService_AlgoClient_Expecter) Context() *AgentService_AlgoClient_Context_Call { + return &AgentService_AlgoClient_Context_Call{Call: _e.mock.On("Context")} +} + +func (_c *AgentService_AlgoClient_Context_Call) Run(run func()) *AgentService_AlgoClient_Context_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_AlgoClient_Context_Call) Return(context1 context.Context) *AgentService_AlgoClient_Context_Call { + _c.Call.Return(context1) + return _c +} + +func (_c *AgentService_AlgoClient_Context_Call) RunAndReturn(run func() context.Context) *AgentService_AlgoClient_Context_Call { + _c.Call.Return(run) + return _c +} + +// Header provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) Header() (metadata.MD, error) { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Header") + } + + var r0 metadata.MD + var r1 error + if returnFunc, ok := ret.Get(0).(func() (metadata.MD, error)); ok { + return returnFunc() + } + if returnFunc, ok := ret.Get(0).(func() metadata.MD); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(metadata.MD) + } + } + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() + } else { + r1 = ret.Error(1) + } + return r0, r1 +} + +// AgentService_AlgoClient_Header_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Header' +type AgentService_AlgoClient_Header_Call struct { + *mock.Call +} + +// Header is a helper method to define mock.On call +func (_e *AgentService_AlgoClient_Expecter) Header() *AgentService_AlgoClient_Header_Call { + return &AgentService_AlgoClient_Header_Call{Call: _e.mock.On("Header")} +} + +func (_c *AgentService_AlgoClient_Header_Call) Run(run func()) *AgentService_AlgoClient_Header_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_AlgoClient_Header_Call) Return(mD metadata.MD, err error) *AgentService_AlgoClient_Header_Call { + _c.Call.Return(mD, err) + return _c +} + +func (_c *AgentService_AlgoClient_Header_Call) RunAndReturn(run func() (metadata.MD, error)) *AgentService_AlgoClient_Header_Call { + _c.Call.Return(run) + return _c +} + +// RecvMsg provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) RecvMsg(m any) error { + ret := _mock.Called(m) + + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(any) error); ok { + r0 = returnFunc(m) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_AlgoClient_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' +type AgentService_AlgoClient_RecvMsg_Call struct { + *mock.Call +} + +// RecvMsg is a helper method to define mock.On call +// - m any +func (_e *AgentService_AlgoClient_Expecter) RecvMsg(m interface{}) *AgentService_AlgoClient_RecvMsg_Call { + return &AgentService_AlgoClient_RecvMsg_Call{Call: _e.mock.On("RecvMsg", m)} +} + +func (_c *AgentService_AlgoClient_RecvMsg_Call) Run(run func(m any)) *AgentService_AlgoClient_RecvMsg_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 any + if args[0] != nil { + arg0 = args[0].(any) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_AlgoClient_RecvMsg_Call) Return(err error) *AgentService_AlgoClient_RecvMsg_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_AlgoClient_RecvMsg_Call) RunAndReturn(run func(m any) error) *AgentService_AlgoClient_RecvMsg_Call { + _c.Call.Return(run) + return _c +} + +// Send provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) Send(algoRequest *agent.AlgoRequest) error { + ret := _mock.Called(algoRequest) + + if len(ret) == 0 { + panic("no return value specified for Send") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(*agent.AlgoRequest) error); ok { + r0 = returnFunc(algoRequest) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_AlgoClient_Send_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Send' +type AgentService_AlgoClient_Send_Call struct { + *mock.Call +} + +// Send is a helper method to define mock.On call +// - algoRequest *agent.AlgoRequest +func (_e *AgentService_AlgoClient_Expecter) Send(algoRequest interface{}) *AgentService_AlgoClient_Send_Call { + return &AgentService_AlgoClient_Send_Call{Call: _e.mock.On("Send", algoRequest)} +} + +func (_c *AgentService_AlgoClient_Send_Call) Run(run func(algoRequest *agent.AlgoRequest)) *AgentService_AlgoClient_Send_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 *agent.AlgoRequest + if args[0] != nil { + arg0 = args[0].(*agent.AlgoRequest) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_AlgoClient_Send_Call) Return(err error) *AgentService_AlgoClient_Send_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_AlgoClient_Send_Call) RunAndReturn(run func(algoRequest *agent.AlgoRequest) error) *AgentService_AlgoClient_Send_Call { + _c.Call.Return(run) + return _c +} + +// SendMsg provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) SendMsg(m any) error { + ret := _mock.Called(m) + + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(any) error); ok { + r0 = returnFunc(m) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_AlgoClient_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' +type AgentService_AlgoClient_SendMsg_Call struct { + *mock.Call +} + +// SendMsg is a helper method to define mock.On call +// - m any +func (_e *AgentService_AlgoClient_Expecter) SendMsg(m interface{}) *AgentService_AlgoClient_SendMsg_Call { + return &AgentService_AlgoClient_SendMsg_Call{Call: _e.mock.On("SendMsg", m)} +} + +func (_c *AgentService_AlgoClient_SendMsg_Call) Run(run func(m any)) *AgentService_AlgoClient_SendMsg_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 any + if args[0] != nil { + arg0 = args[0].(any) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_AlgoClient_SendMsg_Call) Return(err error) *AgentService_AlgoClient_SendMsg_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_AlgoClient_SendMsg_Call) RunAndReturn(run func(m any) error) *AgentService_AlgoClient_SendMsg_Call { + _c.Call.Return(run) + return _c +} + +// Trailer provides a mock function for the type AgentService_AlgoClient +func (_mock *AgentService_AlgoClient) Trailer() metadata.MD { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + + var r0 metadata.MD + if returnFunc, ok := ret.Get(0).(func() metadata.MD); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(metadata.MD) + } + } + return r0 +} + +// AgentService_AlgoClient_Trailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Trailer' +type AgentService_AlgoClient_Trailer_Call struct { + *mock.Call +} + +// Trailer is a helper method to define mock.On call +func (_e *AgentService_AlgoClient_Expecter) Trailer() *AgentService_AlgoClient_Trailer_Call { + return &AgentService_AlgoClient_Trailer_Call{Call: _e.mock.On("Trailer")} +} + +func (_c *AgentService_AlgoClient_Trailer_Call) Run(run func()) *AgentService_AlgoClient_Trailer_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_AlgoClient_Trailer_Call) Return(mD metadata.MD) *AgentService_AlgoClient_Trailer_Call { + _c.Call.Return(mD) + return _c +} + +func (_c *AgentService_AlgoClient_Trailer_Call) RunAndReturn(run func() metadata.MD) *AgentService_AlgoClient_Trailer_Call { + _c.Call.Return(run) + return _c +} diff --git a/agent/mocks/agentservice_dataclient.go b/agent/mocks/agentservice_dataclient.go new file mode 100644 index 00000000..473b46c2 --- /dev/null +++ b/agent/mocks/agentservice_dataclient.go @@ -0,0 +1,442 @@ +// Copyright (c) Ultraviolet +// SPDX-License-Identifier: Apache-2.0 + +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify + +package mocks + +import ( + "context" + + mock "github.com/stretchr/testify/mock" + "github.com/ultravioletrs/cocos/agent" + "google.golang.org/grpc/metadata" +) + +// NewAgentService_DataClient creates a new instance of AgentService_DataClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewAgentService_DataClient(t interface { + mock.TestingT + Cleanup(func()) +}) *AgentService_DataClient { + mock := &AgentService_DataClient{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + +// AgentService_DataClient is an autogenerated mock type for the AgentService_DataClient type +type AgentService_DataClient struct { + mock.Mock +} + +type AgentService_DataClient_Expecter struct { + mock *mock.Mock +} + +func (_m *AgentService_DataClient) EXPECT() *AgentService_DataClient_Expecter { + return &AgentService_DataClient_Expecter{mock: &_m.Mock} +} + +// CloseAndRecv provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) CloseAndRecv() (*agent.DataResponse, error) { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for CloseAndRecv") + } + + var r0 *agent.DataResponse + var r1 error + if returnFunc, ok := ret.Get(0).(func() (*agent.DataResponse, error)); ok { + return returnFunc() + } + if returnFunc, ok := ret.Get(0).(func() *agent.DataResponse); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*agent.DataResponse) + } + } + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() + } else { + r1 = ret.Error(1) + } + return r0, r1 +} + +// AgentService_DataClient_CloseAndRecv_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseAndRecv' +type AgentService_DataClient_CloseAndRecv_Call struct { + *mock.Call +} + +// CloseAndRecv is a helper method to define mock.On call +func (_e *AgentService_DataClient_Expecter) CloseAndRecv() *AgentService_DataClient_CloseAndRecv_Call { + return &AgentService_DataClient_CloseAndRecv_Call{Call: _e.mock.On("CloseAndRecv")} +} + +func (_c *AgentService_DataClient_CloseAndRecv_Call) Run(run func()) *AgentService_DataClient_CloseAndRecv_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_DataClient_CloseAndRecv_Call) Return(dataResponse *agent.DataResponse, err error) *AgentService_DataClient_CloseAndRecv_Call { + _c.Call.Return(dataResponse, err) + return _c +} + +func (_c *AgentService_DataClient_CloseAndRecv_Call) RunAndReturn(run func() (*agent.DataResponse, error)) *AgentService_DataClient_CloseAndRecv_Call { + _c.Call.Return(run) + return _c +} + +// CloseSend provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) CloseSend() error { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_DataClient_CloseSend_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseSend' +type AgentService_DataClient_CloseSend_Call struct { + *mock.Call +} + +// CloseSend is a helper method to define mock.On call +func (_e *AgentService_DataClient_Expecter) CloseSend() *AgentService_DataClient_CloseSend_Call { + return &AgentService_DataClient_CloseSend_Call{Call: _e.mock.On("CloseSend")} +} + +func (_c *AgentService_DataClient_CloseSend_Call) Run(run func()) *AgentService_DataClient_CloseSend_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_DataClient_CloseSend_Call) Return(err error) *AgentService_DataClient_CloseSend_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_DataClient_CloseSend_Call) RunAndReturn(run func() error) *AgentService_DataClient_CloseSend_Call { + _c.Call.Return(run) + return _c +} + +// Context provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) Context() context.Context { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Context") + } + + var r0 context.Context + if returnFunc, ok := ret.Get(0).(func() context.Context); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(context.Context) + } + } + return r0 +} + +// AgentService_DataClient_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' +type AgentService_DataClient_Context_Call struct { + *mock.Call +} + +// Context is a helper method to define mock.On call +func (_e *AgentService_DataClient_Expecter) Context() *AgentService_DataClient_Context_Call { + return &AgentService_DataClient_Context_Call{Call: _e.mock.On("Context")} +} + +func (_c *AgentService_DataClient_Context_Call) Run(run func()) *AgentService_DataClient_Context_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_DataClient_Context_Call) Return(context1 context.Context) *AgentService_DataClient_Context_Call { + _c.Call.Return(context1) + return _c +} + +func (_c *AgentService_DataClient_Context_Call) RunAndReturn(run func() context.Context) *AgentService_DataClient_Context_Call { + _c.Call.Return(run) + return _c +} + +// Header provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) Header() (metadata.MD, error) { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Header") + } + + var r0 metadata.MD + var r1 error + if returnFunc, ok := ret.Get(0).(func() (metadata.MD, error)); ok { + return returnFunc() + } + if returnFunc, ok := ret.Get(0).(func() metadata.MD); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(metadata.MD) + } + } + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() + } else { + r1 = ret.Error(1) + } + return r0, r1 +} + +// AgentService_DataClient_Header_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Header' +type AgentService_DataClient_Header_Call struct { + *mock.Call +} + +// Header is a helper method to define mock.On call +func (_e *AgentService_DataClient_Expecter) Header() *AgentService_DataClient_Header_Call { + return &AgentService_DataClient_Header_Call{Call: _e.mock.On("Header")} +} + +func (_c *AgentService_DataClient_Header_Call) Run(run func()) *AgentService_DataClient_Header_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_DataClient_Header_Call) Return(mD metadata.MD, err error) *AgentService_DataClient_Header_Call { + _c.Call.Return(mD, err) + return _c +} + +func (_c *AgentService_DataClient_Header_Call) RunAndReturn(run func() (metadata.MD, error)) *AgentService_DataClient_Header_Call { + _c.Call.Return(run) + return _c +} + +// RecvMsg provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) RecvMsg(m any) error { + ret := _mock.Called(m) + + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(any) error); ok { + r0 = returnFunc(m) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_DataClient_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' +type AgentService_DataClient_RecvMsg_Call struct { + *mock.Call +} + +// RecvMsg is a helper method to define mock.On call +// - m any +func (_e *AgentService_DataClient_Expecter) RecvMsg(m interface{}) *AgentService_DataClient_RecvMsg_Call { + return &AgentService_DataClient_RecvMsg_Call{Call: _e.mock.On("RecvMsg", m)} +} + +func (_c *AgentService_DataClient_RecvMsg_Call) Run(run func(m any)) *AgentService_DataClient_RecvMsg_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 any + if args[0] != nil { + arg0 = args[0].(any) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_DataClient_RecvMsg_Call) Return(err error) *AgentService_DataClient_RecvMsg_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_DataClient_RecvMsg_Call) RunAndReturn(run func(m any) error) *AgentService_DataClient_RecvMsg_Call { + _c.Call.Return(run) + return _c +} + +// Send provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) Send(dataRequest *agent.DataRequest) error { + ret := _mock.Called(dataRequest) + + if len(ret) == 0 { + panic("no return value specified for Send") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(*agent.DataRequest) error); ok { + r0 = returnFunc(dataRequest) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_DataClient_Send_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Send' +type AgentService_DataClient_Send_Call struct { + *mock.Call +} + +// Send is a helper method to define mock.On call +// - dataRequest *agent.DataRequest +func (_e *AgentService_DataClient_Expecter) Send(dataRequest interface{}) *AgentService_DataClient_Send_Call { + return &AgentService_DataClient_Send_Call{Call: _e.mock.On("Send", dataRequest)} +} + +func (_c *AgentService_DataClient_Send_Call) Run(run func(dataRequest *agent.DataRequest)) *AgentService_DataClient_Send_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 *agent.DataRequest + if args[0] != nil { + arg0 = args[0].(*agent.DataRequest) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_DataClient_Send_Call) Return(err error) *AgentService_DataClient_Send_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_DataClient_Send_Call) RunAndReturn(run func(dataRequest *agent.DataRequest) error) *AgentService_DataClient_Send_Call { + _c.Call.Return(run) + return _c +} + +// SendMsg provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) SendMsg(m any) error { + ret := _mock.Called(m) + + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(any) error); ok { + r0 = returnFunc(m) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_DataClient_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' +type AgentService_DataClient_SendMsg_Call struct { + *mock.Call +} + +// SendMsg is a helper method to define mock.On call +// - m any +func (_e *AgentService_DataClient_Expecter) SendMsg(m interface{}) *AgentService_DataClient_SendMsg_Call { + return &AgentService_DataClient_SendMsg_Call{Call: _e.mock.On("SendMsg", m)} +} + +func (_c *AgentService_DataClient_SendMsg_Call) Run(run func(m any)) *AgentService_DataClient_SendMsg_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 any + if args[0] != nil { + arg0 = args[0].(any) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_DataClient_SendMsg_Call) Return(err error) *AgentService_DataClient_SendMsg_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_DataClient_SendMsg_Call) RunAndReturn(run func(m any) error) *AgentService_DataClient_SendMsg_Call { + _c.Call.Return(run) + return _c +} + +// Trailer provides a mock function for the type AgentService_DataClient +func (_mock *AgentService_DataClient) Trailer() metadata.MD { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + + var r0 metadata.MD + if returnFunc, ok := ret.Get(0).(func() metadata.MD); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(metadata.MD) + } + } + return r0 +} + +// AgentService_DataClient_Trailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Trailer' +type AgentService_DataClient_Trailer_Call struct { + *mock.Call +} + +// Trailer is a helper method to define mock.On call +func (_e *AgentService_DataClient_Expecter) Trailer() *AgentService_DataClient_Trailer_Call { + return &AgentService_DataClient_Trailer_Call{Call: _e.mock.On("Trailer")} +} + +func (_c *AgentService_DataClient_Trailer_Call) Run(run func()) *AgentService_DataClient_Trailer_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_DataClient_Trailer_Call) Return(mD metadata.MD) *AgentService_DataClient_Trailer_Call { + _c.Call.Return(mD) + return _c +} + +func (_c *AgentService_DataClient_Trailer_Call) RunAndReturn(run func() metadata.MD) *AgentService_DataClient_Trailer_Call { + _c.Call.Return(run) + return _c +} diff --git a/agent/mocks/agentservice_imameasurementsclient.go b/agent/mocks/agentservice_imameasurementsclient.go new file mode 100644 index 00000000..d2f8578d --- /dev/null +++ b/agent/mocks/agentservice_imameasurementsclient.go @@ -0,0 +1,391 @@ +// Copyright (c) Ultraviolet +// SPDX-License-Identifier: Apache-2.0 + +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify + +package mocks + +import ( + "context" + + mock "github.com/stretchr/testify/mock" + "github.com/ultravioletrs/cocos/agent" + "google.golang.org/grpc/metadata" +) + +// NewAgentService_IMAMeasurementsClient creates a new instance of AgentService_IMAMeasurementsClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewAgentService_IMAMeasurementsClient(t interface { + mock.TestingT + Cleanup(func()) +}) *AgentService_IMAMeasurementsClient { + mock := &AgentService_IMAMeasurementsClient{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + +// AgentService_IMAMeasurementsClient is an autogenerated mock type for the AgentService_IMAMeasurementsClient type +type AgentService_IMAMeasurementsClient struct { + mock.Mock +} + +type AgentService_IMAMeasurementsClient_Expecter struct { + mock *mock.Mock +} + +func (_m *AgentService_IMAMeasurementsClient) EXPECT() *AgentService_IMAMeasurementsClient_Expecter { + return &AgentService_IMAMeasurementsClient_Expecter{mock: &_m.Mock} +} + +// CloseSend provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) CloseSend() error { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for CloseSend") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_IMAMeasurementsClient_CloseSend_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CloseSend' +type AgentService_IMAMeasurementsClient_CloseSend_Call struct { + *mock.Call +} + +// CloseSend is a helper method to define mock.On call +func (_e *AgentService_IMAMeasurementsClient_Expecter) CloseSend() *AgentService_IMAMeasurementsClient_CloseSend_Call { + return &AgentService_IMAMeasurementsClient_CloseSend_Call{Call: _e.mock.On("CloseSend")} +} + +func (_c *AgentService_IMAMeasurementsClient_CloseSend_Call) Run(run func()) *AgentService_IMAMeasurementsClient_CloseSend_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_CloseSend_Call) Return(err error) *AgentService_IMAMeasurementsClient_CloseSend_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_CloseSend_Call) RunAndReturn(run func() error) *AgentService_IMAMeasurementsClient_CloseSend_Call { + _c.Call.Return(run) + return _c +} + +// Context provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) Context() context.Context { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Context") + } + + var r0 context.Context + if returnFunc, ok := ret.Get(0).(func() context.Context); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(context.Context) + } + } + return r0 +} + +// AgentService_IMAMeasurementsClient_Context_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Context' +type AgentService_IMAMeasurementsClient_Context_Call struct { + *mock.Call +} + +// Context is a helper method to define mock.On call +func (_e *AgentService_IMAMeasurementsClient_Expecter) Context() *AgentService_IMAMeasurementsClient_Context_Call { + return &AgentService_IMAMeasurementsClient_Context_Call{Call: _e.mock.On("Context")} +} + +func (_c *AgentService_IMAMeasurementsClient_Context_Call) Run(run func()) *AgentService_IMAMeasurementsClient_Context_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Context_Call) Return(context1 context.Context) *AgentService_IMAMeasurementsClient_Context_Call { + _c.Call.Return(context1) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Context_Call) RunAndReturn(run func() context.Context) *AgentService_IMAMeasurementsClient_Context_Call { + _c.Call.Return(run) + return _c +} + +// Header provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) Header() (metadata.MD, error) { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Header") + } + + var r0 metadata.MD + var r1 error + if returnFunc, ok := ret.Get(0).(func() (metadata.MD, error)); ok { + return returnFunc() + } + if returnFunc, ok := ret.Get(0).(func() metadata.MD); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(metadata.MD) + } + } + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() + } else { + r1 = ret.Error(1) + } + return r0, r1 +} + +// AgentService_IMAMeasurementsClient_Header_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Header' +type AgentService_IMAMeasurementsClient_Header_Call struct { + *mock.Call +} + +// Header is a helper method to define mock.On call +func (_e *AgentService_IMAMeasurementsClient_Expecter) Header() *AgentService_IMAMeasurementsClient_Header_Call { + return &AgentService_IMAMeasurementsClient_Header_Call{Call: _e.mock.On("Header")} +} + +func (_c *AgentService_IMAMeasurementsClient_Header_Call) Run(run func()) *AgentService_IMAMeasurementsClient_Header_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Header_Call) Return(mD metadata.MD, err error) *AgentService_IMAMeasurementsClient_Header_Call { + _c.Call.Return(mD, err) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Header_Call) RunAndReturn(run func() (metadata.MD, error)) *AgentService_IMAMeasurementsClient_Header_Call { + _c.Call.Return(run) + return _c +} + +// Recv provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) Recv() (*agent.IMAMeasurementsResponse, error) { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Recv") + } + + var r0 *agent.IMAMeasurementsResponse + var r1 error + if returnFunc, ok := ret.Get(0).(func() (*agent.IMAMeasurementsResponse, error)); ok { + return returnFunc() + } + if returnFunc, ok := ret.Get(0).(func() *agent.IMAMeasurementsResponse); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(*agent.IMAMeasurementsResponse) + } + } + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() + } else { + r1 = ret.Error(1) + } + return r0, r1 +} + +// AgentService_IMAMeasurementsClient_Recv_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Recv' +type AgentService_IMAMeasurementsClient_Recv_Call struct { + *mock.Call +} + +// Recv is a helper method to define mock.On call +func (_e *AgentService_IMAMeasurementsClient_Expecter) Recv() *AgentService_IMAMeasurementsClient_Recv_Call { + return &AgentService_IMAMeasurementsClient_Recv_Call{Call: _e.mock.On("Recv")} +} + +func (_c *AgentService_IMAMeasurementsClient_Recv_Call) Run(run func()) *AgentService_IMAMeasurementsClient_Recv_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Recv_Call) Return(iMAMeasurementsResponse *agent.IMAMeasurementsResponse, err error) *AgentService_IMAMeasurementsClient_Recv_Call { + _c.Call.Return(iMAMeasurementsResponse, err) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Recv_Call) RunAndReturn(run func() (*agent.IMAMeasurementsResponse, error)) *AgentService_IMAMeasurementsClient_Recv_Call { + _c.Call.Return(run) + return _c +} + +// RecvMsg provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) RecvMsg(m any) error { + ret := _mock.Called(m) + + if len(ret) == 0 { + panic("no return value specified for RecvMsg") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(any) error); ok { + r0 = returnFunc(m) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_IMAMeasurementsClient_RecvMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RecvMsg' +type AgentService_IMAMeasurementsClient_RecvMsg_Call struct { + *mock.Call +} + +// RecvMsg is a helper method to define mock.On call +// - m any +func (_e *AgentService_IMAMeasurementsClient_Expecter) RecvMsg(m interface{}) *AgentService_IMAMeasurementsClient_RecvMsg_Call { + return &AgentService_IMAMeasurementsClient_RecvMsg_Call{Call: _e.mock.On("RecvMsg", m)} +} + +func (_c *AgentService_IMAMeasurementsClient_RecvMsg_Call) Run(run func(m any)) *AgentService_IMAMeasurementsClient_RecvMsg_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 any + if args[0] != nil { + arg0 = args[0].(any) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_RecvMsg_Call) Return(err error) *AgentService_IMAMeasurementsClient_RecvMsg_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_RecvMsg_Call) RunAndReturn(run func(m any) error) *AgentService_IMAMeasurementsClient_RecvMsg_Call { + _c.Call.Return(run) + return _c +} + +// SendMsg provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) SendMsg(m any) error { + ret := _mock.Called(m) + + if len(ret) == 0 { + panic("no return value specified for SendMsg") + } + + var r0 error + if returnFunc, ok := ret.Get(0).(func(any) error); ok { + r0 = returnFunc(m) + } else { + r0 = ret.Error(0) + } + return r0 +} + +// AgentService_IMAMeasurementsClient_SendMsg_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendMsg' +type AgentService_IMAMeasurementsClient_SendMsg_Call struct { + *mock.Call +} + +// SendMsg is a helper method to define mock.On call +// - m any +func (_e *AgentService_IMAMeasurementsClient_Expecter) SendMsg(m interface{}) *AgentService_IMAMeasurementsClient_SendMsg_Call { + return &AgentService_IMAMeasurementsClient_SendMsg_Call{Call: _e.mock.On("SendMsg", m)} +} + +func (_c *AgentService_IMAMeasurementsClient_SendMsg_Call) Run(run func(m any)) *AgentService_IMAMeasurementsClient_SendMsg_Call { + _c.Call.Run(func(args mock.Arguments) { + var arg0 any + if args[0] != nil { + arg0 = args[0].(any) + } + run( + arg0, + ) + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_SendMsg_Call) Return(err error) *AgentService_IMAMeasurementsClient_SendMsg_Call { + _c.Call.Return(err) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_SendMsg_Call) RunAndReturn(run func(m any) error) *AgentService_IMAMeasurementsClient_SendMsg_Call { + _c.Call.Return(run) + return _c +} + +// Trailer provides a mock function for the type AgentService_IMAMeasurementsClient +func (_mock *AgentService_IMAMeasurementsClient) Trailer() metadata.MD { + ret := _mock.Called() + + if len(ret) == 0 { + panic("no return value specified for Trailer") + } + + var r0 metadata.MD + if returnFunc, ok := ret.Get(0).(func() metadata.MD); ok { + r0 = returnFunc() + } else { + if ret.Get(0) != nil { + r0 = ret.Get(0).(metadata.MD) + } + } + return r0 +} + +// AgentService_IMAMeasurementsClient_Trailer_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Trailer' +type AgentService_IMAMeasurementsClient_Trailer_Call struct { + *mock.Call +} + +// Trailer is a helper method to define mock.On call +func (_e *AgentService_IMAMeasurementsClient_Expecter) Trailer() *AgentService_IMAMeasurementsClient_Trailer_Call { + return &AgentService_IMAMeasurementsClient_Trailer_Call{Call: _e.mock.On("Trailer")} +} + +func (_c *AgentService_IMAMeasurementsClient_Trailer_Call) Run(run func()) *AgentService_IMAMeasurementsClient_Trailer_Call { + _c.Call.Run(func(args mock.Arguments) { + run() + }) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Trailer_Call) Return(mD metadata.MD) *AgentService_IMAMeasurementsClient_Trailer_Call { + _c.Call.Return(mD) + return _c +} + +func (_c *AgentService_IMAMeasurementsClient_Trailer_Call) RunAndReturn(run func() metadata.MD) *AgentService_IMAMeasurementsClient_Trailer_Call { + _c.Call.Return(run) + return _c +} diff --git a/agent/mocks/agent.go b/agent/mocks/service.go similarity index 54% rename from agent/mocks/agent.go rename to agent/mocks/service.go index 1c998266..0ea98b89 100644 --- a/agent/mocks/agent.go +++ b/agent/mocks/service.go @@ -1,19 +1,34 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - agent "github.com/ultravioletrs/cocos/agent" - attestation "github.com/ultravioletrs/cocos/pkg/attestation" - - context "context" + "context" mock "github.com/stretchr/testify/mock" + "github.com/ultravioletrs/cocos/agent" + "github.com/ultravioletrs/cocos/pkg/attestation" ) +// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewService(t interface { + mock.TestingT + Cleanup(func()) +}) *Service { + mock := &Service{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Service is an autogenerated mock type for the Service type type Service struct { mock.Mock @@ -27,21 +42,20 @@ func (_m *Service) EXPECT() *Service_Expecter { return &Service_Expecter{mock: &_m.Mock} } -// Algo provides a mock function with given fields: ctx, algorithm -func (_m *Service) Algo(ctx context.Context, algorithm agent.Algorithm) error { - ret := _m.Called(ctx, algorithm) +// Algo provides a mock function for the type Service +func (_mock *Service) Algo(ctx context.Context, algorithm agent.Algorithm) error { + ret := _mock.Called(ctx, algorithm) if len(ret) == 0 { panic("no return value specified for Algo") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, agent.Algorithm) error); ok { - r0 = rf(ctx, algorithm) + if returnFunc, ok := ret.Get(0).(func(context.Context, agent.Algorithm) error); ok { + r0 = returnFunc(ctx, algorithm) } else { r0 = ret.Error(0) } - return r0 } @@ -59,24 +73,35 @@ func (_e *Service_Expecter) Algo(ctx interface{}, algorithm interface{}) *Servic func (_c *Service_Algo_Call) Run(run func(ctx context.Context, algorithm agent.Algorithm)) *Service_Algo_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(agent.Algorithm)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 agent.Algorithm + if args[1] != nil { + arg1 = args[1].(agent.Algorithm) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_Algo_Call) Return(_a0 error) *Service_Algo_Call { - _c.Call.Return(_a0) +func (_c *Service_Algo_Call) Return(err error) *Service_Algo_Call { + _c.Call.Return(err) return _c } -func (_c *Service_Algo_Call) RunAndReturn(run func(context.Context, agent.Algorithm) error) *Service_Algo_Call { +func (_c *Service_Algo_Call) RunAndReturn(run func(ctx context.Context, algorithm agent.Algorithm) error) *Service_Algo_Call { _c.Call.Return(run) return _c } -// Attestation provides a mock function with given fields: ctx, reportData, nonce, attType -func (_m *Service) Attestation(ctx context.Context, reportData [64]byte, nonce [32]byte, attType attestation.PlatformType) ([]byte, error) { - ret := _m.Called(ctx, reportData, nonce, attType) +// Attestation provides a mock function for the type Service +func (_mock *Service) Attestation(ctx context.Context, reportData [64]byte, nonce [32]byte, attType attestation.PlatformType) ([]byte, error) { + ret := _mock.Called(ctx, reportData, nonce, attType) if len(ret) == 0 { panic("no return value specified for Attestation") @@ -84,23 +109,21 @@ func (_m *Service) Attestation(ctx context.Context, reportData [64]byte, nonce [ var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func(context.Context, [64]byte, [32]byte, attestation.PlatformType) ([]byte, error)); ok { - return rf(ctx, reportData, nonce, attType) + if returnFunc, ok := ret.Get(0).(func(context.Context, [64]byte, [32]byte, attestation.PlatformType) ([]byte, error)); ok { + return returnFunc(ctx, reportData, nonce, attType) } - if rf, ok := ret.Get(0).(func(context.Context, [64]byte, [32]byte, attestation.PlatformType) []byte); ok { - r0 = rf(ctx, reportData, nonce, attType) + if returnFunc, ok := ret.Get(0).(func(context.Context, [64]byte, [32]byte, attestation.PlatformType) []byte); ok { + r0 = returnFunc(ctx, reportData, nonce, attType) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(context.Context, [64]byte, [32]byte, attestation.PlatformType) error); ok { - r1 = rf(ctx, reportData, nonce, attType) + if returnFunc, ok := ret.Get(1).(func(context.Context, [64]byte, [32]byte, attestation.PlatformType) error); ok { + r1 = returnFunc(ctx, reportData, nonce, attType) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -120,24 +143,45 @@ func (_e *Service_Expecter) Attestation(ctx interface{}, reportData interface{}, func (_c *Service_Attestation_Call) Run(run func(ctx context.Context, reportData [64]byte, nonce [32]byte, attType attestation.PlatformType)) *Service_Attestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].([64]byte), args[2].([32]byte), args[3].(attestation.PlatformType)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 [64]byte + if args[1] != nil { + arg1 = args[1].([64]byte) + } + var arg2 [32]byte + if args[2] != nil { + arg2 = args[2].([32]byte) + } + var arg3 attestation.PlatformType + if args[3] != nil { + arg3 = args[3].(attestation.PlatformType) + } + run( + arg0, + arg1, + arg2, + arg3, + ) }) return _c } -func (_c *Service_Attestation_Call) Return(_a0 []byte, _a1 error) *Service_Attestation_Call { - _c.Call.Return(_a0, _a1) +func (_c *Service_Attestation_Call) Return(bytes []byte, err error) *Service_Attestation_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Service_Attestation_Call) RunAndReturn(run func(context.Context, [64]byte, [32]byte, attestation.PlatformType) ([]byte, error)) *Service_Attestation_Call { +func (_c *Service_Attestation_Call) RunAndReturn(run func(ctx context.Context, reportData [64]byte, nonce [32]byte, attType attestation.PlatformType) ([]byte, error)) *Service_Attestation_Call { _c.Call.Return(run) return _c } -// AzureAttestationToken provides a mock function with given fields: ctx, nonce -func (_m *Service) AzureAttestationToken(ctx context.Context, nonce [32]byte) ([]byte, error) { - ret := _m.Called(ctx, nonce) +// AzureAttestationToken provides a mock function for the type Service +func (_mock *Service) AzureAttestationToken(ctx context.Context, nonce [32]byte) ([]byte, error) { + ret := _mock.Called(ctx, nonce) if len(ret) == 0 { panic("no return value specified for AzureAttestationToken") @@ -145,23 +189,21 @@ func (_m *Service) AzureAttestationToken(ctx context.Context, nonce [32]byte) ([ var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func(context.Context, [32]byte) ([]byte, error)); ok { - return rf(ctx, nonce) + if returnFunc, ok := ret.Get(0).(func(context.Context, [32]byte) ([]byte, error)); ok { + return returnFunc(ctx, nonce) } - if rf, ok := ret.Get(0).(func(context.Context, [32]byte) []byte); ok { - r0 = rf(ctx, nonce) + if returnFunc, ok := ret.Get(0).(func(context.Context, [32]byte) []byte); ok { + r0 = returnFunc(ctx, nonce) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(context.Context, [32]byte) error); ok { - r1 = rf(ctx, nonce) + if returnFunc, ok := ret.Get(1).(func(context.Context, [32]byte) error); ok { + r1 = returnFunc(ctx, nonce) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -179,36 +221,46 @@ func (_e *Service_Expecter) AzureAttestationToken(ctx interface{}, nonce interfa func (_c *Service_AzureAttestationToken_Call) Run(run func(ctx context.Context, nonce [32]byte)) *Service_AzureAttestationToken_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].([32]byte)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 [32]byte + if args[1] != nil { + arg1 = args[1].([32]byte) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_AzureAttestationToken_Call) Return(_a0 []byte, _a1 error) *Service_AzureAttestationToken_Call { - _c.Call.Return(_a0, _a1) +func (_c *Service_AzureAttestationToken_Call) Return(bytes []byte, err error) *Service_AzureAttestationToken_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Service_AzureAttestationToken_Call) RunAndReturn(run func(context.Context, [32]byte) ([]byte, error)) *Service_AzureAttestationToken_Call { +func (_c *Service_AzureAttestationToken_Call) RunAndReturn(run func(ctx context.Context, nonce [32]byte) ([]byte, error)) *Service_AzureAttestationToken_Call { _c.Call.Return(run) return _c } -// Data provides a mock function with given fields: ctx, dataset -func (_m *Service) Data(ctx context.Context, dataset agent.Dataset) error { - ret := _m.Called(ctx, dataset) +// Data provides a mock function for the type Service +func (_mock *Service) Data(ctx context.Context, dataset agent.Dataset) error { + ret := _mock.Called(ctx, dataset) if len(ret) == 0 { panic("no return value specified for Data") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, agent.Dataset) error); ok { - r0 = rf(ctx, dataset) + if returnFunc, ok := ret.Get(0).(func(context.Context, agent.Dataset) error); ok { + r0 = returnFunc(ctx, dataset) } else { r0 = ret.Error(0) } - return r0 } @@ -226,24 +278,35 @@ func (_e *Service_Expecter) Data(ctx interface{}, dataset interface{}) *Service_ func (_c *Service_Data_Call) Run(run func(ctx context.Context, dataset agent.Dataset)) *Service_Data_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(agent.Dataset)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 agent.Dataset + if args[1] != nil { + arg1 = args[1].(agent.Dataset) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_Data_Call) Return(_a0 error) *Service_Data_Call { - _c.Call.Return(_a0) +func (_c *Service_Data_Call) Return(err error) *Service_Data_Call { + _c.Call.Return(err) return _c } -func (_c *Service_Data_Call) RunAndReturn(run func(context.Context, agent.Dataset) error) *Service_Data_Call { +func (_c *Service_Data_Call) RunAndReturn(run func(ctx context.Context, dataset agent.Dataset) error) *Service_Data_Call { _c.Call.Return(run) return _c } -// IMAMeasurements provides a mock function with given fields: ctx -func (_m *Service) IMAMeasurements(ctx context.Context) ([]byte, []byte, error) { - ret := _m.Called(ctx) +// IMAMeasurements provides a mock function for the type Service +func (_mock *Service) IMAMeasurements(ctx context.Context) ([]byte, []byte, error) { + ret := _mock.Called(ctx) if len(ret) == 0 { panic("no return value specified for IMAMeasurements") @@ -252,31 +315,28 @@ func (_m *Service) IMAMeasurements(ctx context.Context) ([]byte, []byte, error) var r0 []byte var r1 []byte var r2 error - if rf, ok := ret.Get(0).(func(context.Context) ([]byte, []byte, error)); ok { - return rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) ([]byte, []byte, error)); ok { + return returnFunc(ctx) } - if rf, ok := ret.Get(0).(func(context.Context) []byte); ok { - r0 = rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) []byte); ok { + r0 = returnFunc(ctx) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(context.Context) []byte); ok { - r1 = rf(ctx) + if returnFunc, ok := ret.Get(1).(func(context.Context) []byte); ok { + r1 = returnFunc(ctx) } else { if ret.Get(1) != nil { r1 = ret.Get(1).([]byte) } } - - if rf, ok := ret.Get(2).(func(context.Context) error); ok { - r2 = rf(ctx) + if returnFunc, ok := ret.Get(2).(func(context.Context) error); ok { + r2 = returnFunc(ctx) } else { r2 = ret.Error(2) } - return r0, r1, r2 } @@ -293,36 +353,41 @@ func (_e *Service_Expecter) IMAMeasurements(ctx interface{}) *Service_IMAMeasure func (_c *Service_IMAMeasurements_Call) Run(run func(ctx context.Context)) *Service_IMAMeasurements_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + run( + arg0, + ) }) return _c } -func (_c *Service_IMAMeasurements_Call) Return(_a0 []byte, _a1 []byte, _a2 error) *Service_IMAMeasurements_Call { - _c.Call.Return(_a0, _a1, _a2) +func (_c *Service_IMAMeasurements_Call) Return(bytes []byte, bytes1 []byte, err error) *Service_IMAMeasurements_Call { + _c.Call.Return(bytes, bytes1, err) return _c } -func (_c *Service_IMAMeasurements_Call) RunAndReturn(run func(context.Context) ([]byte, []byte, error)) *Service_IMAMeasurements_Call { +func (_c *Service_IMAMeasurements_Call) RunAndReturn(run func(ctx context.Context) ([]byte, []byte, error)) *Service_IMAMeasurements_Call { _c.Call.Return(run) return _c } -// InitComputation provides a mock function with given fields: ctx, cmp -func (_m *Service) InitComputation(ctx context.Context, cmp agent.Computation) error { - ret := _m.Called(ctx, cmp) +// InitComputation provides a mock function for the type Service +func (_mock *Service) InitComputation(ctx context.Context, cmp agent.Computation) error { + ret := _mock.Called(ctx, cmp) if len(ret) == 0 { panic("no return value specified for InitComputation") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, agent.Computation) error); ok { - r0 = rf(ctx, cmp) + if returnFunc, ok := ret.Get(0).(func(context.Context, agent.Computation) error); ok { + r0 = returnFunc(ctx, cmp) } else { r0 = ret.Error(0) } - return r0 } @@ -340,24 +405,35 @@ func (_e *Service_Expecter) InitComputation(ctx interface{}, cmp interface{}) *S func (_c *Service_InitComputation_Call) Run(run func(ctx context.Context, cmp agent.Computation)) *Service_InitComputation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(agent.Computation)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 agent.Computation + if args[1] != nil { + arg1 = args[1].(agent.Computation) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_InitComputation_Call) Return(_a0 error) *Service_InitComputation_Call { - _c.Call.Return(_a0) +func (_c *Service_InitComputation_Call) Return(err error) *Service_InitComputation_Call { + _c.Call.Return(err) return _c } -func (_c *Service_InitComputation_Call) RunAndReturn(run func(context.Context, agent.Computation) error) *Service_InitComputation_Call { +func (_c *Service_InitComputation_Call) RunAndReturn(run func(ctx context.Context, cmp agent.Computation) error) *Service_InitComputation_Call { _c.Call.Return(run) return _c } -// Result provides a mock function with given fields: ctx -func (_m *Service) Result(ctx context.Context) ([]byte, error) { - ret := _m.Called(ctx) +// Result provides a mock function for the type Service +func (_mock *Service) Result(ctx context.Context) ([]byte, error) { + ret := _mock.Called(ctx) if len(ret) == 0 { panic("no return value specified for Result") @@ -365,23 +441,21 @@ func (_m *Service) Result(ctx context.Context) ([]byte, error) { var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func(context.Context) ([]byte, error)); ok { - return rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) ([]byte, error)); ok { + return returnFunc(ctx) } - if rf, ok := ret.Get(0).(func(context.Context) []byte); ok { - r0 = rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) []byte); ok { + r0 = returnFunc(ctx) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(context.Context) error); ok { - r1 = rf(ctx) + if returnFunc, ok := ret.Get(1).(func(context.Context) error); ok { + r1 = returnFunc(ctx) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -398,36 +472,41 @@ func (_e *Service_Expecter) Result(ctx interface{}) *Service_Result_Call { func (_c *Service_Result_Call) Run(run func(ctx context.Context)) *Service_Result_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + run( + arg0, + ) }) return _c } -func (_c *Service_Result_Call) Return(_a0 []byte, _a1 error) *Service_Result_Call { - _c.Call.Return(_a0, _a1) +func (_c *Service_Result_Call) Return(bytes []byte, err error) *Service_Result_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Service_Result_Call) RunAndReturn(run func(context.Context) ([]byte, error)) *Service_Result_Call { +func (_c *Service_Result_Call) RunAndReturn(run func(ctx context.Context) ([]byte, error)) *Service_Result_Call { _c.Call.Return(run) return _c } -// State provides a mock function with no fields -func (_m *Service) State() string { - ret := _m.Called() +// State provides a mock function for the type Service +func (_mock *Service) State() string { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for State") } var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() string); ok { + r0 = returnFunc() } else { r0 = ret.Get(0).(string) } - return r0 } @@ -448,8 +527,8 @@ func (_c *Service_State_Call) Run(run func()) *Service_State_Call { return _c } -func (_c *Service_State_Call) Return(_a0 string) *Service_State_Call { - _c.Call.Return(_a0) +func (_c *Service_State_Call) Return(s string) *Service_State_Call { + _c.Call.Return(s) return _c } @@ -458,21 +537,20 @@ func (_c *Service_State_Call) RunAndReturn(run func() string) *Service_State_Cal return _c } -// StopComputation provides a mock function with given fields: ctx -func (_m *Service) StopComputation(ctx context.Context) error { - ret := _m.Called(ctx) +// StopComputation provides a mock function for the type Service +func (_mock *Service) StopComputation(ctx context.Context) error { + ret := _mock.Called(ctx) if len(ret) == 0 { panic("no return value specified for StopComputation") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context) error); ok { - r0 = rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) error); ok { + r0 = returnFunc(ctx) } else { r0 = ret.Error(0) } - return r0 } @@ -489,31 +567,23 @@ func (_e *Service_Expecter) StopComputation(ctx interface{}) *Service_StopComput func (_c *Service_StopComputation_Call) Run(run func(ctx context.Context)) *Service_StopComputation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + run( + arg0, + ) }) return _c } -func (_c *Service_StopComputation_Call) Return(_a0 error) *Service_StopComputation_Call { - _c.Call.Return(_a0) +func (_c *Service_StopComputation_Call) Return(err error) *Service_StopComputation_Call { + _c.Call.Return(err) return _c } -func (_c *Service_StopComputation_Call) RunAndReturn(run func(context.Context) error) *Service_StopComputation_Call { +func (_c *Service_StopComputation_Call) RunAndReturn(run func(ctx context.Context) error) *Service_StopComputation_Call { _c.Call.Return(run) return _c } - -// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewService(t interface { - mock.TestingT - Cleanup(func()) -}) *Service { - mock := &Service{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/agent/statemachine/mocks/state.go b/agent/statemachine/mocks/statemachine.go similarity index 68% rename from agent/statemachine/mocks/state.go rename to agent/statemachine/mocks/statemachine.go index deaa7f0c..3d6f9eb6 100644 --- a/agent/statemachine/mocks/state.go +++ b/agent/statemachine/mocks/statemachine.go @@ -1,17 +1,33 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - context "context" + "context" mock "github.com/stretchr/testify/mock" - statemachine "github.com/ultravioletrs/cocos/agent/statemachine" + "github.com/ultravioletrs/cocos/agent/statemachine" ) +// NewStateMachine creates a new instance of StateMachine. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewStateMachine(t interface { + mock.TestingT + Cleanup(func()) +}) *StateMachine { + mock := &StateMachine{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // StateMachine is an autogenerated mock type for the StateMachine type type StateMachine struct { mock.Mock @@ -25,9 +41,10 @@ func (_m *StateMachine) EXPECT() *StateMachine_Expecter { return &StateMachine_Expecter{mock: &_m.Mock} } -// AddTransition provides a mock function with given fields: t -func (_m *StateMachine) AddTransition(t statemachine.Transition) { - _m.Called(t) +// AddTransition provides a mock function for the type StateMachine +func (_mock *StateMachine) AddTransition(t statemachine.Transition) { + _mock.Called(t) + return } // StateMachine_AddTransition_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AddTransition' @@ -43,7 +60,13 @@ func (_e *StateMachine_Expecter) AddTransition(t interface{}) *StateMachine_AddT func (_c *StateMachine_AddTransition_Call) Run(run func(t statemachine.Transition)) *StateMachine_AddTransition_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(statemachine.Transition)) + var arg0 statemachine.Transition + if args[0] != nil { + arg0 = args[0].(statemachine.Transition) + } + run( + arg0, + ) }) return _c } @@ -53,28 +76,27 @@ func (_c *StateMachine_AddTransition_Call) Return() *StateMachine_AddTransition_ return _c } -func (_c *StateMachine_AddTransition_Call) RunAndReturn(run func(statemachine.Transition)) *StateMachine_AddTransition_Call { +func (_c *StateMachine_AddTransition_Call) RunAndReturn(run func(t statemachine.Transition)) *StateMachine_AddTransition_Call { _c.Run(run) return _c } -// GetState provides a mock function with no fields -func (_m *StateMachine) GetState() statemachine.State { - ret := _m.Called() +// GetState provides a mock function for the type StateMachine +func (_mock *StateMachine) GetState() statemachine.State { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for GetState") } var r0 statemachine.State - if rf, ok := ret.Get(0).(func() statemachine.State); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() statemachine.State); ok { + r0 = returnFunc() } else { if ret.Get(0) != nil { r0 = ret.Get(0).(statemachine.State) } } - return r0 } @@ -95,8 +117,8 @@ func (_c *StateMachine_GetState_Call) Run(run func()) *StateMachine_GetState_Cal return _c } -func (_c *StateMachine_GetState_Call) Return(_a0 statemachine.State) *StateMachine_GetState_Call { - _c.Call.Return(_a0) +func (_c *StateMachine_GetState_Call) Return(state statemachine.State) *StateMachine_GetState_Call { + _c.Call.Return(state) return _c } @@ -105,9 +127,10 @@ func (_c *StateMachine_GetState_Call) RunAndReturn(run func() statemachine.State return _c } -// Reset provides a mock function with given fields: initialState -func (_m *StateMachine) Reset(initialState statemachine.State) { - _m.Called(initialState) +// Reset provides a mock function for the type StateMachine +func (_mock *StateMachine) Reset(initialState statemachine.State) { + _mock.Called(initialState) + return } // StateMachine_Reset_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Reset' @@ -123,7 +146,13 @@ func (_e *StateMachine_Expecter) Reset(initialState interface{}) *StateMachine_R func (_c *StateMachine_Reset_Call) Run(run func(initialState statemachine.State)) *StateMachine_Reset_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(statemachine.State)) + var arg0 statemachine.State + if args[0] != nil { + arg0 = args[0].(statemachine.State) + } + run( + arg0, + ) }) return _c } @@ -133,14 +162,15 @@ func (_c *StateMachine_Reset_Call) Return() *StateMachine_Reset_Call { return _c } -func (_c *StateMachine_Reset_Call) RunAndReturn(run func(statemachine.State)) *StateMachine_Reset_Call { +func (_c *StateMachine_Reset_Call) RunAndReturn(run func(initialState statemachine.State)) *StateMachine_Reset_Call { _c.Run(run) return _c } -// SendEvent provides a mock function with given fields: event -func (_m *StateMachine) SendEvent(event statemachine.Event) { - _m.Called(event) +// SendEvent provides a mock function for the type StateMachine +func (_mock *StateMachine) SendEvent(event statemachine.Event) { + _mock.Called(event) + return } // StateMachine_SendEvent_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SendEvent' @@ -156,7 +186,13 @@ func (_e *StateMachine_Expecter) SendEvent(event interface{}) *StateMachine_Send func (_c *StateMachine_SendEvent_Call) Run(run func(event statemachine.Event)) *StateMachine_SendEvent_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(statemachine.Event)) + var arg0 statemachine.Event + if args[0] != nil { + arg0 = args[0].(statemachine.Event) + } + run( + arg0, + ) }) return _c } @@ -166,14 +202,15 @@ func (_c *StateMachine_SendEvent_Call) Return() *StateMachine_SendEvent_Call { return _c } -func (_c *StateMachine_SendEvent_Call) RunAndReturn(run func(statemachine.Event)) *StateMachine_SendEvent_Call { +func (_c *StateMachine_SendEvent_Call) RunAndReturn(run func(event statemachine.Event)) *StateMachine_SendEvent_Call { _c.Run(run) return _c } -// SetAction provides a mock function with given fields: state, action -func (_m *StateMachine) SetAction(state statemachine.State, action statemachine.Action) { - _m.Called(state, action) +// SetAction provides a mock function for the type StateMachine +func (_mock *StateMachine) SetAction(state statemachine.State, action statemachine.Action) { + _mock.Called(state, action) + return } // StateMachine_SetAction_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetAction' @@ -190,7 +227,18 @@ func (_e *StateMachine_Expecter) SetAction(state interface{}, action interface{} func (_c *StateMachine_SetAction_Call) Run(run func(state statemachine.State, action statemachine.Action)) *StateMachine_SetAction_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(statemachine.State), args[1].(statemachine.Action)) + var arg0 statemachine.State + if args[0] != nil { + arg0 = args[0].(statemachine.State) + } + var arg1 statemachine.Action + if args[1] != nil { + arg1 = args[1].(statemachine.Action) + } + run( + arg0, + arg1, + ) }) return _c } @@ -200,26 +248,25 @@ func (_c *StateMachine_SetAction_Call) Return() *StateMachine_SetAction_Call { return _c } -func (_c *StateMachine_SetAction_Call) RunAndReturn(run func(statemachine.State, statemachine.Action)) *StateMachine_SetAction_Call { +func (_c *StateMachine_SetAction_Call) RunAndReturn(run func(state statemachine.State, action statemachine.Action)) *StateMachine_SetAction_Call { _c.Run(run) return _c } -// Start provides a mock function with given fields: ctx -func (_m *StateMachine) Start(ctx context.Context) error { - ret := _m.Called(ctx) +// Start provides a mock function for the type StateMachine +func (_mock *StateMachine) Start(ctx context.Context) error { + ret := _mock.Called(ctx) if len(ret) == 0 { panic("no return value specified for Start") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context) error); ok { - r0 = rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) error); ok { + r0 = returnFunc(ctx) } else { r0 = ret.Error(0) } - return r0 } @@ -236,31 +283,23 @@ func (_e *StateMachine_Expecter) Start(ctx interface{}) *StateMachine_Start_Call func (_c *StateMachine_Start_Call) Run(run func(ctx context.Context)) *StateMachine_Start_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + run( + arg0, + ) }) return _c } -func (_c *StateMachine_Start_Call) Return(_a0 error) *StateMachine_Start_Call { - _c.Call.Return(_a0) +func (_c *StateMachine_Start_Call) Return(err error) *StateMachine_Start_Call { + _c.Call.Return(err) return _c } -func (_c *StateMachine_Start_Call) RunAndReturn(run func(context.Context) error) *StateMachine_Start_Call { +func (_c *StateMachine_Start_Call) RunAndReturn(run func(ctx context.Context) error) *StateMachine_Start_Call { _c.Call.Return(run) return _c } - -// NewStateMachine creates a new instance of StateMachine. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewStateMachine(t interface { - mock.TestingT - Cleanup(func()) -}) *StateMachine { - mock := &StateMachine{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/cli/attestation.go b/cli/attestation.go index f089a139..a05934fa 100644 --- a/cli/attestation.go +++ b/cli/attestation.go @@ -551,7 +551,7 @@ func decodeJWTToJSON(tokenBytes []byte) ([]byte, error) { return nil, fmt.Errorf("invalid JWT: must have at least 2 parts") } - decode := func(seg string) (map[string]interface{}, error) { + decode := func(seg string) (map[string]any, error) { // Add padding if missing if m := len(seg) % 4; m != 0 { seg += strings.Repeat("=", 4-m) @@ -562,7 +562,7 @@ func decodeJWTToJSON(tokenBytes []byte) ([]byte, error) { return nil, err } - var result map[string]interface{} + var result map[string]any if err := json.Unmarshal(data, &result); err != nil { return nil, err } @@ -580,7 +580,7 @@ func decodeJWTToJSON(tokenBytes []byte) ([]byte, error) { return nil, fmt.Errorf("failed to decode payload: %v", err) } - combined := map[string]interface{}{ + combined := map[string]any{ "header": header, "payload": payload, } diff --git a/cli/attestation_snp_test.go b/cli/attestation_snp_test.go index 33fe8187..c5821d7a 100644 --- a/cli/attestation_snp_test.go +++ b/cli/attestation_snp_test.go @@ -369,15 +369,15 @@ func TestGetBase(t *testing.T) { func TestParseConfig(t *testing.T) { tempDir := t.TempDir() - validConfig := map[string]interface{}{ - "rootOfTrust": map[string]interface{}{ + validConfig := map[string]any{ + "rootOfTrust": map[string]any{ "product": "test_product", "cabundlePaths": []string{"test_path"}, "cabundles": []string{"test_bundle"}, "checkCrl": true, "disallowNetwork": true, }, - "policy": map[string]interface{}{ + "policy": map[string]any{ "minimumGuestSvn": 1, "policy": "1", "minimumBuild": 1, diff --git a/cli/attestation_test.go b/cli/attestation_test.go index 57629180..2379b439 100644 --- a/cli/attestation_test.go +++ b/cli/attestation_test.go @@ -526,7 +526,7 @@ func TestAttestationToJSON(t *testing.T) { require.NotNil(t, got) - var js map[string]interface{} + var js map[string]any err = json.Unmarshal(got, &js) assert.NoError(t, err) }) diff --git a/cli/datasets.go b/cli/datasets.go index a5de32ce..0f6bcf72 100644 --- a/cli/datasets.go +++ b/cli/datasets.go @@ -89,7 +89,7 @@ func (cli *CLI) NewDatasetsCmd() *cobra.Command { return cmd } -func decodeKey(b *pem.Block) (interface{}, error) { +func decodeKey(b *pem.Block) (any, error) { if b == nil { return nil, errors.New("error decoding key") } diff --git a/cli/keys.go b/cli/keys.go index 5b12acd4..16acaa36 100644 --- a/cli/keys.go +++ b/cli/keys.go @@ -96,7 +96,7 @@ func (cli *CLI) NewKeysCmd() *cobra.Command { } } -func generateAndWriteKeys(privKey interface{}, pubKeyBytes []byte, keyType string) error { +func generateAndWriteKeys(privKey any, pubKeyBytes []byte, keyType string) error { privFile, err := os.Create(privateKeyFile) if err != nil { return err diff --git a/cli/keys_test.go b/cli/keys_test.go index 32231414..0113655e 100644 --- a/cli/keys_test.go +++ b/cli/keys_test.go @@ -57,7 +57,7 @@ func TestGenerateAndWriteKeys(t *testing.T) { t.Fatalf("Failed to decode private key PEM") } - var privKey interface{} + var privKey any switch tt.keyType { case "rsa": privKey, err = x509.ParsePKCS1PrivateKey(privPem.Bytes) diff --git a/go.mod b/go.mod index db683100..54859237 100644 --- a/go.mod +++ b/go.mod @@ -1,6 +1,6 @@ module github.com/ultravioletrs/cocos -go 1.24.5 +go 1.25.0 require ( github.com/caarlos0/env/v11 v11.3.1 diff --git a/hal/linux/README.md b/hal/linux/README.md index a8492ef6..43310d60 100644 --- a/hal/linux/README.md +++ b/hal/linux/README.md @@ -10,7 +10,7 @@ HAL uses [Buildroot](https://buildroot.org/)'s [_External Tree_ mechanism](https git clone git@github.com:ultravioletrs/cocos.git git clone git@github.com:buildroot/buildroot.git cd buildroot -git checkout 2025.05-rc1 +git checkout 2025.08-rc3 make BR2_EXTERNAL=../cocos/hal/linux cocos_defconfig # Execute 'make menuconfig' only if you want to make additional configuration changes to Buildroot. make menuconfig diff --git a/hal/linux/package/agent/agent.mk b/hal/linux/package/agent/agent.mk index 8fa7c575..66a9eb87 100644 --- a/hal/linux/package/agent/agent.mk +++ b/hal/linux/package/agent/agent.mk @@ -24,4 +24,4 @@ define AGENT_INSTALL_INIT_SYSTEMD $(INSTALL) -D -m 0750 $(@D)/init/systemd/agent_start_script.sh $(TARGET_DIR)/cocos_init/agent_start_script.sh endef -$(eval $(golang-package)) +$(eval $(generic-package)) diff --git a/internal/server/mocks/server.go b/internal/server/mocks/server.go index 7b6438a6..06b3ffcd 100644 --- a/internal/server/mocks/server.go +++ b/internal/server/mocks/server.go @@ -1,11 +1,29 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks -import mock "github.com/stretchr/testify/mock" +import ( + mock "github.com/stretchr/testify/mock" +) + +// NewServer creates a new instance of Server. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewServer(t interface { + mock.TestingT + Cleanup(func()) +}) *Server { + mock := &Server{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} // Server is an autogenerated mock type for the Server type type Server struct { @@ -20,21 +38,20 @@ func (_m *Server) EXPECT() *Server_Expecter { return &Server_Expecter{mock: &_m.Mock} } -// Start provides a mock function with no fields -func (_m *Server) Start() error { - ret := _m.Called() +// Start provides a mock function for the type Server +func (_mock *Server) Start() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Start") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -55,8 +72,8 @@ func (_c *Server_Start_Call) Run(run func()) *Server_Start_Call { return _c } -func (_c *Server_Start_Call) Return(_a0 error) *Server_Start_Call { - _c.Call.Return(_a0) +func (_c *Server_Start_Call) Return(err error) *Server_Start_Call { + _c.Call.Return(err) return _c } @@ -65,21 +82,20 @@ func (_c *Server_Start_Call) RunAndReturn(run func() error) *Server_Start_Call { return _c } -// Stop provides a mock function with no fields -func (_m *Server) Stop() error { - ret := _m.Called() +// Stop provides a mock function for the type Server +func (_mock *Server) Stop() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Stop") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -100,8 +116,8 @@ func (_c *Server_Stop_Call) Run(run func()) *Server_Stop_Call { return _c } -func (_c *Server_Stop_Call) Return(_a0 error) *Server_Stop_Call { - _c.Call.Return(_a0) +func (_c *Server_Stop_Call) Return(err error) *Server_Stop_Call { + _c.Call.Return(err) return _c } @@ -109,17 +125,3 @@ func (_c *Server_Stop_Call) RunAndReturn(run func() error) *Server_Stop_Call { _c.Call.Return(run) return _c } - -// NewServer creates a new instance of Server. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewServer(t interface { - mock.TestingT - Cleanup(func()) -}) *Server { - mock := &Server{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/manager/attestation_policy_test.go b/manager/attestation_policy_test.go index b654975a..25120098 100644 --- a/manager/attestation_policy_test.go +++ b/manager/attestation_policy_test.go @@ -47,10 +47,10 @@ func TestFetchAttestationPolicy(t *testing.T) { testCases := []struct { name string computationId string - vmConfig interface{} + vmConfig any binaryBehavior string expectedError string - expectedResult map[string]interface{} + expectedResult map[string]any }{ { name: "Valid SEV-SNP configuration", @@ -136,7 +136,7 @@ func TestFetchAttestationPolicy(t *testing.T) { assert.NoError(t, err) assert.NotNil(t, result) - var attestationPolicy map[string]interface{} + var attestationPolicy map[string]any err = json.Unmarshal(result, &attestationPolicy) assert.NoError(t, err) diff --git a/manager/manager.pb.go b/manager/manager.pb.go index dede829b..ce16d7e2 100644 --- a/manager/manager.pb.go +++ b/manager/manager.pb.go @@ -3,7 +3,7 @@ // Code generated by protoc-gen-go. DO NOT EDIT. // versions: -// protoc-gen-go v1.36.5 +// protoc-gen-go v1.36.8 // protoc v5.29.0 // source: manager/manager.proto @@ -439,77 +439,45 @@ func (x *CVMInfoReq) GetId() string { var File_manager_manager_proto protoreflect.FileDescriptor -var file_manager_manager_proto_rawDesc = string([]byte{ - 0x0a, 0x15, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2f, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, - 0x72, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x07, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, - 0x1a, 0x1b, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, - 0x66, 0x2f, 0x65, 0x6d, 0x70, 0x74, 0x79, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x22, 0xbb, 0x02, - 0x0a, 0x09, 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x52, 0x65, 0x71, 0x12, 0x26, 0x0a, 0x0f, 0x61, - 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x6c, 0x6f, 0x67, 0x5f, 0x6c, 0x65, 0x76, 0x65, 0x6c, 0x18, 0x01, - 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x4c, 0x6f, 0x67, 0x4c, 0x65, - 0x76, 0x65, 0x6c, 0x12, 0x36, 0x0a, 0x18, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x76, 0x6d, - 0x5f, 0x73, 0x65, 0x72, 0x76, 0x65, 0x72, 0x5f, 0x63, 0x61, 0x5f, 0x63, 0x65, 0x72, 0x74, 0x18, - 0x02, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x14, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x43, 0x76, 0x6d, 0x53, - 0x65, 0x72, 0x76, 0x65, 0x72, 0x43, 0x61, 0x43, 0x65, 0x72, 0x74, 0x12, 0x2f, 0x0a, 0x14, 0x61, - 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x76, 0x6d, 0x5f, 0x63, 0x6c, 0x69, 0x65, 0x6e, 0x74, 0x5f, - 0x6b, 0x65, 0x79, 0x18, 0x03, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x11, 0x61, 0x67, 0x65, 0x6e, 0x74, - 0x43, 0x76, 0x6d, 0x43, 0x6c, 0x69, 0x65, 0x6e, 0x74, 0x4b, 0x65, 0x79, 0x12, 0x31, 0x0a, 0x15, - 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x76, 0x6d, 0x5f, 0x63, 0x6c, 0x69, 0x65, 0x6e, 0x74, - 0x5f, 0x63, 0x65, 0x72, 0x74, 0x18, 0x04, 0x20, 0x01, 0x28, 0x0c, 0x52, 0x12, 0x61, 0x67, 0x65, - 0x6e, 0x74, 0x43, 0x76, 0x6d, 0x43, 0x6c, 0x69, 0x65, 0x6e, 0x74, 0x43, 0x65, 0x72, 0x74, 0x12, - 0x2f, 0x0a, 0x14, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x76, 0x6d, 0x5f, 0x73, 0x65, 0x72, - 0x76, 0x65, 0x72, 0x5f, 0x75, 0x72, 0x6c, 0x18, 0x05, 0x20, 0x01, 0x28, 0x09, 0x52, 0x11, 0x61, - 0x67, 0x65, 0x6e, 0x74, 0x43, 0x76, 0x6d, 0x53, 0x65, 0x72, 0x76, 0x65, 0x72, 0x55, 0x72, 0x6c, - 0x12, 0x27, 0x0a, 0x10, 0x61, 0x67, 0x65, 0x6e, 0x74, 0x5f, 0x63, 0x76, 0x6d, 0x5f, 0x63, 0x61, - 0x5f, 0x75, 0x72, 0x6c, 0x18, 0x06, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0d, 0x61, 0x67, 0x65, 0x6e, - 0x74, 0x43, 0x76, 0x6d, 0x43, 0x61, 0x55, 0x72, 0x6c, 0x12, 0x10, 0x0a, 0x03, 0x74, 0x74, 0x6c, - 0x18, 0x07, 0x20, 0x01, 0x28, 0x09, 0x52, 0x03, 0x74, 0x74, 0x6c, 0x22, 0x49, 0x0a, 0x09, 0x43, - 0x72, 0x65, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x12, 0x25, 0x0a, 0x0e, 0x66, 0x6f, 0x72, 0x77, - 0x61, 0x72, 0x64, 0x65, 0x64, 0x5f, 0x70, 0x6f, 0x72, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, - 0x52, 0x0d, 0x66, 0x6f, 0x72, 0x77, 0x61, 0x72, 0x64, 0x65, 0x64, 0x50, 0x6f, 0x72, 0x74, 0x12, - 0x15, 0x0a, 0x06, 0x63, 0x76, 0x6d, 0x5f, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, - 0x05, 0x63, 0x76, 0x6d, 0x49, 0x64, 0x22, 0x22, 0x0a, 0x09, 0x52, 0x65, 0x6d, 0x6f, 0x76, 0x65, - 0x52, 0x65, 0x71, 0x12, 0x15, 0x0a, 0x06, 0x63, 0x76, 0x6d, 0x5f, 0x69, 0x64, 0x18, 0x01, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x05, 0x63, 0x76, 0x6d, 0x49, 0x64, 0x22, 0x3a, 0x0a, 0x14, 0x41, 0x74, - 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x50, 0x6f, 0x6c, 0x69, 0x63, 0x79, 0x52, - 0x65, 0x73, 0x12, 0x12, 0x0a, 0x04, 0x69, 0x6e, 0x66, 0x6f, 0x18, 0x01, 0x20, 0x01, 0x28, 0x0c, - 0x52, 0x04, 0x69, 0x6e, 0x66, 0x6f, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x02, 0x20, 0x01, - 0x28, 0x09, 0x52, 0x02, 0x69, 0x64, 0x22, 0xb3, 0x01, 0x0a, 0x0a, 0x43, 0x56, 0x4d, 0x49, 0x6e, - 0x66, 0x6f, 0x52, 0x65, 0x73, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, - 0x09, 0x52, 0x02, 0x69, 0x64, 0x12, 0x21, 0x0a, 0x0c, 0x6f, 0x76, 0x6d, 0x66, 0x5f, 0x76, 0x65, - 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x18, 0x02, 0x20, 0x01, 0x28, 0x09, 0x52, 0x0b, 0x6f, 0x76, 0x6d, - 0x66, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x12, 0x17, 0x0a, 0x07, 0x63, 0x70, 0x75, 0x5f, - 0x6e, 0x75, 0x6d, 0x18, 0x03, 0x20, 0x01, 0x28, 0x05, 0x52, 0x06, 0x63, 0x70, 0x75, 0x4e, 0x75, - 0x6d, 0x12, 0x19, 0x0a, 0x08, 0x63, 0x70, 0x75, 0x5f, 0x74, 0x79, 0x70, 0x65, 0x18, 0x04, 0x20, - 0x01, 0x28, 0x09, 0x52, 0x07, 0x63, 0x70, 0x75, 0x54, 0x79, 0x70, 0x65, 0x12, 0x1d, 0x0a, 0x0a, - 0x6b, 0x65, 0x72, 0x6e, 0x65, 0x6c, 0x5f, 0x63, 0x6d, 0x64, 0x18, 0x05, 0x20, 0x01, 0x28, 0x09, - 0x52, 0x09, 0x6b, 0x65, 0x72, 0x6e, 0x65, 0x6c, 0x43, 0x6d, 0x64, 0x12, 0x1f, 0x0a, 0x0b, 0x65, - 0x6f, 0x73, 0x5f, 0x76, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x18, 0x06, 0x20, 0x01, 0x28, 0x09, - 0x52, 0x0a, 0x65, 0x6f, 0x73, 0x56, 0x65, 0x72, 0x73, 0x69, 0x6f, 0x6e, 0x22, 0x26, 0x0a, 0x14, - 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x50, 0x6f, 0x6c, 0x69, 0x63, - 0x79, 0x52, 0x65, 0x71, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, - 0x52, 0x02, 0x69, 0x64, 0x22, 0x1c, 0x0a, 0x0a, 0x43, 0x56, 0x4d, 0x49, 0x6e, 0x66, 0x6f, 0x52, - 0x65, 0x71, 0x12, 0x0e, 0x0a, 0x02, 0x69, 0x64, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52, 0x02, - 0x69, 0x64, 0x32, 0x8c, 0x02, 0x0a, 0x0e, 0x4d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x53, 0x65, - 0x72, 0x76, 0x69, 0x63, 0x65, 0x12, 0x34, 0x0a, 0x08, 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x56, - 0x6d, 0x12, 0x12, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x43, 0x72, 0x65, 0x61, - 0x74, 0x65, 0x52, 0x65, 0x71, 0x1a, 0x12, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, - 0x43, 0x72, 0x65, 0x61, 0x74, 0x65, 0x52, 0x65, 0x73, 0x22, 0x00, 0x12, 0x38, 0x0a, 0x08, 0x52, - 0x65, 0x6d, 0x6f, 0x76, 0x65, 0x56, 0x6d, 0x12, 0x12, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, - 0x72, 0x2e, 0x52, 0x65, 0x6d, 0x6f, 0x76, 0x65, 0x52, 0x65, 0x71, 0x1a, 0x16, 0x2e, 0x67, 0x6f, - 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, 0x6d, - 0x70, 0x74, 0x79, 0x22, 0x00, 0x12, 0x35, 0x0a, 0x07, 0x43, 0x56, 0x4d, 0x49, 0x6e, 0x66, 0x6f, - 0x12, 0x13, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x43, 0x56, 0x4d, 0x49, 0x6e, - 0x66, 0x6f, 0x52, 0x65, 0x71, 0x1a, 0x13, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, - 0x43, 0x56, 0x4d, 0x49, 0x6e, 0x66, 0x6f, 0x52, 0x65, 0x73, 0x22, 0x00, 0x12, 0x53, 0x0a, 0x11, - 0x41, 0x74, 0x74, 0x65, 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x50, 0x6f, 0x6c, 0x69, 0x63, - 0x79, 0x12, 0x1d, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x41, 0x74, 0x74, 0x65, - 0x73, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x50, 0x6f, 0x6c, 0x69, 0x63, 0x79, 0x52, 0x65, 0x71, - 0x1a, 0x1d, 0x2e, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x2e, 0x41, 0x74, 0x74, 0x65, 0x73, - 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, 0x50, 0x6f, 0x6c, 0x69, 0x63, 0x79, 0x52, 0x65, 0x73, 0x22, - 0x00, 0x42, 0x0b, 0x5a, 0x09, 0x2e, 0x2f, 0x6d, 0x61, 0x6e, 0x61, 0x67, 0x65, 0x72, 0x62, 0x06, - 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33, -}) +const file_manager_manager_proto_rawDesc = "" + + "\n" + + "\x15manager/manager.proto\x12\amanager\x1a\x1bgoogle/protobuf/empty.proto\"\xbb\x02\n" + + "\tCreateReq\x12&\n" + + "\x0fagent_log_level\x18\x01 \x01(\tR\ragentLogLevel\x126\n" + + "\x18agent_cvm_server_ca_cert\x18\x02 \x01(\fR\x14agentCvmServerCaCert\x12/\n" + + "\x14agent_cvm_client_key\x18\x03 \x01(\fR\x11agentCvmClientKey\x121\n" + + "\x15agent_cvm_client_cert\x18\x04 \x01(\fR\x12agentCvmClientCert\x12/\n" + + "\x14agent_cvm_server_url\x18\x05 \x01(\tR\x11agentCvmServerUrl\x12'\n" + + "\x10agent_cvm_ca_url\x18\x06 \x01(\tR\ragentCvmCaUrl\x12\x10\n" + + "\x03ttl\x18\a \x01(\tR\x03ttl\"I\n" + + "\tCreateRes\x12%\n" + + "\x0eforwarded_port\x18\x01 \x01(\tR\rforwardedPort\x12\x15\n" + + "\x06cvm_id\x18\x02 \x01(\tR\x05cvmId\"\"\n" + + "\tRemoveReq\x12\x15\n" + + "\x06cvm_id\x18\x01 \x01(\tR\x05cvmId\":\n" + + "\x14AttestationPolicyRes\x12\x12\n" + + "\x04info\x18\x01 \x01(\fR\x04info\x12\x0e\n" + + "\x02id\x18\x02 \x01(\tR\x02id\"\xb3\x01\n" + + "\n" + + "CVMInfoRes\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id\x12!\n" + + "\fovmf_version\x18\x02 \x01(\tR\vovmfVersion\x12\x17\n" + + "\acpu_num\x18\x03 \x01(\x05R\x06cpuNum\x12\x19\n" + + "\bcpu_type\x18\x04 \x01(\tR\acpuType\x12\x1d\n" + + "\n" + + "kernel_cmd\x18\x05 \x01(\tR\tkernelCmd\x12\x1f\n" + + "\veos_version\x18\x06 \x01(\tR\n" + + "eosVersion\"&\n" + + "\x14AttestationPolicyReq\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id\"\x1c\n" + + "\n" + + "CVMInfoReq\x12\x0e\n" + + "\x02id\x18\x01 \x01(\tR\x02id2\x8c\x02\n" + + "\x0eManagerService\x124\n" + + "\bCreateVm\x12\x12.manager.CreateReq\x1a\x12.manager.CreateRes\"\x00\x128\n" + + "\bRemoveVm\x12\x12.manager.RemoveReq\x1a\x16.google.protobuf.Empty\"\x00\x125\n" + + "\aCVMInfo\x12\x13.manager.CVMInfoReq\x1a\x13.manager.CVMInfoRes\"\x00\x12S\n" + + "\x11AttestationPolicy\x12\x1d.manager.AttestationPolicyReq\x1a\x1d.manager.AttestationPolicyRes\"\x00B\vZ\t./managerb\x06proto3" var ( file_manager_manager_proto_rawDescOnce sync.Once diff --git a/manager/mocks/manager_service_client.go b/manager/mocks/managerserviceclient.go similarity index 55% rename from manager/mocks/manager_service_client.go rename to manager/mocks/managerserviceclient.go index 708a24f4..cf3b845c 100644 --- a/manager/mocks/manager_service_client.go +++ b/manager/mocks/managerserviceclient.go @@ -1,21 +1,35 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - context "context" - - grpc "google.golang.org/grpc" - emptypb "google.golang.org/protobuf/types/known/emptypb" - - manager "github.com/ultravioletrs/cocos/manager" + "context" mock "github.com/stretchr/testify/mock" + "github.com/ultravioletrs/cocos/manager" + "google.golang.org/grpc" + "google.golang.org/protobuf/types/known/emptypb" ) +// NewManagerServiceClient creates a new instance of ManagerServiceClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewManagerServiceClient(t interface { + mock.TestingT + Cleanup(func()) +}) *ManagerServiceClient { + mock := &ManagerServiceClient{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // ManagerServiceClient is an autogenerated mock type for the ManagerServiceClient type type ManagerServiceClient struct { mock.Mock @@ -29,8 +43,9 @@ func (_m *ManagerServiceClient) EXPECT() *ManagerServiceClient_Expecter { return &ManagerServiceClient_Expecter{mock: &_m.Mock} } -// AttestationPolicy provides a mock function with given fields: ctx, in, opts -func (_m *ManagerServiceClient) AttestationPolicy(ctx context.Context, in *manager.AttestationPolicyReq, opts ...grpc.CallOption) (*manager.AttestationPolicyRes, error) { +// AttestationPolicy provides a mock function for the type ManagerServiceClient +func (_mock *ManagerServiceClient) AttestationPolicy(ctx context.Context, in *manager.AttestationPolicyReq, opts ...grpc.CallOption) (*manager.AttestationPolicyRes, error) { + // grpc.CallOption _va := make([]interface{}, len(opts)) for _i := range opts { _va[_i] = opts[_i] @@ -38,7 +53,7 @@ func (_m *ManagerServiceClient) AttestationPolicy(ctx context.Context, in *manag var _ca []interface{} _ca = append(_ca, ctx, in) _ca = append(_ca, _va...) - ret := _m.Called(_ca...) + ret := _mock.Called(_ca...) if len(ret) == 0 { panic("no return value specified for AttestationPolicy") @@ -46,23 +61,21 @@ func (_m *ManagerServiceClient) AttestationPolicy(ctx context.Context, in *manag var r0 *manager.AttestationPolicyRes var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) (*manager.AttestationPolicyRes, error)); ok { - return rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) (*manager.AttestationPolicyRes, error)); ok { + return returnFunc(ctx, in, opts...) } - if rf, ok := ret.Get(0).(func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) *manager.AttestationPolicyRes); ok { - r0 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) *manager.AttestationPolicyRes); ok { + r0 = returnFunc(ctx, in, opts...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*manager.AttestationPolicyRes) } } - - if rf, ok := ret.Get(1).(func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) error); ok { - r1 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(1).(func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) error); ok { + r1 = returnFunc(ctx, in, opts...) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -82,29 +95,44 @@ func (_e *ManagerServiceClient_Expecter) AttestationPolicy(ctx interface{}, in i func (_c *ManagerServiceClient_AttestationPolicy_Call) Run(run func(ctx context.Context, in *manager.AttestationPolicyReq, opts ...grpc.CallOption)) *ManagerServiceClient_AttestationPolicy_Call { _c.Call.Run(func(args mock.Arguments) { + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *manager.AttestationPolicyReq + if args[1] != nil { + arg1 = args[1].(*manager.AttestationPolicyReq) + } + var arg2 []grpc.CallOption variadicArgs := make([]grpc.CallOption, len(args)-2) for i, a := range args[2:] { if a != nil { variadicArgs[i] = a.(grpc.CallOption) } } - run(args[0].(context.Context), args[1].(*manager.AttestationPolicyReq), variadicArgs...) + arg2 = variadicArgs + run( + arg0, + arg1, + arg2..., + ) }) return _c } -func (_c *ManagerServiceClient_AttestationPolicy_Call) Return(_a0 *manager.AttestationPolicyRes, _a1 error) *ManagerServiceClient_AttestationPolicy_Call { - _c.Call.Return(_a0, _a1) +func (_c *ManagerServiceClient_AttestationPolicy_Call) Return(attestationPolicyRes *manager.AttestationPolicyRes, err error) *ManagerServiceClient_AttestationPolicy_Call { + _c.Call.Return(attestationPolicyRes, err) return _c } -func (_c *ManagerServiceClient_AttestationPolicy_Call) RunAndReturn(run func(context.Context, *manager.AttestationPolicyReq, ...grpc.CallOption) (*manager.AttestationPolicyRes, error)) *ManagerServiceClient_AttestationPolicy_Call { +func (_c *ManagerServiceClient_AttestationPolicy_Call) RunAndReturn(run func(ctx context.Context, in *manager.AttestationPolicyReq, opts ...grpc.CallOption) (*manager.AttestationPolicyRes, error)) *ManagerServiceClient_AttestationPolicy_Call { _c.Call.Return(run) return _c } -// CVMInfo provides a mock function with given fields: ctx, in, opts -func (_m *ManagerServiceClient) CVMInfo(ctx context.Context, in *manager.CVMInfoReq, opts ...grpc.CallOption) (*manager.CVMInfoRes, error) { +// CVMInfo provides a mock function for the type ManagerServiceClient +func (_mock *ManagerServiceClient) CVMInfo(ctx context.Context, in *manager.CVMInfoReq, opts ...grpc.CallOption) (*manager.CVMInfoRes, error) { + // grpc.CallOption _va := make([]interface{}, len(opts)) for _i := range opts { _va[_i] = opts[_i] @@ -112,7 +140,7 @@ func (_m *ManagerServiceClient) CVMInfo(ctx context.Context, in *manager.CVMInfo var _ca []interface{} _ca = append(_ca, ctx, in) _ca = append(_ca, _va...) - ret := _m.Called(_ca...) + ret := _mock.Called(_ca...) if len(ret) == 0 { panic("no return value specified for CVMInfo") @@ -120,23 +148,21 @@ func (_m *ManagerServiceClient) CVMInfo(ctx context.Context, in *manager.CVMInfo var r0 *manager.CVMInfoRes var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) (*manager.CVMInfoRes, error)); ok { - return rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) (*manager.CVMInfoRes, error)); ok { + return returnFunc(ctx, in, opts...) } - if rf, ok := ret.Get(0).(func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) *manager.CVMInfoRes); ok { - r0 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) *manager.CVMInfoRes); ok { + r0 = returnFunc(ctx, in, opts...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*manager.CVMInfoRes) } } - - if rf, ok := ret.Get(1).(func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) error); ok { - r1 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(1).(func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) error); ok { + r1 = returnFunc(ctx, in, opts...) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -156,29 +182,44 @@ func (_e *ManagerServiceClient_Expecter) CVMInfo(ctx interface{}, in interface{} func (_c *ManagerServiceClient_CVMInfo_Call) Run(run func(ctx context.Context, in *manager.CVMInfoReq, opts ...grpc.CallOption)) *ManagerServiceClient_CVMInfo_Call { _c.Call.Run(func(args mock.Arguments) { + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *manager.CVMInfoReq + if args[1] != nil { + arg1 = args[1].(*manager.CVMInfoReq) + } + var arg2 []grpc.CallOption variadicArgs := make([]grpc.CallOption, len(args)-2) for i, a := range args[2:] { if a != nil { variadicArgs[i] = a.(grpc.CallOption) } } - run(args[0].(context.Context), args[1].(*manager.CVMInfoReq), variadicArgs...) + arg2 = variadicArgs + run( + arg0, + arg1, + arg2..., + ) }) return _c } -func (_c *ManagerServiceClient_CVMInfo_Call) Return(_a0 *manager.CVMInfoRes, _a1 error) *ManagerServiceClient_CVMInfo_Call { - _c.Call.Return(_a0, _a1) +func (_c *ManagerServiceClient_CVMInfo_Call) Return(cVMInfoRes *manager.CVMInfoRes, err error) *ManagerServiceClient_CVMInfo_Call { + _c.Call.Return(cVMInfoRes, err) return _c } -func (_c *ManagerServiceClient_CVMInfo_Call) RunAndReturn(run func(context.Context, *manager.CVMInfoReq, ...grpc.CallOption) (*manager.CVMInfoRes, error)) *ManagerServiceClient_CVMInfo_Call { +func (_c *ManagerServiceClient_CVMInfo_Call) RunAndReturn(run func(ctx context.Context, in *manager.CVMInfoReq, opts ...grpc.CallOption) (*manager.CVMInfoRes, error)) *ManagerServiceClient_CVMInfo_Call { _c.Call.Return(run) return _c } -// CreateVm provides a mock function with given fields: ctx, in, opts -func (_m *ManagerServiceClient) CreateVm(ctx context.Context, in *manager.CreateReq, opts ...grpc.CallOption) (*manager.CreateRes, error) { +// CreateVm provides a mock function for the type ManagerServiceClient +func (_mock *ManagerServiceClient) CreateVm(ctx context.Context, in *manager.CreateReq, opts ...grpc.CallOption) (*manager.CreateRes, error) { + // grpc.CallOption _va := make([]interface{}, len(opts)) for _i := range opts { _va[_i] = opts[_i] @@ -186,7 +227,7 @@ func (_m *ManagerServiceClient) CreateVm(ctx context.Context, in *manager.Create var _ca []interface{} _ca = append(_ca, ctx, in) _ca = append(_ca, _va...) - ret := _m.Called(_ca...) + ret := _mock.Called(_ca...) if len(ret) == 0 { panic("no return value specified for CreateVm") @@ -194,23 +235,21 @@ func (_m *ManagerServiceClient) CreateVm(ctx context.Context, in *manager.Create var r0 *manager.CreateRes var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *manager.CreateReq, ...grpc.CallOption) (*manager.CreateRes, error)); ok { - return rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.CreateReq, ...grpc.CallOption) (*manager.CreateRes, error)); ok { + return returnFunc(ctx, in, opts...) } - if rf, ok := ret.Get(0).(func(context.Context, *manager.CreateReq, ...grpc.CallOption) *manager.CreateRes); ok { - r0 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.CreateReq, ...grpc.CallOption) *manager.CreateRes); ok { + r0 = returnFunc(ctx, in, opts...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*manager.CreateRes) } } - - if rf, ok := ret.Get(1).(func(context.Context, *manager.CreateReq, ...grpc.CallOption) error); ok { - r1 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(1).(func(context.Context, *manager.CreateReq, ...grpc.CallOption) error); ok { + r1 = returnFunc(ctx, in, opts...) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -230,29 +269,44 @@ func (_e *ManagerServiceClient_Expecter) CreateVm(ctx interface{}, in interface{ func (_c *ManagerServiceClient_CreateVm_Call) Run(run func(ctx context.Context, in *manager.CreateReq, opts ...grpc.CallOption)) *ManagerServiceClient_CreateVm_Call { _c.Call.Run(func(args mock.Arguments) { + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *manager.CreateReq + if args[1] != nil { + arg1 = args[1].(*manager.CreateReq) + } + var arg2 []grpc.CallOption variadicArgs := make([]grpc.CallOption, len(args)-2) for i, a := range args[2:] { if a != nil { variadicArgs[i] = a.(grpc.CallOption) } } - run(args[0].(context.Context), args[1].(*manager.CreateReq), variadicArgs...) + arg2 = variadicArgs + run( + arg0, + arg1, + arg2..., + ) }) return _c } -func (_c *ManagerServiceClient_CreateVm_Call) Return(_a0 *manager.CreateRes, _a1 error) *ManagerServiceClient_CreateVm_Call { - _c.Call.Return(_a0, _a1) +func (_c *ManagerServiceClient_CreateVm_Call) Return(createRes *manager.CreateRes, err error) *ManagerServiceClient_CreateVm_Call { + _c.Call.Return(createRes, err) return _c } -func (_c *ManagerServiceClient_CreateVm_Call) RunAndReturn(run func(context.Context, *manager.CreateReq, ...grpc.CallOption) (*manager.CreateRes, error)) *ManagerServiceClient_CreateVm_Call { +func (_c *ManagerServiceClient_CreateVm_Call) RunAndReturn(run func(ctx context.Context, in *manager.CreateReq, opts ...grpc.CallOption) (*manager.CreateRes, error)) *ManagerServiceClient_CreateVm_Call { _c.Call.Return(run) return _c } -// RemoveVm provides a mock function with given fields: ctx, in, opts -func (_m *ManagerServiceClient) RemoveVm(ctx context.Context, in *manager.RemoveReq, opts ...grpc.CallOption) (*emptypb.Empty, error) { +// RemoveVm provides a mock function for the type ManagerServiceClient +func (_mock *ManagerServiceClient) RemoveVm(ctx context.Context, in *manager.RemoveReq, opts ...grpc.CallOption) (*emptypb.Empty, error) { + // grpc.CallOption _va := make([]interface{}, len(opts)) for _i := range opts { _va[_i] = opts[_i] @@ -260,7 +314,7 @@ func (_m *ManagerServiceClient) RemoveVm(ctx context.Context, in *manager.Remove var _ca []interface{} _ca = append(_ca, ctx, in) _ca = append(_ca, _va...) - ret := _m.Called(_ca...) + ret := _mock.Called(_ca...) if len(ret) == 0 { panic("no return value specified for RemoveVm") @@ -268,23 +322,21 @@ func (_m *ManagerServiceClient) RemoveVm(ctx context.Context, in *manager.Remove var r0 *emptypb.Empty var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *manager.RemoveReq, ...grpc.CallOption) (*emptypb.Empty, error)); ok { - return rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.RemoveReq, ...grpc.CallOption) (*emptypb.Empty, error)); ok { + return returnFunc(ctx, in, opts...) } - if rf, ok := ret.Get(0).(func(context.Context, *manager.RemoveReq, ...grpc.CallOption) *emptypb.Empty); ok { - r0 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.RemoveReq, ...grpc.CallOption) *emptypb.Empty); ok { + r0 = returnFunc(ctx, in, opts...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*emptypb.Empty) } } - - if rf, ok := ret.Get(1).(func(context.Context, *manager.RemoveReq, ...grpc.CallOption) error); ok { - r1 = rf(ctx, in, opts...) + if returnFunc, ok := ret.Get(1).(func(context.Context, *manager.RemoveReq, ...grpc.CallOption) error); ok { + r1 = returnFunc(ctx, in, opts...) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -304,37 +356,37 @@ func (_e *ManagerServiceClient_Expecter) RemoveVm(ctx interface{}, in interface{ func (_c *ManagerServiceClient_RemoveVm_Call) Run(run func(ctx context.Context, in *manager.RemoveReq, opts ...grpc.CallOption)) *ManagerServiceClient_RemoveVm_Call { _c.Call.Run(func(args mock.Arguments) { + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *manager.RemoveReq + if args[1] != nil { + arg1 = args[1].(*manager.RemoveReq) + } + var arg2 []grpc.CallOption variadicArgs := make([]grpc.CallOption, len(args)-2) for i, a := range args[2:] { if a != nil { variadicArgs[i] = a.(grpc.CallOption) } } - run(args[0].(context.Context), args[1].(*manager.RemoveReq), variadicArgs...) + arg2 = variadicArgs + run( + arg0, + arg1, + arg2..., + ) }) return _c } -func (_c *ManagerServiceClient_RemoveVm_Call) Return(_a0 *emptypb.Empty, _a1 error) *ManagerServiceClient_RemoveVm_Call { - _c.Call.Return(_a0, _a1) +func (_c *ManagerServiceClient_RemoveVm_Call) Return(empty *emptypb.Empty, err error) *ManagerServiceClient_RemoveVm_Call { + _c.Call.Return(empty, err) return _c } -func (_c *ManagerServiceClient_RemoveVm_Call) RunAndReturn(run func(context.Context, *manager.RemoveReq, ...grpc.CallOption) (*emptypb.Empty, error)) *ManagerServiceClient_RemoveVm_Call { +func (_c *ManagerServiceClient_RemoveVm_Call) RunAndReturn(run func(ctx context.Context, in *manager.RemoveReq, opts ...grpc.CallOption) (*emptypb.Empty, error)) *ManagerServiceClient_RemoveVm_Call { _c.Call.Return(run) return _c } - -// NewManagerServiceClient creates a new instance of ManagerServiceClient. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewManagerServiceClient(t interface { - mock.TestingT - Cleanup(func()) -}) *ManagerServiceClient { - mock := &ManagerServiceClient{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/manager/mocks/service.go b/manager/mocks/service.go index 4b5689cd..0cc21ea2 100644 --- a/manager/mocks/service.go +++ b/manager/mocks/service.go @@ -1,17 +1,33 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - context "context" + "context" mock "github.com/stretchr/testify/mock" - manager "github.com/ultravioletrs/cocos/manager" + "github.com/ultravioletrs/cocos/manager" ) +// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewService(t interface { + mock.TestingT + Cleanup(func()) +}) *Service { + mock := &Service{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Service is an autogenerated mock type for the Service type type Service struct { mock.Mock @@ -25,9 +41,9 @@ func (_m *Service) EXPECT() *Service_Expecter { return &Service_Expecter{mock: &_m.Mock} } -// CreateVM provides a mock function with given fields: ctx, req -func (_m *Service) CreateVM(ctx context.Context, req *manager.CreateReq) (string, string, error) { - ret := _m.Called(ctx, req) +// CreateVM provides a mock function for the type Service +func (_mock *Service) CreateVM(ctx context.Context, req *manager.CreateReq) (string, string, error) { + ret := _mock.Called(ctx, req) if len(ret) == 0 { panic("no return value specified for CreateVM") @@ -36,27 +52,24 @@ func (_m *Service) CreateVM(ctx context.Context, req *manager.CreateReq) (string var r0 string var r1 string var r2 error - if rf, ok := ret.Get(0).(func(context.Context, *manager.CreateReq) (string, string, error)); ok { - return rf(ctx, req) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.CreateReq) (string, string, error)); ok { + return returnFunc(ctx, req) } - if rf, ok := ret.Get(0).(func(context.Context, *manager.CreateReq) string); ok { - r0 = rf(ctx, req) + if returnFunc, ok := ret.Get(0).(func(context.Context, *manager.CreateReq) string); ok { + r0 = returnFunc(ctx, req) } else { r0 = ret.Get(0).(string) } - - if rf, ok := ret.Get(1).(func(context.Context, *manager.CreateReq) string); ok { - r1 = rf(ctx, req) + if returnFunc, ok := ret.Get(1).(func(context.Context, *manager.CreateReq) string); ok { + r1 = returnFunc(ctx, req) } else { r1 = ret.Get(1).(string) } - - if rf, ok := ret.Get(2).(func(context.Context, *manager.CreateReq) error); ok { - r2 = rf(ctx, req) + if returnFunc, ok := ret.Get(2).(func(context.Context, *manager.CreateReq) error); ok { + r2 = returnFunc(ctx, req) } else { r2 = ret.Error(2) } - return r0, r1, r2 } @@ -74,24 +87,35 @@ func (_e *Service_Expecter) CreateVM(ctx interface{}, req interface{}) *Service_ func (_c *Service_CreateVM_Call) Run(run func(ctx context.Context, req *manager.CreateReq)) *Service_CreateVM_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(*manager.CreateReq)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *manager.CreateReq + if args[1] != nil { + arg1 = args[1].(*manager.CreateReq) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_CreateVM_Call) Return(_a0 string, _a1 string, _a2 error) *Service_CreateVM_Call { - _c.Call.Return(_a0, _a1, _a2) +func (_c *Service_CreateVM_Call) Return(s string, s1 string, err error) *Service_CreateVM_Call { + _c.Call.Return(s, s1, err) return _c } -func (_c *Service_CreateVM_Call) RunAndReturn(run func(context.Context, *manager.CreateReq) (string, string, error)) *Service_CreateVM_Call { +func (_c *Service_CreateVM_Call) RunAndReturn(run func(ctx context.Context, req *manager.CreateReq) (string, string, error)) *Service_CreateVM_Call { _c.Call.Return(run) return _c } -// FetchAttestationPolicy provides a mock function with given fields: ctx, computationID -func (_m *Service) FetchAttestationPolicy(ctx context.Context, computationID string) ([]byte, error) { - ret := _m.Called(ctx, computationID) +// FetchAttestationPolicy provides a mock function for the type Service +func (_mock *Service) FetchAttestationPolicy(ctx context.Context, computationID string) ([]byte, error) { + ret := _mock.Called(ctx, computationID) if len(ret) == 0 { panic("no return value specified for FetchAttestationPolicy") @@ -99,23 +123,21 @@ func (_m *Service) FetchAttestationPolicy(ctx context.Context, computationID str var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func(context.Context, string) ([]byte, error)); ok { - return rf(ctx, computationID) + if returnFunc, ok := ret.Get(0).(func(context.Context, string) ([]byte, error)); ok { + return returnFunc(ctx, computationID) } - if rf, ok := ret.Get(0).(func(context.Context, string) []byte); ok { - r0 = rf(ctx, computationID) + if returnFunc, ok := ret.Get(0).(func(context.Context, string) []byte); ok { + r0 = returnFunc(ctx, computationID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(context.Context, string) error); ok { - r1 = rf(ctx, computationID) + if returnFunc, ok := ret.Get(1).(func(context.Context, string) error); ok { + r1 = returnFunc(ctx, computationID) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -133,36 +155,46 @@ func (_e *Service_Expecter) FetchAttestationPolicy(ctx interface{}, computationI func (_c *Service_FetchAttestationPolicy_Call) Run(run func(ctx context.Context, computationID string)) *Service_FetchAttestationPolicy_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(string)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 string + if args[1] != nil { + arg1 = args[1].(string) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_FetchAttestationPolicy_Call) Return(_a0 []byte, _a1 error) *Service_FetchAttestationPolicy_Call { - _c.Call.Return(_a0, _a1) +func (_c *Service_FetchAttestationPolicy_Call) Return(bytes []byte, err error) *Service_FetchAttestationPolicy_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Service_FetchAttestationPolicy_Call) RunAndReturn(run func(context.Context, string) ([]byte, error)) *Service_FetchAttestationPolicy_Call { +func (_c *Service_FetchAttestationPolicy_Call) RunAndReturn(run func(ctx context.Context, computationID string) ([]byte, error)) *Service_FetchAttestationPolicy_Call { _c.Call.Return(run) return _c } -// RemoveVM provides a mock function with given fields: ctx, computationID -func (_m *Service) RemoveVM(ctx context.Context, computationID string) error { - ret := _m.Called(ctx, computationID) +// RemoveVM provides a mock function for the type Service +func (_mock *Service) RemoveVM(ctx context.Context, computationID string) error { + ret := _mock.Called(ctx, computationID) if len(ret) == 0 { panic("no return value specified for RemoveVM") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, string) error); ok { - r0 = rf(ctx, computationID) + if returnFunc, ok := ret.Get(0).(func(context.Context, string) error); ok { + r0 = returnFunc(ctx, computationID) } else { r0 = ret.Error(0) } - return r0 } @@ -180,24 +212,35 @@ func (_e *Service_Expecter) RemoveVM(ctx interface{}, computationID interface{}) func (_c *Service_RemoveVM_Call) Run(run func(ctx context.Context, computationID string)) *Service_RemoveVM_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(string)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 string + if args[1] != nil { + arg1 = args[1].(string) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Service_RemoveVM_Call) Return(_a0 error) *Service_RemoveVM_Call { - _c.Call.Return(_a0) +func (_c *Service_RemoveVM_Call) Return(err error) *Service_RemoveVM_Call { + _c.Call.Return(err) return _c } -func (_c *Service_RemoveVM_Call) RunAndReturn(run func(context.Context, string) error) *Service_RemoveVM_Call { +func (_c *Service_RemoveVM_Call) RunAndReturn(run func(ctx context.Context, computationID string) error) *Service_RemoveVM_Call { _c.Call.Return(run) return _c } -// ReturnCVMInfo provides a mock function with given fields: ctx -func (_m *Service) ReturnCVMInfo(ctx context.Context) (string, int, string, string) { - ret := _m.Called(ctx) +// ReturnCVMInfo provides a mock function for the type Service +func (_mock *Service) ReturnCVMInfo(ctx context.Context) (string, int, string, string) { + ret := _mock.Called(ctx) if len(ret) == 0 { panic("no return value specified for ReturnCVMInfo") @@ -207,33 +250,29 @@ func (_m *Service) ReturnCVMInfo(ctx context.Context) (string, int, string, stri var r1 int var r2 string var r3 string - if rf, ok := ret.Get(0).(func(context.Context) (string, int, string, string)); ok { - return rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) (string, int, string, string)); ok { + return returnFunc(ctx) } - if rf, ok := ret.Get(0).(func(context.Context) string); ok { - r0 = rf(ctx) + if returnFunc, ok := ret.Get(0).(func(context.Context) string); ok { + r0 = returnFunc(ctx) } else { r0 = ret.Get(0).(string) } - - if rf, ok := ret.Get(1).(func(context.Context) int); ok { - r1 = rf(ctx) + if returnFunc, ok := ret.Get(1).(func(context.Context) int); ok { + r1 = returnFunc(ctx) } else { r1 = ret.Get(1).(int) } - - if rf, ok := ret.Get(2).(func(context.Context) string); ok { - r2 = rf(ctx) + if returnFunc, ok := ret.Get(2).(func(context.Context) string); ok { + r2 = returnFunc(ctx) } else { r2 = ret.Get(2).(string) } - - if rf, ok := ret.Get(3).(func(context.Context) string); ok { - r3 = rf(ctx) + if returnFunc, ok := ret.Get(3).(func(context.Context) string); ok { + r3 = returnFunc(ctx) } else { r3 = ret.Get(3).(string) } - return r0, r1, r2, r3 } @@ -250,36 +289,41 @@ func (_e *Service_Expecter) ReturnCVMInfo(ctx interface{}) *Service_ReturnCVMInf func (_c *Service_ReturnCVMInfo_Call) Run(run func(ctx context.Context)) *Service_ReturnCVMInfo_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + run( + arg0, + ) }) return _c } -func (_c *Service_ReturnCVMInfo_Call) Return(_a0 string, _a1 int, _a2 string, _a3 string) *Service_ReturnCVMInfo_Call { - _c.Call.Return(_a0, _a1, _a2, _a3) +func (_c *Service_ReturnCVMInfo_Call) Return(s string, n int, s1 string, s2 string) *Service_ReturnCVMInfo_Call { + _c.Call.Return(s, n, s1, s2) return _c } -func (_c *Service_ReturnCVMInfo_Call) RunAndReturn(run func(context.Context) (string, int, string, string)) *Service_ReturnCVMInfo_Call { +func (_c *Service_ReturnCVMInfo_Call) RunAndReturn(run func(ctx context.Context) (string, int, string, string)) *Service_ReturnCVMInfo_Call { _c.Call.Return(run) return _c } -// Shutdown provides a mock function with no fields -func (_m *Service) Shutdown() error { - ret := _m.Called() +// Shutdown provides a mock function for the type Service +func (_mock *Service) Shutdown() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Shutdown") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -300,8 +344,8 @@ func (_c *Service_Shutdown_Call) Run(run func()) *Service_Shutdown_Call { return _c } -func (_c *Service_Shutdown_Call) Return(_a0 error) *Service_Shutdown_Call { - _c.Call.Return(_a0) +func (_c *Service_Shutdown_Call) Return(err error) *Service_Shutdown_Call { + _c.Call.Return(err) return _c } @@ -309,17 +353,3 @@ func (_c *Service_Shutdown_Call) RunAndReturn(run func() error) *Service_Shutdow _c.Call.Return(run) return _c } - -// NewService creates a new instance of Service. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewService(t interface { - mock.TestingT - Cleanup(func()) -}) *Service { - mock := &Service{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/manager/qemu/mocks/persistence.go b/manager/qemu/mocks/persistence.go index 124aa689..e07590fe 100644 --- a/manager/qemu/mocks/persistence.go +++ b/manager/qemu/mocks/persistence.go @@ -1,15 +1,31 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( mock "github.com/stretchr/testify/mock" - qemu "github.com/ultravioletrs/cocos/manager/qemu" + "github.com/ultravioletrs/cocos/manager/qemu" ) +// NewPersistence creates a new instance of Persistence. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewPersistence(t interface { + mock.TestingT + Cleanup(func()) +}) *Persistence { + mock := &Persistence{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Persistence is an autogenerated mock type for the Persistence type type Persistence struct { mock.Mock @@ -23,21 +39,20 @@ func (_m *Persistence) EXPECT() *Persistence_Expecter { return &Persistence_Expecter{mock: &_m.Mock} } -// DeleteVM provides a mock function with given fields: id -func (_m *Persistence) DeleteVM(id string) error { - ret := _m.Called(id) +// DeleteVM provides a mock function for the type Persistence +func (_mock *Persistence) DeleteVM(id string) error { + ret := _mock.Called(id) if len(ret) == 0 { panic("no return value specified for DeleteVM") } var r0 error - if rf, ok := ret.Get(0).(func(string) error); ok { - r0 = rf(id) + if returnFunc, ok := ret.Get(0).(func(string) error); ok { + r0 = returnFunc(id) } else { r0 = ret.Error(0) } - return r0 } @@ -54,24 +69,30 @@ func (_e *Persistence_Expecter) DeleteVM(id interface{}) *Persistence_DeleteVM_C func (_c *Persistence_DeleteVM_Call) Run(run func(id string)) *Persistence_DeleteVM_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(string)) + var arg0 string + if args[0] != nil { + arg0 = args[0].(string) + } + run( + arg0, + ) }) return _c } -func (_c *Persistence_DeleteVM_Call) Return(_a0 error) *Persistence_DeleteVM_Call { - _c.Call.Return(_a0) +func (_c *Persistence_DeleteVM_Call) Return(err error) *Persistence_DeleteVM_Call { + _c.Call.Return(err) return _c } -func (_c *Persistence_DeleteVM_Call) RunAndReturn(run func(string) error) *Persistence_DeleteVM_Call { +func (_c *Persistence_DeleteVM_Call) RunAndReturn(run func(id string) error) *Persistence_DeleteVM_Call { _c.Call.Return(run) return _c } -// LoadVMs provides a mock function with no fields -func (_m *Persistence) LoadVMs() ([]qemu.VMState, error) { - ret := _m.Called() +// LoadVMs provides a mock function for the type Persistence +func (_mock *Persistence) LoadVMs() ([]qemu.VMState, error) { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for LoadVMs") @@ -79,23 +100,21 @@ func (_m *Persistence) LoadVMs() ([]qemu.VMState, error) { var r0 []qemu.VMState var r1 error - if rf, ok := ret.Get(0).(func() ([]qemu.VMState, error)); ok { - return rf() + if returnFunc, ok := ret.Get(0).(func() ([]qemu.VMState, error)); ok { + return returnFunc() } - if rf, ok := ret.Get(0).(func() []qemu.VMState); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() []qemu.VMState); ok { + r0 = returnFunc() } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]qemu.VMState) } } - - if rf, ok := ret.Get(1).(func() error); ok { - r1 = rf() + if returnFunc, ok := ret.Get(1).(func() error); ok { + r1 = returnFunc() } else { r1 = ret.Error(1) } - return r0, r1 } @@ -116,8 +135,8 @@ func (_c *Persistence_LoadVMs_Call) Run(run func()) *Persistence_LoadVMs_Call { return _c } -func (_c *Persistence_LoadVMs_Call) Return(_a0 []qemu.VMState, _a1 error) *Persistence_LoadVMs_Call { - _c.Call.Return(_a0, _a1) +func (_c *Persistence_LoadVMs_Call) Return(vMStates []qemu.VMState, err error) *Persistence_LoadVMs_Call { + _c.Call.Return(vMStates, err) return _c } @@ -126,21 +145,20 @@ func (_c *Persistence_LoadVMs_Call) RunAndReturn(run func() ([]qemu.VMState, err return _c } -// SaveVM provides a mock function with given fields: state -func (_m *Persistence) SaveVM(state qemu.VMState) error { - ret := _m.Called(state) +// SaveVM provides a mock function for the type Persistence +func (_mock *Persistence) SaveVM(state qemu.VMState) error { + ret := _mock.Called(state) if len(ret) == 0 { panic("no return value specified for SaveVM") } var r0 error - if rf, ok := ret.Get(0).(func(qemu.VMState) error); ok { - r0 = rf(state) + if returnFunc, ok := ret.Get(0).(func(qemu.VMState) error); ok { + r0 = returnFunc(state) } else { r0 = ret.Error(0) } - return r0 } @@ -157,31 +175,23 @@ func (_e *Persistence_Expecter) SaveVM(state interface{}) *Persistence_SaveVM_Ca func (_c *Persistence_SaveVM_Call) Run(run func(state qemu.VMState)) *Persistence_SaveVM_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(qemu.VMState)) + var arg0 qemu.VMState + if args[0] != nil { + arg0 = args[0].(qemu.VMState) + } + run( + arg0, + ) }) return _c } -func (_c *Persistence_SaveVM_Call) Return(_a0 error) *Persistence_SaveVM_Call { - _c.Call.Return(_a0) +func (_c *Persistence_SaveVM_Call) Return(err error) *Persistence_SaveVM_Call { + _c.Call.Return(err) return _c } -func (_c *Persistence_SaveVM_Call) RunAndReturn(run func(qemu.VMState) error) *Persistence_SaveVM_Call { +func (_c *Persistence_SaveVM_Call) RunAndReturn(run func(state qemu.VMState) error) *Persistence_SaveVM_Call { _c.Call.Return(run) return _c } - -// NewPersistence creates a new instance of Persistence. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewPersistence(t interface { - mock.TestingT - Cleanup(func()) -}) *Persistence { - mock := &Persistence{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/manager/qemu/vm.go b/manager/qemu/vm.go index 9c756144..bf603a54 100644 --- a/manager/qemu/vm.go +++ b/manager/qemu/vm.go @@ -39,7 +39,7 @@ type qemuVM struct { vm.StateMachine } -func NewVM(config interface{}, cvmId string, logger *slog.Logger) vm.VM { +func NewVM(config any, cvmId string, logger *slog.Logger) vm.VM { return &qemuVM{ vmi: config.(VMInfo), cvmId: cvmId, @@ -192,7 +192,7 @@ func processExists(pid int) bool { return false } -func (v *qemuVM) GetConfig() interface{} { +func (v *qemuVM) GetConfig() any { return v.vmi } diff --git a/manager/vm/mocks/provider.go b/manager/vm/mocks/provider.go index 674d039b..459ef193 100644 --- a/manager/vm/mocks/provider.go +++ b/manager/vm/mocks/provider.go @@ -27,7 +27,7 @@ func (_m *Provider) EXPECT() *Provider_Expecter { } // Execute provides a mock function with given fields: config, computationId, logger -func (_m *Provider) Execute(config interface{}, computationId string, logger *slog.Logger) vm.VM { +func (_m *Provider) Execute(config any, computationId string, logger *slog.Logger) vm.VM { ret := _m.Called(config, computationId, logger) if len(ret) == 0 { @@ -35,7 +35,7 @@ func (_m *Provider) Execute(config interface{}, computationId string, logger *sl } var r0 vm.VM - if rf, ok := ret.Get(0).(func(interface{}, string, *slog.Logger) vm.VM); ok { + if rf, ok := ret.Get(0).(func(any, string, *slog.Logger) vm.VM); ok { r0 = rf(config, computationId, logger) } else { if ret.Get(0) != nil { @@ -52,16 +52,16 @@ type Provider_Execute_Call struct { } // Execute is a helper method to define mock.On call -// - config interface{} +// - config any // - computationId string // - logger *slog.Logger -func (_e *Provider_Expecter) Execute(config interface{}, computationId interface{}, logger interface{}) *Provider_Execute_Call { +func (_e *Provider_Expecter) Execute(config any, computationId any, logger any) *Provider_Execute_Call { return &Provider_Execute_Call{Call: _e.mock.On("Execute", config, computationId, logger)} } -func (_c *Provider_Execute_Call) Run(run func(config interface{}, computationId string, logger *slog.Logger)) *Provider_Execute_Call { +func (_c *Provider_Execute_Call) Run(run func(config any, computationId string, logger *slog.Logger)) *Provider_Execute_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(interface{}), args[1].(string), args[2].(*slog.Logger)) + run(args[0].(any), args[1].(string), args[2].(*slog.Logger)) }) return _c } @@ -71,7 +71,7 @@ func (_c *Provider_Execute_Call) Return(_a0 vm.VM) *Provider_Execute_Call { return _c } -func (_c *Provider_Execute_Call) RunAndReturn(run func(interface{}, string, *slog.Logger) vm.VM) *Provider_Execute_Call { +func (_c *Provider_Execute_Call) RunAndReturn(run func(any, string, *slog.Logger) vm.VM) *Provider_Execute_Call { _c.Call.Return(run) return _c } diff --git a/manager/vm/mocks/state_machine.go b/manager/vm/mocks/statemachine.go similarity index 70% rename from manager/vm/mocks/state_machine.go rename to manager/vm/mocks/statemachine.go index 85c08c76..a34cc29a 100644 --- a/manager/vm/mocks/state_machine.go +++ b/manager/vm/mocks/statemachine.go @@ -1,15 +1,31 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( mock "github.com/stretchr/testify/mock" - manager "github.com/ultravioletrs/cocos/pkg/manager" + "github.com/ultravioletrs/cocos/pkg/manager" ) +// NewStateMachine creates a new instance of StateMachine. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewStateMachine(t interface { + mock.TestingT + Cleanup(func()) +}) *StateMachine { + mock := &StateMachine{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // StateMachine is an autogenerated mock type for the StateMachine type type StateMachine struct { mock.Mock @@ -23,21 +39,20 @@ func (_m *StateMachine) EXPECT() *StateMachine_Expecter { return &StateMachine_Expecter{mock: &_m.Mock} } -// State provides a mock function with no fields -func (_m *StateMachine) State() string { - ret := _m.Called() +// State provides a mock function for the type StateMachine +func (_mock *StateMachine) State() string { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for State") } var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() string); ok { + r0 = returnFunc() } else { r0 = ret.Get(0).(string) } - return r0 } @@ -58,8 +73,8 @@ func (_c *StateMachine_State_Call) Run(run func()) *StateMachine_State_Call { return _c } -func (_c *StateMachine_State_Call) Return(_a0 string) *StateMachine_State_Call { - _c.Call.Return(_a0) +func (_c *StateMachine_State_Call) Return(s string) *StateMachine_State_Call { + _c.Call.Return(s) return _c } @@ -68,21 +83,20 @@ func (_c *StateMachine_State_Call) RunAndReturn(run func() string) *StateMachine return _c } -// Transition provides a mock function with given fields: newState -func (_m *StateMachine) Transition(newState manager.ManagerState) error { - ret := _m.Called(newState) +// Transition provides a mock function for the type StateMachine +func (_mock *StateMachine) Transition(newState manager.ManagerState) error { + ret := _mock.Called(newState) if len(ret) == 0 { panic("no return value specified for Transition") } var r0 error - if rf, ok := ret.Get(0).(func(manager.ManagerState) error); ok { - r0 = rf(newState) + if returnFunc, ok := ret.Get(0).(func(manager.ManagerState) error); ok { + r0 = returnFunc(newState) } else { r0 = ret.Error(0) } - return r0 } @@ -99,31 +113,23 @@ func (_e *StateMachine_Expecter) Transition(newState interface{}) *StateMachine_ func (_c *StateMachine_Transition_Call) Run(run func(newState manager.ManagerState)) *StateMachine_Transition_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(manager.ManagerState)) + var arg0 manager.ManagerState + if args[0] != nil { + arg0 = args[0].(manager.ManagerState) + } + run( + arg0, + ) }) return _c } -func (_c *StateMachine_Transition_Call) Return(_a0 error) *StateMachine_Transition_Call { - _c.Call.Return(_a0) +func (_c *StateMachine_Transition_Call) Return(err error) *StateMachine_Transition_Call { + _c.Call.Return(err) return _c } -func (_c *StateMachine_Transition_Call) RunAndReturn(run func(manager.ManagerState) error) *StateMachine_Transition_Call { +func (_c *StateMachine_Transition_Call) RunAndReturn(run func(newState manager.ManagerState) error) *StateMachine_Transition_Call { _c.Call.Return(run) return _c } - -// NewStateMachine creates a new instance of StateMachine. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewStateMachine(t interface { - mock.TestingT - Cleanup(func()) -}) *StateMachine { - mock := &StateMachine{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/manager/vm/mocks/vm.go b/manager/vm/mocks/vm.go index 79c99380..29bbc1f3 100644 --- a/manager/vm/mocks/vm.go +++ b/manager/vm/mocks/vm.go @@ -1,15 +1,31 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( mock "github.com/stretchr/testify/mock" - manager "github.com/ultravioletrs/cocos/pkg/manager" + "github.com/ultravioletrs/cocos/pkg/manager" ) +// NewVM creates a new instance of VM. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewVM(t interface { + mock.TestingT + Cleanup(func()) +}) *VM { + mock := &VM{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // VM is an autogenerated mock type for the VM type type VM struct { mock.Mock @@ -23,23 +39,22 @@ func (_m *VM) EXPECT() *VM_Expecter { return &VM_Expecter{mock: &_m.Mock} } -// GetConfig provides a mock function with no fields -func (_m *VM) GetConfig() interface{} { - ret := _m.Called() +// GetConfig provides a mock function for the type VM +func (_mock *VM) GetConfig() any { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for GetConfig") } - var r0 interface{} - if rf, ok := ret.Get(0).(func() interface{}); ok { - r0 = rf() + var r0 any + if returnFunc, ok := ret.Get(0).(func() any); ok { + r0 = returnFunc() } else { if ret.Get(0) != nil { - r0 = ret.Get(0).(interface{}) + r0 = ret.Get(0).(any) } } - return r0 } @@ -60,31 +75,30 @@ func (_c *VM_GetConfig_Call) Run(run func()) *VM_GetConfig_Call { return _c } -func (_c *VM_GetConfig_Call) Return(_a0 interface{}) *VM_GetConfig_Call { - _c.Call.Return(_a0) +func (_c *VM_GetConfig_Call) Return(v any) *VM_GetConfig_Call { + _c.Call.Return(v) return _c } -func (_c *VM_GetConfig_Call) RunAndReturn(run func() interface{}) *VM_GetConfig_Call { +func (_c *VM_GetConfig_Call) RunAndReturn(run func() any) *VM_GetConfig_Call { _c.Call.Return(run) return _c } -// GetProcess provides a mock function with no fields -func (_m *VM) GetProcess() int { - ret := _m.Called() +// GetProcess provides a mock function for the type VM +func (_mock *VM) GetProcess() int { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for GetProcess") } var r0 int - if rf, ok := ret.Get(0).(func() int); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() int); ok { + r0 = returnFunc() } else { r0 = ret.Get(0).(int) } - return r0 } @@ -105,8 +119,8 @@ func (_c *VM_GetProcess_Call) Run(run func()) *VM_GetProcess_Call { return _c } -func (_c *VM_GetProcess_Call) Return(_a0 int) *VM_GetProcess_Call { - _c.Call.Return(_a0) +func (_c *VM_GetProcess_Call) Return(n int) *VM_GetProcess_Call { + _c.Call.Return(n) return _c } @@ -115,21 +129,20 @@ func (_c *VM_GetProcess_Call) RunAndReturn(run func() int) *VM_GetProcess_Call { return _c } -// SetProcess provides a mock function with given fields: pid -func (_m *VM) SetProcess(pid int) error { - ret := _m.Called(pid) +// SetProcess provides a mock function for the type VM +func (_mock *VM) SetProcess(pid int) error { + ret := _mock.Called(pid) if len(ret) == 0 { panic("no return value specified for SetProcess") } var r0 error - if rf, ok := ret.Get(0).(func(int) error); ok { - r0 = rf(pid) + if returnFunc, ok := ret.Get(0).(func(int) error); ok { + r0 = returnFunc(pid) } else { r0 = ret.Error(0) } - return r0 } @@ -146,36 +159,41 @@ func (_e *VM_Expecter) SetProcess(pid interface{}) *VM_SetProcess_Call { func (_c *VM_SetProcess_Call) Run(run func(pid int)) *VM_SetProcess_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(int)) + var arg0 int + if args[0] != nil { + arg0 = args[0].(int) + } + run( + arg0, + ) }) return _c } -func (_c *VM_SetProcess_Call) Return(_a0 error) *VM_SetProcess_Call { - _c.Call.Return(_a0) +func (_c *VM_SetProcess_Call) Return(err error) *VM_SetProcess_Call { + _c.Call.Return(err) return _c } -func (_c *VM_SetProcess_Call) RunAndReturn(run func(int) error) *VM_SetProcess_Call { +func (_c *VM_SetProcess_Call) RunAndReturn(run func(pid int) error) *VM_SetProcess_Call { _c.Call.Return(run) return _c } -// Start provides a mock function with no fields -func (_m *VM) Start() error { - ret := _m.Called() +// Start provides a mock function for the type VM +func (_mock *VM) Start() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Start") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -196,8 +214,8 @@ func (_c *VM_Start_Call) Run(run func()) *VM_Start_Call { return _c } -func (_c *VM_Start_Call) Return(_a0 error) *VM_Start_Call { - _c.Call.Return(_a0) +func (_c *VM_Start_Call) Return(err error) *VM_Start_Call { + _c.Call.Return(err) return _c } @@ -206,21 +224,20 @@ func (_c *VM_Start_Call) RunAndReturn(run func() error) *VM_Start_Call { return _c } -// State provides a mock function with no fields -func (_m *VM) State() string { - ret := _m.Called() +// State provides a mock function for the type VM +func (_mock *VM) State() string { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for State") } var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() string); ok { + r0 = returnFunc() } else { r0 = ret.Get(0).(string) } - return r0 } @@ -241,8 +258,8 @@ func (_c *VM_State_Call) Run(run func()) *VM_State_Call { return _c } -func (_c *VM_State_Call) Return(_a0 string) *VM_State_Call { - _c.Call.Return(_a0) +func (_c *VM_State_Call) Return(s string) *VM_State_Call { + _c.Call.Return(s) return _c } @@ -251,21 +268,20 @@ func (_c *VM_State_Call) RunAndReturn(run func() string) *VM_State_Call { return _c } -// Stop provides a mock function with no fields -func (_m *VM) Stop() error { - ret := _m.Called() +// Stop provides a mock function for the type VM +func (_mock *VM) Stop() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Stop") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -286,8 +302,8 @@ func (_c *VM_Stop_Call) Run(run func()) *VM_Stop_Call { return _c } -func (_c *VM_Stop_Call) Return(_a0 error) *VM_Stop_Call { - _c.Call.Return(_a0) +func (_c *VM_Stop_Call) Return(err error) *VM_Stop_Call { + _c.Call.Return(err) return _c } @@ -296,21 +312,20 @@ func (_c *VM_Stop_Call) RunAndReturn(run func() error) *VM_Stop_Call { return _c } -// Transition provides a mock function with given fields: newState -func (_m *VM) Transition(newState manager.ManagerState) error { - ret := _m.Called(newState) +// Transition provides a mock function for the type VM +func (_mock *VM) Transition(newState manager.ManagerState) error { + ret := _mock.Called(newState) if len(ret) == 0 { panic("no return value specified for Transition") } var r0 error - if rf, ok := ret.Get(0).(func(manager.ManagerState) error); ok { - r0 = rf(newState) + if returnFunc, ok := ret.Get(0).(func(manager.ManagerState) error); ok { + r0 = returnFunc(newState) } else { r0 = ret.Error(0) } - return r0 } @@ -327,31 +342,23 @@ func (_e *VM_Expecter) Transition(newState interface{}) *VM_Transition_Call { func (_c *VM_Transition_Call) Run(run func(newState manager.ManagerState)) *VM_Transition_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(manager.ManagerState)) + var arg0 manager.ManagerState + if args[0] != nil { + arg0 = args[0].(manager.ManagerState) + } + run( + arg0, + ) }) return _c } -func (_c *VM_Transition_Call) Return(_a0 error) *VM_Transition_Call { - _c.Call.Return(_a0) +func (_c *VM_Transition_Call) Return(err error) *VM_Transition_Call { + _c.Call.Return(err) return _c } -func (_c *VM_Transition_Call) RunAndReturn(run func(manager.ManagerState) error) *VM_Transition_Call { +func (_c *VM_Transition_Call) RunAndReturn(run func(newState manager.ManagerState) error) *VM_Transition_Call { _c.Call.Return(run) return _c } - -// NewVM creates a new instance of VM. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewVM(t interface { - mock.TestingT - Cleanup(func()) -}) *VM { - mock := &VM{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/manager/vm/vm.go b/manager/vm/vm.go index 41ac7028..45dc19df 100644 --- a/manager/vm/vm.go +++ b/manager/vm/vm.go @@ -17,10 +17,10 @@ type VM interface { GetProcess() int Transition(newState pkgmanager.ManagerState) error State() string - GetConfig() interface{} + GetConfig() any } -type Provider func(config interface{}, computationId string, logger *slog.Logger) VM +type Provider func(config any, computationId string, logger *slog.Logger) VM type Event struct { EventType string diff --git a/mockery.yml b/mockery.yml deleted file mode 100644 index 58291268..00000000 --- a/mockery.yml +++ /dev/null @@ -1,145 +0,0 @@ -with-expecter: true -filename: "{{.InterfaceName}}.go" -outpkg: "mocks" -boilerplate-file: "./boilerplate.txt" -packages: - github.com/ultravioletrs/cocos/agent: - interfaces: - Service: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "agent.go" - mockname: "{{.InterfaceName}}" - AgentService_DataClient: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "agent_grpc_data.go" - mockname: "{{.InterfaceName}}" - AgentService_AlgoClient: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "agent_grpc_algo.go" - mockname: "{{.InterfaceName}}" - AgentService_IMAMeasurementsClient: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "agent_grpc_ima.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/agent/auth: - interfaces: - Authenticator: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "auth.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/agent/events: - interfaces: - Service: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "events.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/agent/statemachine: - interfaces: - StateMachine: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "state.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/internal/server: - interfaces: - Server: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "server.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/manager: - interfaces: - Service: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "service.go" - mockname: "{{.InterfaceName}}" - ManagerServiceClient: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "manager_service_client.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/manager/qemu: - interfaces: - Persistence: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "persistence.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/manager/vm: - interfaces: - StateMachine: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "state_machine.go" - mockname: "{{.InterfaceName}}" - VM: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "vm.go" - mockname: "{{.InterfaceName}}" - Provider: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "provider.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/pkg/sdk: - interfaces: - SDK: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "sdk.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/agent/cvms/server: - interfaces: - AgentServer: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "server.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/agent/cvms/api/grpc/storage: - interfaces: - Storage: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "storage.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/pkg/clients/grpc: - interfaces: - Client: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "client.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/pkg/attestation: - interfaces: - Provider: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "attestation.go" - mockname: "{{.InterfaceName}}" - Verifier: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "verifier.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/agent/algorithm: - interfaces: - Algorithm: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "algorithm.go" - mockname: "{{.InterfaceName}}" - github.com/ultravioletrs/cocos/pkg/attestation/cmdconfig: - interfaces: - MeasurementProvider: - config: - dir: "{{.InterfaceDir}}/mocks" - filename: "measurement_provider.go" - mockname: "{{.InterfaceName}}" diff --git a/pkg/attestation/azure/snp.go b/pkg/attestation/azure/snp.go index ef336c0b..2944319f 100644 --- a/pkg/attestation/azure/snp.go +++ b/pkg/attestation/azure/snp.go @@ -165,7 +165,7 @@ func GenerateAttestationPolicy(token, product string, policy uint64) (*attestati return nil, fmt.Errorf("failed to validate token: %w", err) } - tee, ok := claims["x-ms-isolation-tee"].(map[string]interface{}) + tee, ok := claims["x-ms-isolation-tee"].(map[string]any) if !ok { return nil, fmt.Errorf("failed to get tee from claims") } @@ -284,7 +284,7 @@ func FetchAzureAttestationToken(tokenNonce []byte, maaURL string) ([]byte, error return []byte(token), nil } -func validateToken(token string) (map[string]interface{}, error) { +func validateToken(token string) (map[string]any, error) { unverifiedToken, _, err := new(jwt.Parser).ParseUnverified(token, jwt.MapClaims{}) if err != nil { return nil, fmt.Errorf("failed to parse token: %w", err) diff --git a/pkg/attestation/azure/snp_test.go b/pkg/attestation/azure/snp_test.go index 5828872e..f0c5587f 100644 --- a/pkg/attestation/azure/snp_test.go +++ b/pkg/attestation/azure/snp_test.go @@ -452,7 +452,7 @@ func TestIntegration_FullAttestationFlow(t *testing.T) { maaServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/attest": - response := map[string]interface{}{ + response := map[string]any{ "token": createMockJWT(), } w.Header().Set("Content-Type", "application/json") @@ -460,7 +460,7 @@ func TestIntegration_FullAttestationFlow(t *testing.T) { t.Fatalf("Failed to encode response: %v", err) } case "/.well-known/openid_configuration": - config := map[string]interface{}{ + config := map[string]any{ "jwks_uri": "maaServer.URL" + "/certs", } w.Header().Set("Content-Type", "application/json") @@ -468,8 +468,8 @@ func TestIntegration_FullAttestationFlow(t *testing.T) { t.Fatalf("Failed to encode OpenID configuration: %v", err) } case "/certs": - jwks := map[string]interface{}{ - "keys": []map[string]interface{}{ + jwks := map[string]any{ + "keys": []map[string]any{ { "kid": "test-kid", "kty": "RSA", @@ -555,7 +555,7 @@ func createMockJWT() string { "aud": "test-audience", "exp": time.Now().Add(time.Hour).Unix(), "iat": time.Now().Unix(), - "x-ms-isolation-tee": map[string]interface{}{ + "x-ms-isolation-tee": map[string]any{ "x-ms-sevsnpvm-familyId": "1234567890abcdef", "x-ms-sevsnpvm-imageId": "fedcba0987654321", "x-ms-sevsnpvm-launchmeasurement": "abcdef1234567890", diff --git a/pkg/attestation/cmdconfig/mocks/measurement_provider.go b/pkg/attestation/cmdconfig/mocks/measurementprovider.go similarity index 67% rename from pkg/attestation/cmdconfig/mocks/measurement_provider.go rename to pkg/attestation/cmdconfig/mocks/measurementprovider.go index 44a4dbe0..64a7bd64 100644 --- a/pkg/attestation/cmdconfig/mocks/measurement_provider.go +++ b/pkg/attestation/cmdconfig/mocks/measurementprovider.go @@ -1,11 +1,29 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks -import mock "github.com/stretchr/testify/mock" +import ( + mock "github.com/stretchr/testify/mock" +) + +// NewMeasurementProvider creates a new instance of MeasurementProvider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewMeasurementProvider(t interface { + mock.TestingT + Cleanup(func()) +}) *MeasurementProvider { + mock := &MeasurementProvider{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} // MeasurementProvider is an autogenerated mock type for the MeasurementProvider type type MeasurementProvider struct { @@ -20,9 +38,9 @@ func (_m *MeasurementProvider) EXPECT() *MeasurementProvider_Expecter { return &MeasurementProvider_Expecter{mock: &_m.Mock} } -// Run provides a mock function with given fields: binaryPath -func (_m *MeasurementProvider) Run(binaryPath string) ([]byte, error) { - ret := _m.Called(binaryPath) +// Run provides a mock function for the type MeasurementProvider +func (_mock *MeasurementProvider) Run(binaryPath string) ([]byte, error) { + ret := _mock.Called(binaryPath) if len(ret) == 0 { panic("no return value specified for Run") @@ -30,23 +48,21 @@ func (_m *MeasurementProvider) Run(binaryPath string) ([]byte, error) { var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func(string) ([]byte, error)); ok { - return rf(binaryPath) + if returnFunc, ok := ret.Get(0).(func(string) ([]byte, error)); ok { + return returnFunc(binaryPath) } - if rf, ok := ret.Get(0).(func(string) []byte); ok { - r0 = rf(binaryPath) + if returnFunc, ok := ret.Get(0).(func(string) []byte); ok { + r0 = returnFunc(binaryPath) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(string) error); ok { - r1 = rf(binaryPath) + if returnFunc, ok := ret.Get(1).(func(string) error); ok { + r1 = returnFunc(binaryPath) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -63,36 +79,41 @@ func (_e *MeasurementProvider_Expecter) Run(binaryPath interface{}) *Measurement func (_c *MeasurementProvider_Run_Call) Run(run func(binaryPath string)) *MeasurementProvider_Run_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(string)) + var arg0 string + if args[0] != nil { + arg0 = args[0].(string) + } + run( + arg0, + ) }) return _c } -func (_c *MeasurementProvider_Run_Call) Return(_a0 []byte, _a1 error) *MeasurementProvider_Run_Call { - _c.Call.Return(_a0, _a1) +func (_c *MeasurementProvider_Run_Call) Return(bytes []byte, err error) *MeasurementProvider_Run_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *MeasurementProvider_Run_Call) RunAndReturn(run func(string) ([]byte, error)) *MeasurementProvider_Run_Call { +func (_c *MeasurementProvider_Run_Call) RunAndReturn(run func(binaryPath string) ([]byte, error)) *MeasurementProvider_Run_Call { _c.Call.Return(run) return _c } -// Stop provides a mock function with no fields -func (_m *MeasurementProvider) Stop() error { - ret := _m.Called() +// Stop provides a mock function for the type MeasurementProvider +func (_mock *MeasurementProvider) Stop() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Stop") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -113,8 +134,8 @@ func (_c *MeasurementProvider_Stop_Call) Run(run func()) *MeasurementProvider_St return _c } -func (_c *MeasurementProvider_Stop_Call) Return(_a0 error) *MeasurementProvider_Stop_Call { - _c.Call.Return(_a0) +func (_c *MeasurementProvider_Stop_Call) Return(err error) *MeasurementProvider_Stop_Call { + _c.Call.Return(err) return _c } @@ -122,17 +143,3 @@ func (_c *MeasurementProvider_Stop_Call) RunAndReturn(run func() error) *Measure _c.Call.Return(run) return _c } - -// NewMeasurementProvider creates a new instance of MeasurementProvider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewMeasurementProvider(t interface { - mock.TestingT - Cleanup(func()) -}) *MeasurementProvider { - mock := &MeasurementProvider{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/pkg/attestation/mocks/attestation.go b/pkg/attestation/mocks/provider.go similarity index 56% rename from pkg/attestation/mocks/attestation.go rename to pkg/attestation/mocks/provider.go index 9eb8cb2a..74cc2c2d 100644 --- a/pkg/attestation/mocks/attestation.go +++ b/pkg/attestation/mocks/provider.go @@ -1,11 +1,29 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks -import mock "github.com/stretchr/testify/mock" +import ( + mock "github.com/stretchr/testify/mock" +) + +// NewProvider creates a new instance of Provider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewProvider(t interface { + mock.TestingT + Cleanup(func()) +}) *Provider { + mock := &Provider{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} // Provider is an autogenerated mock type for the Provider type type Provider struct { @@ -20,9 +38,9 @@ func (_m *Provider) EXPECT() *Provider_Expecter { return &Provider_Expecter{mock: &_m.Mock} } -// Attestation provides a mock function with given fields: teeNonce, vTpmNonce -func (_m *Provider) Attestation(teeNonce []byte, vTpmNonce []byte) ([]byte, error) { - ret := _m.Called(teeNonce, vTpmNonce) +// Attestation provides a mock function for the type Provider +func (_mock *Provider) Attestation(teeNonce []byte, vTpmNonce []byte) ([]byte, error) { + ret := _mock.Called(teeNonce, vTpmNonce) if len(ret) == 0 { panic("no return value specified for Attestation") @@ -30,23 +48,21 @@ func (_m *Provider) Attestation(teeNonce []byte, vTpmNonce []byte) ([]byte, erro var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func([]byte, []byte) ([]byte, error)); ok { - return rf(teeNonce, vTpmNonce) + if returnFunc, ok := ret.Get(0).(func([]byte, []byte) ([]byte, error)); ok { + return returnFunc(teeNonce, vTpmNonce) } - if rf, ok := ret.Get(0).(func([]byte, []byte) []byte); ok { - r0 = rf(teeNonce, vTpmNonce) + if returnFunc, ok := ret.Get(0).(func([]byte, []byte) []byte); ok { + r0 = returnFunc(teeNonce, vTpmNonce) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func([]byte, []byte) error); ok { - r1 = rf(teeNonce, vTpmNonce) + if returnFunc, ok := ret.Get(1).(func([]byte, []byte) error); ok { + r1 = returnFunc(teeNonce, vTpmNonce) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -64,24 +80,35 @@ func (_e *Provider_Expecter) Attestation(teeNonce interface{}, vTpmNonce interfa func (_c *Provider_Attestation_Call) Run(run func(teeNonce []byte, vTpmNonce []byte)) *Provider_Attestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte), args[1].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + var arg1 []byte + if args[1] != nil { + arg1 = args[1].([]byte) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Provider_Attestation_Call) Return(_a0 []byte, _a1 error) *Provider_Attestation_Call { - _c.Call.Return(_a0, _a1) +func (_c *Provider_Attestation_Call) Return(bytes []byte, err error) *Provider_Attestation_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Provider_Attestation_Call) RunAndReturn(run func([]byte, []byte) ([]byte, error)) *Provider_Attestation_Call { +func (_c *Provider_Attestation_Call) RunAndReturn(run func(teeNonce []byte, vTpmNonce []byte) ([]byte, error)) *Provider_Attestation_Call { _c.Call.Return(run) return _c } -// AzureAttestationToken provides a mock function with given fields: tokenNonce -func (_m *Provider) AzureAttestationToken(tokenNonce []byte) ([]byte, error) { - ret := _m.Called(tokenNonce) +// AzureAttestationToken provides a mock function for the type Provider +func (_mock *Provider) AzureAttestationToken(tokenNonce []byte) ([]byte, error) { + ret := _mock.Called(tokenNonce) if len(ret) == 0 { panic("no return value specified for AzureAttestationToken") @@ -89,23 +116,21 @@ func (_m *Provider) AzureAttestationToken(tokenNonce []byte) ([]byte, error) { var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok { - return rf(tokenNonce) + if returnFunc, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok { + return returnFunc(tokenNonce) } - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(tokenNonce) + if returnFunc, ok := ret.Get(0).(func([]byte) []byte); ok { + r0 = returnFunc(tokenNonce) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(tokenNonce) + if returnFunc, ok := ret.Get(1).(func([]byte) error); ok { + r1 = returnFunc(tokenNonce) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -122,24 +147,30 @@ func (_e *Provider_Expecter) AzureAttestationToken(tokenNonce interface{}) *Prov func (_c *Provider_AzureAttestationToken_Call) Run(run func(tokenNonce []byte)) *Provider_AzureAttestationToken_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + run( + arg0, + ) }) return _c } -func (_c *Provider_AzureAttestationToken_Call) Return(_a0 []byte, _a1 error) *Provider_AzureAttestationToken_Call { - _c.Call.Return(_a0, _a1) +func (_c *Provider_AzureAttestationToken_Call) Return(bytes []byte, err error) *Provider_AzureAttestationToken_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Provider_AzureAttestationToken_Call) RunAndReturn(run func([]byte) ([]byte, error)) *Provider_AzureAttestationToken_Call { +func (_c *Provider_AzureAttestationToken_Call) RunAndReturn(run func(tokenNonce []byte) ([]byte, error)) *Provider_AzureAttestationToken_Call { _c.Call.Return(run) return _c } -// TeeAttestation provides a mock function with given fields: teeNonce -func (_m *Provider) TeeAttestation(teeNonce []byte) ([]byte, error) { - ret := _m.Called(teeNonce) +// TeeAttestation provides a mock function for the type Provider +func (_mock *Provider) TeeAttestation(teeNonce []byte) ([]byte, error) { + ret := _mock.Called(teeNonce) if len(ret) == 0 { panic("no return value specified for TeeAttestation") @@ -147,23 +178,21 @@ func (_m *Provider) TeeAttestation(teeNonce []byte) ([]byte, error) { var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok { - return rf(teeNonce) + if returnFunc, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok { + return returnFunc(teeNonce) } - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(teeNonce) + if returnFunc, ok := ret.Get(0).(func([]byte) []byte); ok { + r0 = returnFunc(teeNonce) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(teeNonce) + if returnFunc, ok := ret.Get(1).(func([]byte) error); ok { + r1 = returnFunc(teeNonce) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -180,24 +209,30 @@ func (_e *Provider_Expecter) TeeAttestation(teeNonce interface{}) *Provider_TeeA func (_c *Provider_TeeAttestation_Call) Run(run func(teeNonce []byte)) *Provider_TeeAttestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + run( + arg0, + ) }) return _c } -func (_c *Provider_TeeAttestation_Call) Return(_a0 []byte, _a1 error) *Provider_TeeAttestation_Call { - _c.Call.Return(_a0, _a1) +func (_c *Provider_TeeAttestation_Call) Return(bytes []byte, err error) *Provider_TeeAttestation_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Provider_TeeAttestation_Call) RunAndReturn(run func([]byte) ([]byte, error)) *Provider_TeeAttestation_Call { +func (_c *Provider_TeeAttestation_Call) RunAndReturn(run func(teeNonce []byte) ([]byte, error)) *Provider_TeeAttestation_Call { _c.Call.Return(run) return _c } -// VTpmAttestation provides a mock function with given fields: vTpmNonce -func (_m *Provider) VTpmAttestation(vTpmNonce []byte) ([]byte, error) { - ret := _m.Called(vTpmNonce) +// VTpmAttestation provides a mock function for the type Provider +func (_mock *Provider) VTpmAttestation(vTpmNonce []byte) ([]byte, error) { + ret := _mock.Called(vTpmNonce) if len(ret) == 0 { panic("no return value specified for VTpmAttestation") @@ -205,23 +240,21 @@ func (_m *Provider) VTpmAttestation(vTpmNonce []byte) ([]byte, error) { var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok { - return rf(vTpmNonce) + if returnFunc, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok { + return returnFunc(vTpmNonce) } - if rf, ok := ret.Get(0).(func([]byte) []byte); ok { - r0 = rf(vTpmNonce) + if returnFunc, ok := ret.Get(0).(func([]byte) []byte); ok { + r0 = returnFunc(vTpmNonce) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func([]byte) error); ok { - r1 = rf(vTpmNonce) + if returnFunc, ok := ret.Get(1).(func([]byte) error); ok { + r1 = returnFunc(vTpmNonce) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -238,31 +271,23 @@ func (_e *Provider_Expecter) VTpmAttestation(vTpmNonce interface{}) *Provider_VT func (_c *Provider_VTpmAttestation_Call) Run(run func(vTpmNonce []byte)) *Provider_VTpmAttestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + run( + arg0, + ) }) return _c } -func (_c *Provider_VTpmAttestation_Call) Return(_a0 []byte, _a1 error) *Provider_VTpmAttestation_Call { - _c.Call.Return(_a0, _a1) +func (_c *Provider_VTpmAttestation_Call) Return(bytes []byte, err error) *Provider_VTpmAttestation_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *Provider_VTpmAttestation_Call) RunAndReturn(run func([]byte) ([]byte, error)) *Provider_VTpmAttestation_Call { +func (_c *Provider_VTpmAttestation_Call) RunAndReturn(run func(vTpmNonce []byte) ([]byte, error)) *Provider_VTpmAttestation_Call { _c.Call.Return(run) return _c } - -// NewProvider creates a new instance of Provider. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewProvider(t interface { - mock.TestingT - Cleanup(func()) -}) *Provider { - mock := &Provider{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/pkg/attestation/mocks/verifier.go b/pkg/attestation/mocks/verifier.go index 5962ee5a..4d95db55 100644 --- a/pkg/attestation/mocks/verifier.go +++ b/pkg/attestation/mocks/verifier.go @@ -1,11 +1,29 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks -import mock "github.com/stretchr/testify/mock" +import ( + mock "github.com/stretchr/testify/mock" +) + +// NewVerifier creates a new instance of Verifier. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewVerifier(t interface { + mock.TestingT + Cleanup(func()) +}) *Verifier { + mock := &Verifier{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} // Verifier is an autogenerated mock type for the Verifier type type Verifier struct { @@ -20,21 +38,20 @@ func (_m *Verifier) EXPECT() *Verifier_Expecter { return &Verifier_Expecter{mock: &_m.Mock} } -// JSONToPolicy provides a mock function with given fields: path -func (_m *Verifier) JSONToPolicy(path string) error { - ret := _m.Called(path) +// JSONToPolicy provides a mock function for the type Verifier +func (_mock *Verifier) JSONToPolicy(path string) error { + ret := _mock.Called(path) if len(ret) == 0 { panic("no return value specified for JSONToPolicy") } var r0 error - if rf, ok := ret.Get(0).(func(string) error); ok { - r0 = rf(path) + if returnFunc, ok := ret.Get(0).(func(string) error); ok { + r0 = returnFunc(path) } else { r0 = ret.Error(0) } - return r0 } @@ -51,36 +68,41 @@ func (_e *Verifier_Expecter) JSONToPolicy(path interface{}) *Verifier_JSONToPoli func (_c *Verifier_JSONToPolicy_Call) Run(run func(path string)) *Verifier_JSONToPolicy_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(string)) + var arg0 string + if args[0] != nil { + arg0 = args[0].(string) + } + run( + arg0, + ) }) return _c } -func (_c *Verifier_JSONToPolicy_Call) Return(_a0 error) *Verifier_JSONToPolicy_Call { - _c.Call.Return(_a0) +func (_c *Verifier_JSONToPolicy_Call) Return(err error) *Verifier_JSONToPolicy_Call { + _c.Call.Return(err) return _c } -func (_c *Verifier_JSONToPolicy_Call) RunAndReturn(run func(string) error) *Verifier_JSONToPolicy_Call { +func (_c *Verifier_JSONToPolicy_Call) RunAndReturn(run func(path string) error) *Verifier_JSONToPolicy_Call { _c.Call.Return(run) return _c } -// VerifTeeAttestation provides a mock function with given fields: report, teeNonce -func (_m *Verifier) VerifTeeAttestation(report []byte, teeNonce []byte) error { - ret := _m.Called(report, teeNonce) +// VerifTeeAttestation provides a mock function for the type Verifier +func (_mock *Verifier) VerifTeeAttestation(report []byte, teeNonce []byte) error { + ret := _mock.Called(report, teeNonce) if len(ret) == 0 { panic("no return value specified for VerifTeeAttestation") } var r0 error - if rf, ok := ret.Get(0).(func([]byte, []byte) error); ok { - r0 = rf(report, teeNonce) + if returnFunc, ok := ret.Get(0).(func([]byte, []byte) error); ok { + r0 = returnFunc(report, teeNonce) } else { r0 = ret.Error(0) } - return r0 } @@ -98,36 +120,46 @@ func (_e *Verifier_Expecter) VerifTeeAttestation(report interface{}, teeNonce in func (_c *Verifier_VerifTeeAttestation_Call) Run(run func(report []byte, teeNonce []byte)) *Verifier_VerifTeeAttestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte), args[1].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + var arg1 []byte + if args[1] != nil { + arg1 = args[1].([]byte) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Verifier_VerifTeeAttestation_Call) Return(_a0 error) *Verifier_VerifTeeAttestation_Call { - _c.Call.Return(_a0) +func (_c *Verifier_VerifTeeAttestation_Call) Return(err error) *Verifier_VerifTeeAttestation_Call { + _c.Call.Return(err) return _c } -func (_c *Verifier_VerifTeeAttestation_Call) RunAndReturn(run func([]byte, []byte) error) *Verifier_VerifTeeAttestation_Call { +func (_c *Verifier_VerifTeeAttestation_Call) RunAndReturn(run func(report []byte, teeNonce []byte) error) *Verifier_VerifTeeAttestation_Call { _c.Call.Return(run) return _c } -// VerifVTpmAttestation provides a mock function with given fields: report, vTpmNonce -func (_m *Verifier) VerifVTpmAttestation(report []byte, vTpmNonce []byte) error { - ret := _m.Called(report, vTpmNonce) +// VerifVTpmAttestation provides a mock function for the type Verifier +func (_mock *Verifier) VerifVTpmAttestation(report []byte, vTpmNonce []byte) error { + ret := _mock.Called(report, vTpmNonce) if len(ret) == 0 { panic("no return value specified for VerifVTpmAttestation") } var r0 error - if rf, ok := ret.Get(0).(func([]byte, []byte) error); ok { - r0 = rf(report, vTpmNonce) + if returnFunc, ok := ret.Get(0).(func([]byte, []byte) error); ok { + r0 = returnFunc(report, vTpmNonce) } else { r0 = ret.Error(0) } - return r0 } @@ -145,36 +177,46 @@ func (_e *Verifier_Expecter) VerifVTpmAttestation(report interface{}, vTpmNonce func (_c *Verifier_VerifVTpmAttestation_Call) Run(run func(report []byte, vTpmNonce []byte)) *Verifier_VerifVTpmAttestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte), args[1].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + var arg1 []byte + if args[1] != nil { + arg1 = args[1].([]byte) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *Verifier_VerifVTpmAttestation_Call) Return(_a0 error) *Verifier_VerifVTpmAttestation_Call { - _c.Call.Return(_a0) +func (_c *Verifier_VerifVTpmAttestation_Call) Return(err error) *Verifier_VerifVTpmAttestation_Call { + _c.Call.Return(err) return _c } -func (_c *Verifier_VerifVTpmAttestation_Call) RunAndReturn(run func([]byte, []byte) error) *Verifier_VerifVTpmAttestation_Call { +func (_c *Verifier_VerifVTpmAttestation_Call) RunAndReturn(run func(report []byte, vTpmNonce []byte) error) *Verifier_VerifVTpmAttestation_Call { _c.Call.Return(run) return _c } -// VerifyAttestation provides a mock function with given fields: report, teeNonce, vTpmNonce -func (_m *Verifier) VerifyAttestation(report []byte, teeNonce []byte, vTpmNonce []byte) error { - ret := _m.Called(report, teeNonce, vTpmNonce) +// VerifyAttestation provides a mock function for the type Verifier +func (_mock *Verifier) VerifyAttestation(report []byte, teeNonce []byte, vTpmNonce []byte) error { + ret := _mock.Called(report, teeNonce, vTpmNonce) if len(ret) == 0 { panic("no return value specified for VerifyAttestation") } var r0 error - if rf, ok := ret.Get(0).(func([]byte, []byte, []byte) error); ok { - r0 = rf(report, teeNonce, vTpmNonce) + if returnFunc, ok := ret.Get(0).(func([]byte, []byte, []byte) error); ok { + r0 = returnFunc(report, teeNonce, vTpmNonce) } else { r0 = ret.Error(0) } - return r0 } @@ -193,31 +235,33 @@ func (_e *Verifier_Expecter) VerifyAttestation(report interface{}, teeNonce inte func (_c *Verifier_VerifyAttestation_Call) Run(run func(report []byte, teeNonce []byte, vTpmNonce []byte)) *Verifier_VerifyAttestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].([]byte), args[1].([]byte), args[2].([]byte)) + var arg0 []byte + if args[0] != nil { + arg0 = args[0].([]byte) + } + var arg1 []byte + if args[1] != nil { + arg1 = args[1].([]byte) + } + var arg2 []byte + if args[2] != nil { + arg2 = args[2].([]byte) + } + run( + arg0, + arg1, + arg2, + ) }) return _c } -func (_c *Verifier_VerifyAttestation_Call) Return(_a0 error) *Verifier_VerifyAttestation_Call { - _c.Call.Return(_a0) +func (_c *Verifier_VerifyAttestation_Call) Return(err error) *Verifier_VerifyAttestation_Call { + _c.Call.Return(err) return _c } -func (_c *Verifier_VerifyAttestation_Call) RunAndReturn(run func([]byte, []byte, []byte) error) *Verifier_VerifyAttestation_Call { +func (_c *Verifier_VerifyAttestation_Call) RunAndReturn(run func(report []byte, teeNonce []byte, vTpmNonce []byte) error) *Verifier_VerifyAttestation_Call { _c.Call.Return(run) return _c } - -// NewVerifier creates a new instance of Verifier. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewVerifier(t interface { - mock.TestingT - Cleanup(func()) -}) *Verifier { - mock := &Verifier{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/pkg/attestation/vtpm/vtpm.go b/pkg/attestation/vtpm/vtpm.go index 1337ad69..c2ac7c54 100644 --- a/pkg/attestation/vtpm/vtpm.go +++ b/pkg/attestation/vtpm/vtpm.go @@ -415,7 +415,7 @@ func ConvertPolicyToJSON(attestationConfiguration *attestation.Config) ([]byte, return nil, errors.Wrap(ErrProtoMarshalFailed, err) } - var pbMap map[string]interface{} + var pbMap map[string]any if err := json.Unmarshal(pbJson, &pbMap); err != nil { return nil, errors.Wrap(ErrJsonUnarshalFailed, err) } @@ -425,7 +425,7 @@ func ConvertPolicyToJSON(attestationConfiguration *attestation.Config) ([]byte, return nil, errors.Wrap(ErrJsonMarshalFailed, err) } - var pcrMap map[string]interface{} + var pcrMap map[string]any if err := json.Unmarshal(pcrJson, &pcrMap); err != nil { return nil, errors.Wrap(ErrJsonUnarshalFailed, err) } diff --git a/pkg/attestation/vtpm/vtpm_test.go b/pkg/attestation/vtpm/vtpm_test.go index 5404a673..6d8c3321 100644 --- a/pkg/attestation/vtpm/vtpm_test.go +++ b/pkg/attestation/vtpm/vtpm_test.go @@ -371,17 +371,17 @@ func TestReadPolicy(t *testing.T) { require.NoError(t, err) defer os.RemoveAll(tempDir) - validPolicy := map[string]interface{}{ - "policy": map[string]interface{}{ - "product": map[string]interface{}{ + validPolicy := map[string]any{ + "policy": map[string]any{ + "product": map[string]any{ "name": "test-product", }, }, - "rootOfTrust": map[string]interface{}{ + "rootOfTrust": map[string]any{ "productLine": "test-line", }, - "pcrConfig": map[string]interface{}{ - "pcrValues": map[string]interface{}{ + "pcrConfig": map[string]any{ + "pcrValues": map[string]any{ "sha256": map[string]string{ "0": "0000000000000000000000000000000000000000000000000000000000000000", }, @@ -557,7 +557,7 @@ func TestConvertPolicyToJSON(t *testing.T) { assert.NoError(t, err) assert.NotNil(t, jsonData) - var result map[string]interface{} + var result map[string]any err = json.Unmarshal(jsonData, &result) assert.NoError(t, err) } diff --git a/pkg/clients/grpc/mocks/client.go b/pkg/clients/grpc/mocks/client.go index 76154ad4..c40cfca8 100644 --- a/pkg/clients/grpc/mocks/client.go +++ b/pkg/clients/grpc/mocks/client.go @@ -1,16 +1,31 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - google_golang_orggrpc "google.golang.org/grpc" - mock "github.com/stretchr/testify/mock" + "google.golang.org/grpc" ) +// NewClient creates a new instance of Client. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewClient(t interface { + mock.TestingT + Cleanup(func()) +}) *Client { + mock := &Client{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // Client is an autogenerated mock type for the Client type type Client struct { mock.Mock @@ -24,21 +39,20 @@ func (_m *Client) EXPECT() *Client_Expecter { return &Client_Expecter{mock: &_m.Mock} } -// Close provides a mock function with no fields -func (_m *Client) Close() error { - ret := _m.Called() +// Close provides a mock function for the type Client +func (_mock *Client) Close() error { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Close") } var r0 error - if rf, ok := ret.Get(0).(func() error); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() error); ok { + r0 = returnFunc() } else { r0 = ret.Error(0) } - return r0 } @@ -59,8 +73,8 @@ func (_c *Client_Close_Call) Run(run func()) *Client_Close_Call { return _c } -func (_c *Client_Close_Call) Return(_a0 error) *Client_Close_Call { - _c.Call.Return(_a0) +func (_c *Client_Close_Call) Return(err error) *Client_Close_Call { + _c.Call.Return(err) return _c } @@ -69,23 +83,22 @@ func (_c *Client_Close_Call) RunAndReturn(run func() error) *Client_Close_Call { return _c } -// Connection provides a mock function with no fields -func (_m *Client) Connection() *google_golang_orggrpc.ClientConn { - ret := _m.Called() +// Connection provides a mock function for the type Client +func (_mock *Client) Connection() *grpc.ClientConn { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Connection") } - var r0 *google_golang_orggrpc.ClientConn - if rf, ok := ret.Get(0).(func() *google_golang_orggrpc.ClientConn); ok { - r0 = rf() + var r0 *grpc.ClientConn + if returnFunc, ok := ret.Get(0).(func() *grpc.ClientConn); ok { + r0 = returnFunc() } else { if ret.Get(0) != nil { - r0 = ret.Get(0).(*google_golang_orggrpc.ClientConn) + r0 = ret.Get(0).(*grpc.ClientConn) } } - return r0 } @@ -106,31 +119,30 @@ func (_c *Client_Connection_Call) Run(run func()) *Client_Connection_Call { return _c } -func (_c *Client_Connection_Call) Return(_a0 *google_golang_orggrpc.ClientConn) *Client_Connection_Call { - _c.Call.Return(_a0) +func (_c *Client_Connection_Call) Return(clientConn *grpc.ClientConn) *Client_Connection_Call { + _c.Call.Return(clientConn) return _c } -func (_c *Client_Connection_Call) RunAndReturn(run func() *google_golang_orggrpc.ClientConn) *Client_Connection_Call { +func (_c *Client_Connection_Call) RunAndReturn(run func() *grpc.ClientConn) *Client_Connection_Call { _c.Call.Return(run) return _c } -// Secure provides a mock function with no fields -func (_m *Client) Secure() string { - ret := _m.Called() +// Secure provides a mock function for the type Client +func (_mock *Client) Secure() string { + ret := _mock.Called() if len(ret) == 0 { panic("no return value specified for Secure") } var r0 string - if rf, ok := ret.Get(0).(func() string); ok { - r0 = rf() + if returnFunc, ok := ret.Get(0).(func() string); ok { + r0 = returnFunc() } else { r0 = ret.Get(0).(string) } - return r0 } @@ -151,8 +163,8 @@ func (_c *Client_Secure_Call) Run(run func()) *Client_Secure_Call { return _c } -func (_c *Client_Secure_Call) Return(_a0 string) *Client_Secure_Call { - _c.Call.Return(_a0) +func (_c *Client_Secure_Call) Return(s string) *Client_Secure_Call { + _c.Call.Return(s) return _c } @@ -160,17 +172,3 @@ func (_c *Client_Secure_Call) RunAndReturn(run func() string) *Client_Secure_Cal _c.Call.Return(run) return _c } - -// NewClient creates a new instance of Client. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewClient(t interface { - mock.TestingT - Cleanup(func()) -}) *Client { - mock := &Client{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -} diff --git a/pkg/progressbar/progress_test.go b/pkg/progressbar/progress_test.go index 62f2428a..3cb2e21c 100644 --- a/pkg/progressbar/progress_test.go +++ b/pkg/progressbar/progress_test.go @@ -77,7 +77,7 @@ func TestSendAlgorithm(t *testing.T) { req, err = os.Open(req.Name()) assert.NoError(t, err) - algoStream := new(mocks.AgentService_AlgoClient[agent.AlgoRequest, agent.AlgoResponse]) + algoStream := new(mocks.AgentService_AlgoClient) algoStream.On("Send", mock.Anything).Return(tc.sendError) algoStream.On("CloseAndRecv").Return(&agent.AlgoResponse{}, tc.closeRecvError) mockStream := &mockAlgoStream{stream: algoStream} @@ -141,7 +141,7 @@ func TestSendData(t *testing.T) { dataset, err = os.Open(dataset.Name()) assert.NoError(t, err) - dataStream := new(mocks.AgentService_DataClient[agent.DataRequest, agent.DataResponse]) + dataStream := new(mocks.AgentService_DataClient) dataStream.On("Send", mock.Anything).Return(tc.sendError) dataStream.On("CloseAndRecv").Return(&agent.DataResponse{}, tc.closeRecvError) mockStream := &mockDataStream{stream: dataStream} @@ -566,7 +566,7 @@ func TestReceiverIMAMeasurements(t *testing.T) { for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { - mockStream := new(mocks.AgentService_IMAMeasurementsClient[agent.IMAMeasurementsResponse]) + mockStream := new(mocks.AgentService_IMAMeasurementsClient) p := New(true) p.TerminalWidthFunc = func() (int, error) { return 100, nil } diff --git a/pkg/progressbar/progressbar.go b/pkg/progressbar/progressbar.go index a32fd27d..12bf8557 100644 --- a/pkg/progressbar/progressbar.go +++ b/pkg/progressbar/progressbar.go @@ -27,15 +27,15 @@ var ( ) type streamSender interface { - Send(interface{}) error - CloseAndRecv() (interface{}, error) + Send(any) error + CloseAndRecv() (any, error) } type algoClientWrapper struct { client agent.AgentService_AlgoClient } -func (a *algoClientWrapper) Send(req interface{}) error { +func (a *algoClientWrapper) Send(req any) error { algoReq, ok := req.(*agent.AlgoRequest) if !ok { return fmt.Errorf("expected *AlgoRequest, got %T", req) @@ -44,7 +44,7 @@ func (a *algoClientWrapper) Send(req interface{}) error { return a.client.Send(algoReq) } -func (a *algoClientWrapper) CloseAndRecv() (interface{}, error) { +func (a *algoClientWrapper) CloseAndRecv() (any, error) { return a.client.CloseAndRecv() } @@ -52,7 +52,7 @@ type dataClientWrapper struct { client agent.AgentService_DataClient } -func (a *dataClientWrapper) Send(req interface{}) error { +func (a *dataClientWrapper) Send(req any) error { dataReq, ok := req.(*agent.DataRequest) if !ok { return fmt.Errorf("expected *DataRequest, got %T", req) @@ -61,7 +61,7 @@ func (a *dataClientWrapper) Send(req interface{}) error { return a.client.Send(dataReq) } -func (a *dataClientWrapper) CloseAndRecv() (interface{}, error) { +func (a *dataClientWrapper) CloseAndRecv() (any, error) { return a.client.CloseAndRecv() } @@ -104,7 +104,7 @@ func (p *ProgressBar) SendAlgorithm(description string, algo, req *os.File, stre // Send req first if req != nil { - if err := p.sendBuffer(req, wrapper, func(data []byte) interface{} { + if err := p.sendBuffer(req, wrapper, func(data []byte) any { return &agent.AlgoRequest{Requirements: data} }); err != nil { return err @@ -112,7 +112,7 @@ func (p *ProgressBar) SendAlgorithm(description string, algo, req *os.File, stre } // Then send algo - if err := p.sendBuffer(algo, wrapper, func(data []byte) interface{} { + if err := p.sendBuffer(algo, wrapper, func(data []byte) any { return &agent.AlgoRequest{Algorithm: data} }); err != nil { return err @@ -131,12 +131,12 @@ func (p *ProgressBar) SendAlgorithm(description string, algo, req *os.File, stre } func (p *ProgressBar) SendData(description, filename string, file *os.File, stream agent.AgentService_DataClient) error { - return p.sendData(description, file, &dataClientWrapper{client: stream}, func(data []byte) interface{} { + return p.sendData(description, file, &dataClientWrapper{client: stream}, func(data []byte) any { return &agent.DataRequest{Dataset: data, Filename: filename} }) } -func (p *ProgressBar) sendData(description string, file *os.File, stream streamSender, createRequest func([]byte) interface{}) error { +func (p *ProgressBar) sendData(description string, file *os.File, stream streamSender, createRequest func([]byte) any) error { dataInfo, err := file.Stat() if err != nil { return err @@ -176,7 +176,7 @@ func (p *ProgressBar) sendData(description string, file *os.File, stream streamS return err } -func (p *ProgressBar) sendBuffer(file *os.File, stream streamSender, createRequest func([]byte) interface{}) error { +func (p *ProgressBar) sendBuffer(file *os.File, stream streamSender, createRequest func([]byte) any) error { buf := make([]byte, bufferSize) for { diff --git a/pkg/sdk/mocks/sdk.go b/pkg/sdk/mocks/sdk.go index c6e0f0ac..d0630658 100644 --- a/pkg/sdk/mocks/sdk.go +++ b/pkg/sdk/mocks/sdk.go @@ -1,17 +1,33 @@ // Copyright (c) Ultraviolet // SPDX-License-Identifier: Apache-2.0 -// Code generated by mockery v2.53.3. DO NOT EDIT. +// Code generated by mockery; DO NOT EDIT. +// github.com/vektra/mockery +// template: testify package mocks import ( - context "context" - os "os" + "context" + "os" mock "github.com/stretchr/testify/mock" ) +// NewSDK creates a new instance of SDK. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. +// The first argument is typically a *testing.T value. +func NewSDK(t interface { + mock.TestingT + Cleanup(func()) +}) *SDK { + mock := &SDK{} + mock.Mock.Test(t) + + t.Cleanup(func() { mock.AssertExpectations(t) }) + + return mock +} + // SDK is an autogenerated mock type for the SDK type type SDK struct { mock.Mock @@ -25,21 +41,20 @@ func (_m *SDK) EXPECT() *SDK_Expecter { return &SDK_Expecter{mock: &_m.Mock} } -// Algo provides a mock function with given fields: ctx, algorithm, requirements, privKey -func (_m *SDK) Algo(ctx context.Context, algorithm *os.File, requirements *os.File, privKey interface{}) error { - ret := _m.Called(ctx, algorithm, requirements, privKey) +// Algo provides a mock function for the type SDK +func (_mock *SDK) Algo(ctx context.Context, algorithm *os.File, requirements *os.File, privKey any) error { + ret := _mock.Called(ctx, algorithm, requirements, privKey) if len(ret) == 0 { panic("no return value specified for Algo") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *os.File, *os.File, interface{}) error); ok { - r0 = rf(ctx, algorithm, requirements, privKey) + if returnFunc, ok := ret.Get(0).(func(context.Context, *os.File, *os.File, any) error); ok { + r0 = returnFunc(ctx, algorithm, requirements, privKey) } else { r0 = ret.Error(0) } - return r0 } @@ -52,43 +67,63 @@ type SDK_Algo_Call struct { // - ctx context.Context // - algorithm *os.File // - requirements *os.File -// - privKey interface{} +// - privKey any func (_e *SDK_Expecter) Algo(ctx interface{}, algorithm interface{}, requirements interface{}, privKey interface{}) *SDK_Algo_Call { return &SDK_Algo_Call{Call: _e.mock.On("Algo", ctx, algorithm, requirements, privKey)} } -func (_c *SDK_Algo_Call) Run(run func(ctx context.Context, algorithm *os.File, requirements *os.File, privKey interface{})) *SDK_Algo_Call { +func (_c *SDK_Algo_Call) Run(run func(ctx context.Context, algorithm *os.File, requirements *os.File, privKey any)) *SDK_Algo_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(*os.File), args[2].(*os.File), args[3].(interface{})) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *os.File + if args[1] != nil { + arg1 = args[1].(*os.File) + } + var arg2 *os.File + if args[2] != nil { + arg2 = args[2].(*os.File) + } + var arg3 any + if args[3] != nil { + arg3 = args[3].(any) + } + run( + arg0, + arg1, + arg2, + arg3, + ) }) return _c } -func (_c *SDK_Algo_Call) Return(_a0 error) *SDK_Algo_Call { - _c.Call.Return(_a0) +func (_c *SDK_Algo_Call) Return(err error) *SDK_Algo_Call { + _c.Call.Return(err) return _c } -func (_c *SDK_Algo_Call) RunAndReturn(run func(context.Context, *os.File, *os.File, interface{}) error) *SDK_Algo_Call { +func (_c *SDK_Algo_Call) RunAndReturn(run func(ctx context.Context, algorithm *os.File, requirements *os.File, privKey any) error) *SDK_Algo_Call { _c.Call.Return(run) return _c } -// Attestation provides a mock function with given fields: ctx, reportData, nonce, attType, attestationFile -func (_m *SDK) Attestation(ctx context.Context, reportData [64]byte, nonce [32]byte, attType int, attestationFile *os.File) error { - ret := _m.Called(ctx, reportData, nonce, attType, attestationFile) +// Attestation provides a mock function for the type SDK +func (_mock *SDK) Attestation(ctx context.Context, reportData [64]byte, nonce [32]byte, attType int, attestationFile *os.File) error { + ret := _mock.Called(ctx, reportData, nonce, attType, attestationFile) if len(ret) == 0 { panic("no return value specified for Attestation") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, [64]byte, [32]byte, int, *os.File) error); ok { - r0 = rf(ctx, reportData, nonce, attType, attestationFile) + if returnFunc, ok := ret.Get(0).(func(context.Context, [64]byte, [32]byte, int, *os.File) error); ok { + r0 = returnFunc(ctx, reportData, nonce, attType, attestationFile) } else { r0 = ret.Error(0) } - return r0 } @@ -109,36 +144,61 @@ func (_e *SDK_Expecter) Attestation(ctx interface{}, reportData interface{}, non func (_c *SDK_Attestation_Call) Run(run func(ctx context.Context, reportData [64]byte, nonce [32]byte, attType int, attestationFile *os.File)) *SDK_Attestation_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].([64]byte), args[2].([32]byte), args[3].(int), args[4].(*os.File)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 [64]byte + if args[1] != nil { + arg1 = args[1].([64]byte) + } + var arg2 [32]byte + if args[2] != nil { + arg2 = args[2].([32]byte) + } + var arg3 int + if args[3] != nil { + arg3 = args[3].(int) + } + var arg4 *os.File + if args[4] != nil { + arg4 = args[4].(*os.File) + } + run( + arg0, + arg1, + arg2, + arg3, + arg4, + ) }) return _c } -func (_c *SDK_Attestation_Call) Return(_a0 error) *SDK_Attestation_Call { - _c.Call.Return(_a0) +func (_c *SDK_Attestation_Call) Return(err error) *SDK_Attestation_Call { + _c.Call.Return(err) return _c } -func (_c *SDK_Attestation_Call) RunAndReturn(run func(context.Context, [64]byte, [32]byte, int, *os.File) error) *SDK_Attestation_Call { +func (_c *SDK_Attestation_Call) RunAndReturn(run func(ctx context.Context, reportData [64]byte, nonce [32]byte, attType int, attestationFile *os.File) error) *SDK_Attestation_Call { _c.Call.Return(run) return _c } -// AttestationToken provides a mock function with given fields: ctx, nonce, attType, attestationFile -func (_m *SDK) AttestationToken(ctx context.Context, nonce [32]byte, attType int, attestationFile *os.File) error { - ret := _m.Called(ctx, nonce, attType, attestationFile) +// AttestationToken provides a mock function for the type SDK +func (_mock *SDK) AttestationToken(ctx context.Context, nonce [32]byte, attType int, attestationFile *os.File) error { + ret := _mock.Called(ctx, nonce, attType, attestationFile) if len(ret) == 0 { panic("no return value specified for AttestationToken") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, [32]byte, int, *os.File) error); ok { - r0 = rf(ctx, nonce, attType, attestationFile) + if returnFunc, ok := ret.Get(0).(func(context.Context, [32]byte, int, *os.File) error); ok { + r0 = returnFunc(ctx, nonce, attType, attestationFile) } else { r0 = ret.Error(0) } - return r0 } @@ -158,36 +218,56 @@ func (_e *SDK_Expecter) AttestationToken(ctx interface{}, nonce interface{}, att func (_c *SDK_AttestationToken_Call) Run(run func(ctx context.Context, nonce [32]byte, attType int, attestationFile *os.File)) *SDK_AttestationToken_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].([32]byte), args[2].(int), args[3].(*os.File)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 [32]byte + if args[1] != nil { + arg1 = args[1].([32]byte) + } + var arg2 int + if args[2] != nil { + arg2 = args[2].(int) + } + var arg3 *os.File + if args[3] != nil { + arg3 = args[3].(*os.File) + } + run( + arg0, + arg1, + arg2, + arg3, + ) }) return _c } -func (_c *SDK_AttestationToken_Call) Return(_a0 error) *SDK_AttestationToken_Call { - _c.Call.Return(_a0) +func (_c *SDK_AttestationToken_Call) Return(err error) *SDK_AttestationToken_Call { + _c.Call.Return(err) return _c } -func (_c *SDK_AttestationToken_Call) RunAndReturn(run func(context.Context, [32]byte, int, *os.File) error) *SDK_AttestationToken_Call { +func (_c *SDK_AttestationToken_Call) RunAndReturn(run func(ctx context.Context, nonce [32]byte, attType int, attestationFile *os.File) error) *SDK_AttestationToken_Call { _c.Call.Return(run) return _c } -// Data provides a mock function with given fields: ctx, dataset, filename, privKey -func (_m *SDK) Data(ctx context.Context, dataset *os.File, filename string, privKey interface{}) error { - ret := _m.Called(ctx, dataset, filename, privKey) +// Data provides a mock function for the type SDK +func (_mock *SDK) Data(ctx context.Context, dataset *os.File, filename string, privKey any) error { + ret := _mock.Called(ctx, dataset, filename, privKey) if len(ret) == 0 { panic("no return value specified for Data") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, *os.File, string, interface{}) error); ok { - r0 = rf(ctx, dataset, filename, privKey) + if returnFunc, ok := ret.Get(0).(func(context.Context, *os.File, string, any) error); ok { + r0 = returnFunc(ctx, dataset, filename, privKey) } else { r0 = ret.Error(0) } - return r0 } @@ -200,31 +280,52 @@ type SDK_Data_Call struct { // - ctx context.Context // - dataset *os.File // - filename string -// - privKey interface{} +// - privKey any func (_e *SDK_Expecter) Data(ctx interface{}, dataset interface{}, filename interface{}, privKey interface{}) *SDK_Data_Call { return &SDK_Data_Call{Call: _e.mock.On("Data", ctx, dataset, filename, privKey)} } -func (_c *SDK_Data_Call) Run(run func(ctx context.Context, dataset *os.File, filename string, privKey interface{})) *SDK_Data_Call { +func (_c *SDK_Data_Call) Run(run func(ctx context.Context, dataset *os.File, filename string, privKey any)) *SDK_Data_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(*os.File), args[2].(string), args[3].(interface{})) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *os.File + if args[1] != nil { + arg1 = args[1].(*os.File) + } + var arg2 string + if args[2] != nil { + arg2 = args[2].(string) + } + var arg3 any + if args[3] != nil { + arg3 = args[3].(any) + } + run( + arg0, + arg1, + arg2, + arg3, + ) }) return _c } -func (_c *SDK_Data_Call) Return(_a0 error) *SDK_Data_Call { - _c.Call.Return(_a0) +func (_c *SDK_Data_Call) Return(err error) *SDK_Data_Call { + _c.Call.Return(err) return _c } -func (_c *SDK_Data_Call) RunAndReturn(run func(context.Context, *os.File, string, interface{}) error) *SDK_Data_Call { +func (_c *SDK_Data_Call) RunAndReturn(run func(ctx context.Context, dataset *os.File, filename string, privKey any) error) *SDK_Data_Call { _c.Call.Return(run) return _c } -// IMAMeasurements provides a mock function with given fields: ctx, resultFile -func (_m *SDK) IMAMeasurements(ctx context.Context, resultFile *os.File) ([]byte, error) { - ret := _m.Called(ctx, resultFile) +// IMAMeasurements provides a mock function for the type SDK +func (_mock *SDK) IMAMeasurements(ctx context.Context, resultFile *os.File) ([]byte, error) { + ret := _mock.Called(ctx, resultFile) if len(ret) == 0 { panic("no return value specified for IMAMeasurements") @@ -232,23 +333,21 @@ func (_m *SDK) IMAMeasurements(ctx context.Context, resultFile *os.File) ([]byte var r0 []byte var r1 error - if rf, ok := ret.Get(0).(func(context.Context, *os.File) ([]byte, error)); ok { - return rf(ctx, resultFile) + if returnFunc, ok := ret.Get(0).(func(context.Context, *os.File) ([]byte, error)); ok { + return returnFunc(ctx, resultFile) } - if rf, ok := ret.Get(0).(func(context.Context, *os.File) []byte); ok { - r0 = rf(ctx, resultFile) + if returnFunc, ok := ret.Get(0).(func(context.Context, *os.File) []byte); ok { + r0 = returnFunc(ctx, resultFile) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]byte) } } - - if rf, ok := ret.Get(1).(func(context.Context, *os.File) error); ok { - r1 = rf(ctx, resultFile) + if returnFunc, ok := ret.Get(1).(func(context.Context, *os.File) error); ok { + r1 = returnFunc(ctx, resultFile) } else { r1 = ret.Error(1) } - return r0, r1 } @@ -266,36 +365,46 @@ func (_e *SDK_Expecter) IMAMeasurements(ctx interface{}, resultFile interface{}) func (_c *SDK_IMAMeasurements_Call) Run(run func(ctx context.Context, resultFile *os.File)) *SDK_IMAMeasurements_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(*os.File)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 *os.File + if args[1] != nil { + arg1 = args[1].(*os.File) + } + run( + arg0, + arg1, + ) }) return _c } -func (_c *SDK_IMAMeasurements_Call) Return(_a0 []byte, _a1 error) *SDK_IMAMeasurements_Call { - _c.Call.Return(_a0, _a1) +func (_c *SDK_IMAMeasurements_Call) Return(bytes []byte, err error) *SDK_IMAMeasurements_Call { + _c.Call.Return(bytes, err) return _c } -func (_c *SDK_IMAMeasurements_Call) RunAndReturn(run func(context.Context, *os.File) ([]byte, error)) *SDK_IMAMeasurements_Call { +func (_c *SDK_IMAMeasurements_Call) RunAndReturn(run func(ctx context.Context, resultFile *os.File) ([]byte, error)) *SDK_IMAMeasurements_Call { _c.Call.Return(run) return _c } -// Result provides a mock function with given fields: ctx, privKey, resultFile -func (_m *SDK) Result(ctx context.Context, privKey interface{}, resultFile *os.File) error { - ret := _m.Called(ctx, privKey, resultFile) +// Result provides a mock function for the type SDK +func (_mock *SDK) Result(ctx context.Context, privKey any, resultFile *os.File) error { + ret := _mock.Called(ctx, privKey, resultFile) if len(ret) == 0 { panic("no return value specified for Result") } var r0 error - if rf, ok := ret.Get(0).(func(context.Context, interface{}, *os.File) error); ok { - r0 = rf(ctx, privKey, resultFile) + if returnFunc, ok := ret.Get(0).(func(context.Context, any, *os.File) error); ok { + r0 = returnFunc(ctx, privKey, resultFile) } else { r0 = ret.Error(0) } - return r0 } @@ -306,39 +415,41 @@ type SDK_Result_Call struct { // Result is a helper method to define mock.On call // - ctx context.Context -// - privKey interface{} +// - privKey any // - resultFile *os.File func (_e *SDK_Expecter) Result(ctx interface{}, privKey interface{}, resultFile interface{}) *SDK_Result_Call { return &SDK_Result_Call{Call: _e.mock.On("Result", ctx, privKey, resultFile)} } -func (_c *SDK_Result_Call) Run(run func(ctx context.Context, privKey interface{}, resultFile *os.File)) *SDK_Result_Call { +func (_c *SDK_Result_Call) Run(run func(ctx context.Context, privKey any, resultFile *os.File)) *SDK_Result_Call { _c.Call.Run(func(args mock.Arguments) { - run(args[0].(context.Context), args[1].(interface{}), args[2].(*os.File)) + var arg0 context.Context + if args[0] != nil { + arg0 = args[0].(context.Context) + } + var arg1 any + if args[1] != nil { + arg1 = args[1].(any) + } + var arg2 *os.File + if args[2] != nil { + arg2 = args[2].(*os.File) + } + run( + arg0, + arg1, + arg2, + ) }) return _c } -func (_c *SDK_Result_Call) Return(_a0 error) *SDK_Result_Call { - _c.Call.Return(_a0) +func (_c *SDK_Result_Call) Return(err error) *SDK_Result_Call { + _c.Call.Return(err) return _c } -func (_c *SDK_Result_Call) RunAndReturn(run func(context.Context, interface{}, *os.File) error) *SDK_Result_Call { +func (_c *SDK_Result_Call) RunAndReturn(run func(ctx context.Context, privKey any, resultFile *os.File) error) *SDK_Result_Call { _c.Call.Return(run) return _c } - -// NewSDK creates a new instance of SDK. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. -// The first argument is typically a *testing.T value. -func NewSDK(t interface { - mock.TestingT - Cleanup(func()) -}) *SDK { - mock := &SDK{} - mock.Mock.Test(t) - - t.Cleanup(func() { mock.AssertExpectations(t) }) - - return mock -}