github.com/khulnasoft-lab/tunnel-db@v0.0.0-20231117205118-74e1113bd007/pkg/dbtest/mock_bucketer.go (about)

     1  // Code generated by mockery v1.0.0. DO NOT EDIT.
     2  
     3  package dbtest
     4  
     5  import (
     6  	mock "github.com/stretchr/testify/mock"
     7  	bbolt "go.etcd.io/bbolt"
     8  )
     9  
    10  // mockBucketer is an autogenerated mock type for the bucketer type
    11  type mockBucketer struct {
    12  	mock.Mock
    13  }
    14  
    15  type bucketerBucketArgs struct {
    16  	Name         []byte
    17  	NameAnything bool
    18  }
    19  
    20  type bucketerBucketReturns struct {
    21  	_a0 *bbolt.Bucket
    22  }
    23  
    24  type bucketerBucketExpectation struct {
    25  	Args    bucketerBucketArgs
    26  	Returns bucketerBucketReturns
    27  }
    28  
    29  func (_m *mockBucketer) ApplyBucketExpectation(e bucketerBucketExpectation) {
    30  	var args []interface{}
    31  	if e.Args.NameAnything {
    32  		args = append(args, mock.Anything)
    33  	} else {
    34  		args = append(args, e.Args.Name)
    35  	}
    36  	_m.On("Bucket", args...).Return(e.Returns._a0)
    37  }
    38  
    39  func (_m *mockBucketer) ApplyBucketExpectations(expectations []bucketerBucketExpectation) {
    40  	for _, e := range expectations {
    41  		_m.ApplyBucketExpectation(e)
    42  	}
    43  }
    44  
    45  // Bucket provides a mock function with given fields: name
    46  func (_m *mockBucketer) Bucket(name []byte) *bbolt.Bucket {
    47  	ret := _m.Called(name)
    48  
    49  	var r0 *bbolt.Bucket
    50  	if rf, ok := ret.Get(0).(func([]byte) *bbolt.Bucket); ok {
    51  		r0 = rf(name)
    52  	} else {
    53  		if ret.Get(0) != nil {
    54  			r0 = ret.Get(0).(*bbolt.Bucket)
    55  		}
    56  	}
    57  
    58  	return r0
    59  }