mirror of
https://github.com/ultravioletrs/cocos.git
synced 2026-06-23 04:10:25 +00:00
3102114ff3
* add CC platform identification capability * add token verification * add snp azure * add azure snp report verification * fix linter errors * fix agent tests * expand the CC provider * fix azure atls * rebase branch * add nonce check for azure token * rename package attestations * remove alias attestations --------- Co-authored-by: Ubuntu <azureuser@UVCTestCVM.bu0p0zdolasezg1jifpyqhaxuc.dx.internal.cloudapp.net>
353 lines
10 KiB
Go
353 lines
10 KiB
Go
// Copyright (c) Ultraviolet
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
// Code generated by mockery v2.53.3. DO NOT EDIT.
|
|
|
|
package mocks
|
|
|
|
import mock "github.com/stretchr/testify/mock"
|
|
|
|
// Provider is an autogenerated mock type for the Provider type
|
|
type Provider struct {
|
|
mock.Mock
|
|
}
|
|
|
|
type Provider_Expecter struct {
|
|
mock *mock.Mock
|
|
}
|
|
|
|
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)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Attestation")
|
|
}
|
|
|
|
var r0 []byte
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func([]byte, []byte) ([]byte, error)); ok {
|
|
return rf(teeNonce, vTpmNonce)
|
|
}
|
|
if rf, ok := ret.Get(0).(func([]byte, []byte) []byte); ok {
|
|
r0 = rf(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)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// Provider_Attestation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Attestation'
|
|
type Provider_Attestation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Attestation is a helper method to define mock.On call
|
|
// - teeNonce []byte
|
|
// - vTpmNonce []byte
|
|
func (_e *Provider_Expecter) Attestation(teeNonce interface{}, vTpmNonce interface{}) *Provider_Attestation_Call {
|
|
return &Provider_Attestation_Call{Call: _e.mock.On("Attestation", teeNonce, vTpmNonce)}
|
|
}
|
|
|
|
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))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_Attestation_Call) Return(_a0 []byte, _a1 error) *Provider_Attestation_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_Attestation_Call) RunAndReturn(run func([]byte, []byte) ([]byte, error)) *Provider_Attestation_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)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for TeeAttestation")
|
|
}
|
|
|
|
var r0 []byte
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok {
|
|
return rf(teeNonce)
|
|
}
|
|
if rf, ok := ret.Get(0).(func([]byte) []byte); ok {
|
|
r0 = rf(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)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// Provider_TeeAttestation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'TeeAttestation'
|
|
type Provider_TeeAttestation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// TeeAttestation is a helper method to define mock.On call
|
|
// - teeNonce []byte
|
|
func (_e *Provider_Expecter) TeeAttestation(teeNonce interface{}) *Provider_TeeAttestation_Call {
|
|
return &Provider_TeeAttestation_Call{Call: _e.mock.On("TeeAttestation", teeNonce)}
|
|
}
|
|
|
|
func (_c *Provider_TeeAttestation_Call) Run(run func(teeNonce []byte)) *Provider_TeeAttestation_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].([]byte))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_TeeAttestation_Call) Return(_a0 []byte, _a1 error) *Provider_TeeAttestation_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_TeeAttestation_Call) RunAndReturn(run func([]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)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for VTpmAttestation")
|
|
}
|
|
|
|
var r0 []byte
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func([]byte) ([]byte, error)); ok {
|
|
return rf(vTpmNonce)
|
|
}
|
|
if rf, ok := ret.Get(0).(func([]byte) []byte); ok {
|
|
r0 = rf(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)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// Provider_VTpmAttestation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'VTpmAttestation'
|
|
type Provider_VTpmAttestation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// VTpmAttestation is a helper method to define mock.On call
|
|
// - vTpmNonce []byte
|
|
func (_e *Provider_Expecter) VTpmAttestation(vTpmNonce interface{}) *Provider_VTpmAttestation_Call {
|
|
return &Provider_VTpmAttestation_Call{Call: _e.mock.On("VTpmAttestation", vTpmNonce)}
|
|
}
|
|
|
|
func (_c *Provider_VTpmAttestation_Call) Run(run func(vTpmNonce []byte)) *Provider_VTpmAttestation_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].([]byte))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VTpmAttestation_Call) Return(_a0 []byte, _a1 error) *Provider_VTpmAttestation_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VTpmAttestation_Call) RunAndReturn(run func([]byte) ([]byte, error)) *Provider_VTpmAttestation_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// VerifTeeAttestation provides a mock function with given fields: report, teeNonce
|
|
func (_m *Provider) VerifTeeAttestation(report []byte, teeNonce []byte) error {
|
|
ret := _m.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)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// Provider_VerifTeeAttestation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'VerifTeeAttestation'
|
|
type Provider_VerifTeeAttestation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// VerifTeeAttestation is a helper method to define mock.On call
|
|
// - report []byte
|
|
// - teeNonce []byte
|
|
func (_e *Provider_Expecter) VerifTeeAttestation(report interface{}, teeNonce interface{}) *Provider_VerifTeeAttestation_Call {
|
|
return &Provider_VerifTeeAttestation_Call{Call: _e.mock.On("VerifTeeAttestation", report, teeNonce)}
|
|
}
|
|
|
|
func (_c *Provider_VerifTeeAttestation_Call) Run(run func(report []byte, teeNonce []byte)) *Provider_VerifTeeAttestation_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].([]byte), args[1].([]byte))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VerifTeeAttestation_Call) Return(_a0 error) *Provider_VerifTeeAttestation_Call {
|
|
_c.Call.Return(_a0)
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VerifTeeAttestation_Call) RunAndReturn(run func([]byte, []byte) error) *Provider_VerifTeeAttestation_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// VerifVTpmAttestation provides a mock function with given fields: report, vTpmNonce
|
|
func (_m *Provider) VerifVTpmAttestation(report []byte, vTpmNonce []byte) error {
|
|
ret := _m.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)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// Provider_VerifVTpmAttestation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'VerifVTpmAttestation'
|
|
type Provider_VerifVTpmAttestation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// VerifVTpmAttestation is a helper method to define mock.On call
|
|
// - report []byte
|
|
// - vTpmNonce []byte
|
|
func (_e *Provider_Expecter) VerifVTpmAttestation(report interface{}, vTpmNonce interface{}) *Provider_VerifVTpmAttestation_Call {
|
|
return &Provider_VerifVTpmAttestation_Call{Call: _e.mock.On("VerifVTpmAttestation", report, vTpmNonce)}
|
|
}
|
|
|
|
func (_c *Provider_VerifVTpmAttestation_Call) Run(run func(report []byte, vTpmNonce []byte)) *Provider_VerifVTpmAttestation_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].([]byte), args[1].([]byte))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VerifVTpmAttestation_Call) Return(_a0 error) *Provider_VerifVTpmAttestation_Call {
|
|
_c.Call.Return(_a0)
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VerifVTpmAttestation_Call) RunAndReturn(run func([]byte, []byte) error) *Provider_VerifVTpmAttestation_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// VerifyAttestation provides a mock function with given fields: report, teeNonce, vTpmNonce
|
|
func (_m *Provider) VerifyAttestation(report []byte, teeNonce []byte, vTpmNonce []byte) error {
|
|
ret := _m.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)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// Provider_VerifyAttestation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'VerifyAttestation'
|
|
type Provider_VerifyAttestation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// VerifyAttestation is a helper method to define mock.On call
|
|
// - report []byte
|
|
// - teeNonce []byte
|
|
// - vTpmNonce []byte
|
|
func (_e *Provider_Expecter) VerifyAttestation(report interface{}, teeNonce interface{}, vTpmNonce interface{}) *Provider_VerifyAttestation_Call {
|
|
return &Provider_VerifyAttestation_Call{Call: _e.mock.On("VerifyAttestation", report, teeNonce, vTpmNonce)}
|
|
}
|
|
|
|
func (_c *Provider_VerifyAttestation_Call) Run(run func(report []byte, teeNonce []byte, vTpmNonce []byte)) *Provider_VerifyAttestation_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].([]byte), args[1].([]byte), args[2].([]byte))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VerifyAttestation_Call) Return(_a0 error) *Provider_VerifyAttestation_Call {
|
|
_c.Call.Return(_a0)
|
|
return _c
|
|
}
|
|
|
|
func (_c *Provider_VerifyAttestation_Call) RunAndReturn(run func([]byte, []byte, []byte) error) *Provider_VerifyAttestation_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
|
|
}
|