worker/registration: fix waiting for deregister event

.changelog/4662.bugfix.md

@@ -0,0 +1 @@
+Fix waiting for deregister event on `RequestShutdown`

go/oasis-node/cmd/common/genesis/staking.go

@@ -110,6 +110,9 @@ func NewAppendableStakingState() (*AppendableStakingState, error) {
 	st := &AppendableStakingState{
 		State: staking.Genesis{
 			Ledger: make(map[staking.Address]*staking.Account),
+			Parameters: staking.ConsensusParameters{
+				DebondingInterval: 1, // Minimum valid debonding interval.
+			},
 		},
 	}
 	if err := st.setDefaultFeeSplit(); err != nil {

go/oasis-test-runner/scenario/e2e/runtime/keymanager_upgrade.go

@@ -5,6 +5,7 @@ import (
 	"context"
 	"fmt"
 	"path/filepath"
+	"time"
 
 	"github.com/oasisprotocol/oasis-core/go/common"
 	"github.com/oasisprotocol/oasis-core/go/common/cbor"
@@ -283,13 +284,32 @@ func (sc *kmUpgradeImpl) Run(childEnv *env.Env) error {
 		return err
 	}
 
+	nodeCh, nodeSub, err := sc.Net.Controller().Registry.WatchNodes(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to watch nodes: %w", err)
+	}
+	defer nodeSub.Close()
+
 	// Shutdown old keymanager and make sure it deregisters.
 	sc.Logger.Info("shutting down old keymanager")
 	oldKm := sc.Net.Keymanagers()[0]
 	if err := oldKm.RequestShutdown(ctx, true); err != nil {
 		return fmt.Errorf("failed to request shutdown: %w", err)
 	}
 
+	// Ensure keymanager deregisters.
+OUTER:
+	for {
+		select {
+		case ev := <-nodeCh:
+			if !ev.IsRegistration && ev.Node.ID.Equal(oldKm.NodeID) {
+				break OUTER
+			}
+		case <-time.After(10 * time.Second):
+			return fmt.Errorf("failed to wait for keymanager to de-register")
+		}
+	}
+
 	// Run client again.
 	sc.Logger.Info("starting a second client to check if key manager works")
 	newTestClient := sc.testClient.Clone().(*KeyValueEncTestClient)

go/worker/registration/worker.go

@@ -652,9 +652,6 @@ func (w *Worker) doNodeRegistration() {
 
 		case <-w.stopCh:
 			return
-
-		case <-w.stopRegCh:
-			return
 		}
 	}
 }