github.com/treeverse/lakefs@v1.24.1-0.20240520134607-95648127bfb0/pkg/graveler/testutil/graveler_mock.go (about)

     1  package testutil
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/golang/mock/gomock"
     7  	"github.com/treeverse/lakefs/pkg/graveler"
     8  	"github.com/treeverse/lakefs/pkg/graveler/mock"
     9  	kvmock "github.com/treeverse/lakefs/pkg/kv/mock"
    10  )
    11  
    12  type GravelerTest struct {
    13  	Controller               *gomock.Controller
    14  	CommittedManager         *mock.MockCommittedManager
    15  	RefManager               *mock.MockRefManager
    16  	StagingManager           *mock.MockStagingManager
    17  	ProtectedBranchesManager *mock.MockProtectedBranchesManager
    18  	GarbageCollectionManager *mock.MockGarbageCollectionManager
    19  	KVStore                  *kvmock.MockStore
    20  	Sut                      *graveler.Graveler
    21  }
    22  
    23  func InitGravelerTest(t *testing.T) *GravelerTest {
    24  	ctrl := gomock.NewController(t)
    25  
    26  	test := &GravelerTest{
    27  		Controller:               ctrl,
    28  		CommittedManager:         mock.NewMockCommittedManager(ctrl),
    29  		StagingManager:           mock.NewMockStagingManager(ctrl),
    30  		RefManager:               mock.NewMockRefManager(ctrl),
    31  		GarbageCollectionManager: mock.NewMockGarbageCollectionManager(ctrl),
    32  		ProtectedBranchesManager: mock.NewMockProtectedBranchesManager(ctrl),
    33  		KVStore:                  kvmock.NewMockStore(ctrl),
    34  	}
    35  
    36  	test.Sut = graveler.NewGraveler(test.CommittedManager, test.StagingManager, test.RefManager, test.GarbageCollectionManager, test.ProtectedBranchesManager, nil)
    37  
    38  	return test
    39  }