bcd25b53a6
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Unskip cloud-map export assertions, correct cloud venue class fixtures, and update tests/README.md to 960 server / 670 agent Test* totals.
31 lines
822 B
Go
31 lines
822 B
Go
package api
|
|
|
|
import (
|
|
"testing"
|
|
|
|
fleetai "crypto-miner-server/internal/ai"
|
|
"crypto-miner-server/internal/db"
|
|
)
|
|
|
|
func TestCloudVenueIngest(t *testing.T) {
|
|
dbi, e := db.New(t.TempDir())
|
|
if e != nil {
|
|
t.Fatal(e)
|
|
}
|
|
t.Cleanup(func() { _ = dbi.Close() })
|
|
h := NewWSHub(dbi)
|
|
h.ingestCloudVenueReport("a", fleetai.CloudVenueReport{InstanceType: "g4dn.xlarge", OrganizationalUnit: "ou/gpu"})
|
|
if len(h.cloudVenueSnapshot()) != 1 {
|
|
t.Fatal("expected venue snapshot")
|
|
}
|
|
}
|
|
|
|
func TestCloudVenueClassBatchSpot(t *testing.T) {
|
|
if fleetai.InferCloudVenueClass(fleetai.CloudVenueReport{Workload: "batch"}) != fleetai.CloudVenueBatch {
|
|
t.Fatal("expected batch class")
|
|
}
|
|
if fleetai.InferCloudVenueClass(fleetai.CloudVenueReport{InstanceLifecycle: "spot"}) != fleetai.CloudVenueSpot {
|
|
t.Fatal("expected spot class")
|
|
}
|
|
}
|