NVIDIA GPU

/*

**   Copyright 2001-2024 Zabbix SIA

**

**   Licensed under the Apache License, Version 2.0 (the "License");

**   you may not use this file except in compliance with the License.

**   You may obtain a copy of the License at

**

**       http://www.apache.org/licenses/LICENSE-2.0

**

**   Unless required by applicable law or agreed to in writing, software

**   distributed under the License is distributed on an "AS IS" BASIS,

**   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

**   See the License for the specific language governing permissions and

**   limitations under the License.

**/

​

package nvmlmock

​

import (

    "testing"

​

    "github.com/google/go-cmp/cmp"

    "golang.zabbix.com/plugin/nvidia/pkg/nvml"

)

​

var (

    _ nvml.Runner = (*MockRunner)(nil)

    _ Mocker      = (*MockRunner)(nil)

)

​

// Mocker any mock should implement to collect information if all its and submock calls are done.

type Mocker interface {

    ExpectedCallsDone() bool

    SubMocks() []Mocker

}

​

// MockRunner is mock for NVML Runner.

type MockRunner struct {

    nvml.Runner

    expectations []*Expectation

    callIdx      int

    t            *testing.T

}

​

// NewMockRunner creates new mock runner.

func NewMockRunner(t *testing.T) *MockRunner {

    t.Helper()

​

    return &MockRunner{

        t:            t,

        expectations: []*Expectation{},

    }

}

​

// ExpectCalls sets calls that are expected by mock.

func (m *MockRunner) ExpectCalls(expectations ...*Expectation) *MockRunner {

    m.expectations = expectations

​

    return m

}

​

// SubMocks returns submocks of the mock.

func (m *MockRunner) SubMocks() []Mocker {