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.
54 lines
1.6 KiB
Go
54 lines
1.6 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"testing"
|
|
|
|
"crypto-miner-server/internal/erasure"
|
|
)
|
|
|
|
type okStore struct{}
|
|
|
|
func (okStore) PutShard(context.Context, string, string, []byte) error { return nil }
|
|
func (okStore) HeadBucket(context.Context, string) error { return nil }
|
|
|
|
func TestErasureSwarmGetPolicyJSON(t *testing.T) {
|
|
h := NewErasureSwarmHandler(func() erasure.AWSSwarmSettings {
|
|
return erasure.AWSSwarmSettings{S3Bucket: "lab-shards", CloudFrontDomain: "d.cf.net"}
|
|
}, nil)
|
|
req := httptest.NewRequest(http.MethodGet, "/api/v1/erasure-swarm/policy?bucket=lab-shards", nil)
|
|
rec := httptest.NewRecorder()
|
|
h.GetPolicyJSON(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("status=%d body=%s", rec.Code, rec.Body.String())
|
|
}
|
|
var body map[string]interface{}
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if body["iam_policy"] == nil || body["bucket_policy"] == nil {
|
|
t.Fatalf("body=%v", body)
|
|
}
|
|
}
|
|
|
|
func TestErasureSwarmPostTestOK(t *testing.T) {
|
|
os.Setenv("AF_AWS_ACCESS_KEY_ID", "A")
|
|
os.Setenv("AF_AWS_SECRET_ACCESS_KEY", "s")
|
|
defer os.Unsetenv("AF_AWS_ACCESS_KEY_ID")
|
|
defer os.Unsetenv("AF_AWS_SECRET_ACCESS_KEY")
|
|
h := NewErasureSwarmHandler(func() erasure.AWSSwarmSettings {
|
|
return erasure.AWSSwarmSettings{S3Bucket: "b", CloudFrontDomain: "d.cf.net"}
|
|
}, func() erasure.ShardObjectStore { return okStore{} })
|
|
rec := httptest.NewRecorder()
|
|
h.PostTest(rec, httptest.NewRequest(http.MethodPost, "/", nil))
|
|
var body erasureSwarmTestResponse
|
|
_ = json.NewDecoder(rec.Body).Decode(&body)
|
|
if !body.OK {
|
|
t.Fatalf("%+v", body)
|
|
}
|
|
}
|