65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
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)
|
|
}
|
|
}
|