Initial commit: AetherForge Linux (forge-mesh) v0.1.0-dev

internal/mining/chain_test.go

@@ -0,0 +1,64 @@
+package mining
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"forge-mesh/internal/api/types"
+	"forge-mesh/internal/policy"
+)
+
+func TestChainAdvancesOnTimeout(t *testing.T) {
+	profile := policy.DefaultMiningProfile("test-wallet")
+	profile.Tiers = []types.MiningTierSpec{
+		{Type: "oci", Duration: 0},
+		{Type: "xmrig", Duration: 0},
+	}
+
+	cfg := DefaultChainConfig()
+	cfg.HostID = "test-host"
+	cfg.GateWindow = 200 * time.Millisecond
+	cfg.PollInterval = 50 * time.Millisecond
+	cfg.DefaultProbe = 2 * time.Second
+	cfg.MinHashrateHps = 1500
+
+	chain := NewChain(profile, cfg)
+	ctx, cancel := context.WithTimeout(context.Background(), 15*time.Second)
+	defer cancel()
+
+	go func() { _ = chain.Run(ctx) }()
+
+	deadline := time.Now().Add(12 * time.Second)
+	for time.Now().Before(deadline) {
+		st := chain.Status()
+		if st.CurrentTier >= 2 && st.HashrateHps >= cfg.MinHashrateHps {
+			if st.Wallet != "test-wallet" {
+				t.Fatalf("wallet pin broken: %q", st.Wallet)
+			}
+			chain.Stop()
+			return
+		}
+		time.Sleep(100 * time.Millisecond)
+	}
+	t.Fatal("chain did not advance to tier 2 with hashrate")
+}
+
+func TestWalletPinnedAcrossTiers(t *testing.T) {
+	profile := policy.DefaultMiningProfile("pinned-wallet-abc")
+	cfg := DefaultChainConfig()
+	cfg.HostID = "host-abc"
+	cfg.MinHashrateHps = 1
+
+	chain := NewChain(profile, cfg)
+	for _, spec := range profile.Tiers {
+		runner, err := chain.buildRunner(spec)
+		if err != nil {
+			t.Fatal(err)
+		}
+		_ = runner
+	}
+	if chain.Status().Wallet != "pinned-wallet-abc" {
+		t.Fatalf("expected pinned wallet, got %q", chain.Status().Wallet)
+	}
+}