Fix AssetManager startup retry

internal/assets/manager.go

@@ -1,4 +1,4 @@
-// Copyright © 2023 Kaleido, Inc.
+// Copyright © 2024 Kaleido, Inc.
 //
 // SPDX-License-Identifier: Apache-2.0
 //
@@ -73,7 +73,7 @@ type Manager interface {
 	RunOperation(ctx context.Context, op *core.PreparedOperation) (outputs fftypes.JSONObject, phase core.OpPhase, err error)
 
 	// Starts the namespace on each of the configured token plugins
-	Start(ctx context.Context) error
+	Start() error
 }
 
 type assetManager struct {
@@ -173,9 +173,9 @@ func (am *assetManager) GetTokenConnectors(ctx context.Context) []*core.TokenCon
 	return connectors
 }
 
-func (am *assetManager) Start(ctx context.Context) error {
-	f := database.TokenPoolQueryFactory.NewFilter(ctx).And()
-	pools, _, err := am.database.GetTokenPools(ctx, am.namespace, f)
+func (am *assetManager) Start() error {
+	f := database.TokenPoolQueryFactory.NewFilter(am.ctx).And()
+	pools, _, err := am.database.GetTokenPools(am.ctx, am.namespace, f)
 	if err != nil {
 		return err
 	}
@@ -187,7 +187,7 @@ func (am *assetManager) Start(ctx context.Context) error {
 				activePools = append(activePools, pool)
 			}
 		}
-		err := plugin.StartNamespace(ctx, am.namespace, activePools)
+		err := plugin.StartNamespace(am.ctx, am.namespace, activePools)
 		if err != nil {
 			return err
 		}

internal/assets/manager_test.go

@@ -188,7 +188,7 @@ func TestStart(t *testing.T) {
 	txHelper, _ := txcommon.NewTransactionHelper(context.Background(), "ns1", mdi, mdm, cmi)
 	am, err := NewAssetManager(context.Background(), "ns1", "blockchain_plugin", mdi, map[string]tokens.Plugin{"magic-tokens": mti}, mim, msa, mbm, mpm, mm, mom, mcm, txHelper, cmi)
 	assert.NoError(t, err)
-	err = am.Start(context.Background())
+	err = am.Start()
 	assert.NoError(t, err)
 }
 
@@ -211,7 +211,7 @@ func TestStartDBError(t *testing.T) {
 	txHelper, _ := txcommon.NewTransactionHelper(context.Background(), "ns1", mdi, mdm, cmi)
 	am, err := NewAssetManager(context.Background(), "ns1", "blockchain_plugin", mdi, map[string]tokens.Plugin{"magic-tokens": mti}, mim, msa, mbm, mpm, mm, mom, mcm, txHelper, cmi)
 	assert.NoError(t, err)
-	err = am.Start(context.Background())
+	err = am.Start()
 	assert.Regexp(t, "pop", err)
 }
 
@@ -241,6 +241,6 @@ func TestStartError(t *testing.T) {
 	txHelper, _ := txcommon.NewTransactionHelper(context.Background(), "ns1", mdi, mdm, cmi)
 	am, err := NewAssetManager(context.Background(), "ns1", "blockchain_plugin", mdi, map[string]tokens.Plugin{"magic-tokens": mti}, mim, msa, mbm, mpm, mm, mom, mcm, txHelper, cmi)
 	assert.NoError(t, err)
-	err = am.Start(context.Background())
+	err = am.Start()
 	assert.Regexp(t, "pop", err)
 }

internal/orchestrator/orchestrator.go

@@ -297,6 +297,9 @@ func (or *orchestrator) Start() (err error) {
 	if err == nil {
 		or.txWriter.Start()
 	}
+	if err == nil {
+		err = or.assets.Start()
+	}
 
 	or.started = true
 	return err
@@ -544,9 +547,6 @@ func (or *orchestrator) initManagers(ctx context.Context) (err error) {
 		if err != nil {
 			return err
 		}
-		if err := or.assets.Start(ctx); err != nil {
-			return err
-		}
 	}
 
 	if or.defsender == nil {

internal/orchestrator/orchestrator_test.go

@@ -305,18 +305,6 @@ func TestInitMessagingComponentFail(t *testing.T) {
 	assert.Regexp(t, "FF10128", err)
 }
 
-func TestInitTokensFail(t *testing.T) {
-	or := newTestOrchestrator()
-	defer or.cleanup(t)
-	or.assets = nil
-	or.mom.On("RegisterHandler", mock.Anything, mock.Anything, mock.Anything)
-	or.mmp.On("ConfigureContract", mock.Anything).Return(nil)
-	or.mdi.On("GetTokenPools", mock.Anything, "ns", mock.Anything).Return([]*core.TokenPool{}, nil, nil)
-	or.mti.On("StartNamespace", mock.Anything, "ns", mock.Anything).Return(fmt.Errorf("pop"))
-	err := or.initManagers(context.Background())
-	assert.Regexp(t, "pop", err)
-}
-
 func TestInitEventsComponentFail(t *testing.T) {
 	or := newTestOrchestrator()
 	defer or.cleanup(t)
@@ -496,6 +484,7 @@ func TestStartStopOk(t *testing.T) {
 	or.msd.On("Start").Return(nil)
 	or.mom.On("Start").Return(nil)
 	or.mtw.On("Start").Return()
+	or.mam.On("Start").Return(nil)
 	or.mba.On("WaitStop").Return(nil)
 	or.mbm.On("WaitStop").Return(nil)
 	or.mdm.On("WaitStop").Return(nil)
@@ -518,6 +507,7 @@ func TestStartStopOk(t *testing.T) {
 	or.msd.On("Start").Return(nil)
 	or.mom.On("Start").Return(nil)
 	or.mtw.On("Start").Return()
+	or.mam.On("Start").Return(nil)
 	or.mba.On("WaitStop").Return(nil)
 	or.mbm.On("WaitStop").Return(nil)
 	or.mdm.On("WaitStop").Return(nil)