supplycommit: export mock functionality for use in supplyverifier

Author: ffranr

universe/supplycommit/mock.go

@@ -59,21 +59,21 @@ func (m *mockSupplyTreeView) FetchSupplyLeavesByHeight(_ context.Context,
 	return args.Get(0).(lfn.Result[SupplyLeaves])
 }
 
-// mockCommitmentTracker is a mock implementation of the CommitmentTracker
+// MockCommitmentTracker is a mock implementation of the CommitmentTracker
 // interface.
-type mockCommitmentTracker struct {
+type MockCommitmentTracker struct {
 	mock.Mock
 }
 
-func (m *mockCommitmentTracker) UnspentPrecommits(ctx context.Context,
+func (m *MockCommitmentTracker) UnspentPrecommits(ctx context.Context,
 	assetSpec asset.Specifier,
 	localIssuerOnly bool) lfn.Result[PreCommits] {
 
 	args := m.Called(ctx, assetSpec, localIssuerOnly)
 	return args.Get(0).(lfn.Result[PreCommits])
 }
 
-func (m *mockCommitmentTracker) SupplyCommit(ctx context.Context,
+func (m *MockCommitmentTracker) SupplyCommit(ctx context.Context,
 	assetSpec asset.Specifier) RootCommitResp {
 
 	args := m.Called(ctx, assetSpec)
@@ -432,12 +432,12 @@ func (c *mockIgnoreCheckerCache) InvalidateCache(groupKey btcec.PublicKey) {
 	c.Called(groupKey)
 }
 
-// mockAssetLookup is a mock implementation of the AssetLookup interface.
-type mockAssetLookup struct {
+// MockAssetLookup is a mock implementation of the AssetLookup interface.
+type MockAssetLookup struct {
 	mock.Mock
 }
 
-func (m *mockAssetLookup) FetchSupplyCommitAssets(ctx context.Context,
+func (m *MockAssetLookup) FetchSupplyCommitAssets(ctx context.Context,
 	localControlled bool) ([]btcec.PublicKey, error) {
 
 	args := m.Called(ctx, localControlled)
@@ -447,7 +447,7 @@ func (m *mockAssetLookup) FetchSupplyCommitAssets(ctx context.Context,
 	return args.Get(0).([]btcec.PublicKey), args.Error(1)
 }
 
-func (m *mockAssetLookup) QueryAssetGroupByID(ctx context.Context,
+func (m *MockAssetLookup) QueryAssetGroupByID(ctx context.Context,
 	assetID asset.ID) (*asset.AssetGroup, error) {
 
 	args := m.Called(ctx, assetID)
@@ -457,7 +457,7 @@ func (m *mockAssetLookup) QueryAssetGroupByID(ctx context.Context,
 	return args.Get(0).(*asset.AssetGroup), args.Error(1)
 }
 
-func (m *mockAssetLookup) QueryAssetGroupByGroupKey(ctx context.Context,
+func (m *MockAssetLookup) QueryAssetGroupByGroupKey(ctx context.Context,
 	groupKey *btcec.PublicKey) (*asset.AssetGroup, error) {
 
 	args := m.Called(ctx, groupKey)
@@ -467,7 +467,7 @@ func (m *mockAssetLookup) QueryAssetGroupByGroupKey(ctx context.Context,
 	return args.Get(0).(*asset.AssetGroup), args.Error(1)
 }
 
-func (m *mockAssetLookup) FetchAssetMetaForAsset(ctx context.Context,
+func (m *MockAssetLookup) FetchAssetMetaForAsset(ctx context.Context,
 	assetID asset.ID) (*proof.MetaReveal, error) {
 
 	args := m.Called(ctx, assetID)
@@ -477,7 +477,7 @@ func (m *mockAssetLookup) FetchAssetMetaForAsset(ctx context.Context,
 	return args.Get(0).(*proof.MetaReveal), args.Error(1)
 }
 
-func (m *mockAssetLookup) FetchInternalKeyLocator(ctx context.Context,
+func (m *MockAssetLookup) FetchInternalKeyLocator(ctx context.Context,
 	rawKey *btcec.PublicKey) (keychain.KeyLocator, error) {
 
 	args := m.Called(ctx, rawKey)

universe/supplycommit/state_machine_test.go

@@ -121,15 +121,15 @@ type supplyCommitTestHarness struct {
 	env          *Environment
 
 	mockTreeView     *mockSupplyTreeView
-	mockCommits      *mockCommitmentTracker
+	mockCommits      *MockCommitmentTracker
 	mockWallet       *mockWallet
 	mockKeyRing      *mockKeyRing
 	mockChain        *mockChainBridge
 	mockStateLog     *mockStateMachineStore
 	mockCache        *mockIgnoreCheckerCache
 	mockDaemon       *mockDaemonAdapters
 	mockErrReporter  *mockErrorReporter
-	mockAssetLookup  *mockAssetLookup
+	MockAssetLookup  *MockAssetLookup
 	mockSupplySyncer *mockSupplySyncer
 
 	stateSub protofsm.StateSubscriber[Event, *Environment]
@@ -139,15 +139,15 @@ func newSupplyCommitTestHarness(t *testing.T,
 	cfg *harnessCfg) *supplyCommitTestHarness {
 
 	mTreeView := &mockSupplyTreeView{}
-	mCommits := &mockCommitmentTracker{}
+	mCommits := &MockCommitmentTracker{}
 	mWallet := &mockWallet{}
 	mKey := &mockKeyRing{}
 	mChain := &mockChainBridge{}
 	mStateLog := &mockStateMachineStore{}
 	mDaemon := newMockDaemonAdapters()
 	mErrReporter := &mockErrorReporter{}
 	mCache := &mockIgnoreCheckerCache{}
-	mAssetLookup := &mockAssetLookup{}
+	mAssetLookup := &MockAssetLookup{}
 	mSupplySyncer := &mockSupplySyncer{}
 
 	env := &Environment{
@@ -190,7 +190,7 @@ func newSupplyCommitTestHarness(t *testing.T,
 		mockCache:        mCache,
 		mockDaemon:       mDaemon,
 		mockErrReporter:  mErrReporter,
-		mockAssetLookup:  mAssetLookup,
+		MockAssetLookup:  mAssetLookup,
 		mockSupplySyncer: mSupplySyncer,
 	}
 
@@ -552,7 +552,7 @@ func (h *supplyCommitTestHarness) expectAssetLookup() {
 		},
 	}
 
-	h.mockAssetLookup.On(
+	h.MockAssetLookup.On(
 		"QueryAssetGroupByGroupKey", mock.Anything, mock.Anything,
 	).Return(dummyAssetGroup, nil).Maybe()
 
@@ -562,7 +562,7 @@ func (h *supplyCommitTestHarness) expectAssetLookup() {
 		Type: proof.MetaOpaque,
 	}
 
-	h.mockAssetLookup.On(
+	h.MockAssetLookup.On(
 		"FetchAssetMetaForAsset", mock.Anything, mock.Anything,
 	).Return(dummyMetaReveal, nil).Maybe()
 }