Skip to content

Commit

Permalink
feat(shm): add e2e tests
Browse files Browse the repository at this point in the history
Signed-off-by: Artur Troian <[email protected]>
  • Loading branch information
troian committed Mar 21, 2024
1 parent 6a96fc9 commit 60335ab
Show file tree
Hide file tree
Showing 9 changed files with 184 additions and 1 deletion.
5 changes: 5 additions & 0 deletions _run/kube/provider.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,3 +11,8 @@ attributes:
value: true
- key: capabilities/storage/2/class
value: beta2
- key: capabilities/storage/3/class
value: ram
- key: capabilities/storage/3/persistent
value: false

1 change: 1 addition & 0 deletions go.mod
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ require (
github.com/gorilla/context v1.1.1
github.com/gorilla/mux v1.8.0
github.com/gorilla/websocket v1.5.0
github.com/gyuho/linux-inspect v0.0.0-20180929231013-a492bfc5f12a
github.com/jaypipes/ghw v0.12.0
github.com/moby/term v0.5.0
github.com/pkg/errors v0.9.1
Expand Down
2 changes: 2 additions & 0 deletions go.sum
Original file line number Diff line number Diff line change
Expand Up @@ -953,6 +953,8 @@ github.com/gtank/merlin v0.1.1 h1:eQ90iG7K9pOhtereWsmyRJ6RAwcP4tHTDBHXNg+u5is=
github.com/gtank/merlin v0.1.1/go.mod h1:T86dnYJhcGOh5BjZFCJWTDeTK7XW8uE+E21Cy/bIQ+s=
github.com/gtank/ristretto255 v0.1.2 h1:JEqUCPA1NvLq5DwYtuzigd7ss8fwbYay9fi4/5uMzcc=
github.com/gtank/ristretto255 v0.1.2/go.mod h1:Ph5OpO6c7xKUGROZfWVLiJf9icMDwUeIvY4OmlYW69o=
github.com/gyuho/linux-inspect v0.0.0-20180929231013-a492bfc5f12a h1:rhA92KsUcMmncbNtK/0qfF8oLhVbohNkPrfPuNO2og0=
github.com/gyuho/linux-inspect v0.0.0-20180929231013-a492bfc5f12a/go.mod h1:u2l1k3IAT/RZKOAwDCcFBlG2CLl9+JE/6hWy6KBLJa0=
github.com/h2non/parth v0.0.0-20190131123155-b4df798d6542/go.mod h1:Ow0tF8D4Kplbc8s8sSb3V2oUCygFHVp8gC3Dn6U4MNI=
github.com/hailocab/go-hostpool v0.0.0-20160125115350-e80d13ce29ed/go.mod h1:tMWxXQ9wFIaZeTI9F+hmhFiGpFmhOHzyShyFUhRm0H4=
github.com/hashicorp/consul-template v0.25.0/go.mod h1:/vUsrJvDuuQHcxEw0zik+YXTS7ZKWZjQeaQhshBmfH0=
Expand Down
1 change: 0 additions & 1 deletion integration/deployment_update_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -257,5 +257,4 @@ func (s *E2EDeploymentUpdate) TestE2ELeaseShell() {
lID, 99, false, false, "notaservice", "/bin/echo", "/foo")
require.Error(s.T(), err)
require.Regexp(s.T(), ".*no such service exists with that name.*", err.Error())

}
1 change: 1 addition & 0 deletions integration/e2e_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -579,6 +579,7 @@ func TestIntegrationTestSuite(t *testing.T) {
suite.Run(t, new(E2EPersistentStorageDefault))
suite.Run(t, new(E2EPersistentStorageBeta2))
suite.Run(t, new(E2EPersistentStorageDeploymentUpdate))
suite.Run(t, new(E2EStorageClassRam))
suite.Run(t, new(E2EMigrateHostname))
suite.Run(t, new(E2EJWTServer))
suite.Run(t, new(E2ECustomCurrency))
Expand Down
129 changes: 129 additions & 0 deletions integration/storageclassram_test.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,129 @@
//go:build e2e

package integration

import (
"context"
"fmt"
"path/filepath"
"time"

"github.com/cosmos/cosmos-sdk/client/flags"
sdktest "github.com/cosmos/cosmos-sdk/testutil"
"github.com/gyuho/linux-inspect/df"

dtypes "github.com/akash-network/akash-api/go/node/deployment/v1beta3"
mtypes "github.com/akash-network/akash-api/go/node/market/v1beta4"
clitestutil "github.com/akash-network/node/testutil/cli"
deploycli "github.com/akash-network/node/x/deployment/client/cli"
mcli "github.com/akash-network/node/x/market/client/cli"

ptestutil "github.com/akash-network/provider/testutil/provider"
)

type E2EStorageClassRam struct {
IntegrationTestSuite
}

func (s *E2EStorageClassRam) TestRAM() {
deploymentPath, err := filepath.Abs("../testdata/deployment/deployment-v2-storage-ram.yaml")
s.Require().NoError(err)

deploymentID := dtypes.DeploymentID{
Owner: s.keyTenant.GetAddress().String(),
DSeq: uint64(100),
}

// Create Deployments
res, err := deploycli.TxCreateDeploymentExec(
s.validator.ClientCtx,
s.keyTenant.GetAddress(),
deploymentPath,
cliGlobalFlags(fmt.Sprintf("--dseq=%v", deploymentID.DSeq))...,
)
s.Require().NoError(err)
s.Require().NoError(s.waitForBlocksCommitted(7))
clitestutil.ValidateTxSuccessful(s.T(), s.validator.ClientCtx, res.Bytes())

bidID := mtypes.MakeBidID(
mtypes.MakeOrderID(dtypes.MakeGroupID(deploymentID, 1), 1),
s.keyProvider.GetAddress(),
)

_, err = mcli.QueryBidExec(s.validator.ClientCtx, bidID)
s.Require().NoError(err)

_, err = mcli.TxCreateLeaseExec(
s.validator.ClientCtx,
bidID,
s.keyTenant.GetAddress(),
cliGlobalFlags()...,
)
s.Require().NoError(err)
s.Require().NoError(s.waitForBlocksCommitted(2))
clitestutil.ValidateTxSuccessful(s.T(), s.validator.ClientCtx, res.Bytes())

lid := bidID.LeaseID()

// Send Manifest to Provider ----------------------------------------------
_, err = ptestutil.TestSendManifest(
s.validator.ClientCtx.WithOutputFormat("json"),
lid.BidID(),
deploymentPath,
fmt.Sprintf("--%s=%s", flags.FlagFrom, s.keyTenant.GetAddress().String()),
fmt.Sprintf("--%s=%s", flags.FlagHome, s.validator.ClientCtx.HomeDir),
)
s.Require().NoError(err)
s.Require().NoError(s.waitForBlocksCommitted(2))

var out sdktest.BufferWriter
leaseShellCtx, cancel := context.WithTimeout(s.ctx, time.Minute)
defer cancel()

extraArgs := []string{
fmt.Sprintf("--%s=%s", flags.FlagFrom, s.keyTenant.GetAddress().String()),
fmt.Sprintf("--%s=%s", flags.FlagHome, s.validator.ClientCtx.HomeDir),
}

logged := make(map[string]struct{})

cmd := `df --all --sync --block-size=1024 --output=source,target,fstype,file,itotal,iavail,iused,ipcent,size,avail,used,pcent`

// Loop until we get a shell or the context times out
for {
select {
case <-leaseShellCtx.Done():
s.T().Fatalf("context is done while trying to run lease-shell: %v", leaseShellCtx.Err())
return
default:
}
out, err = ptestutil.TestLeaseShell(leaseShellCtx, s.validator.ClientCtx.WithOutputFormat("json"), extraArgs, lid, 0, false, false, "web", cmd)
if err != nil {
_, hasBeenLogged := logged[err.Error()]
if !hasBeenLogged {
// Don't spam an error message in a test, that is very annoying
s.T().Logf("encountered %v, waiting before next attempt", err)
logged[err.Error()] = struct{}{}
}
time.Sleep(100 * time.Millisecond)
continue // Try again until the context times out
}
s.Require().NotNil(s.T(), out)
break
}

dfRes, err := df.Parse(out.String())
s.Require().NoError(err)

var found *df.Row

for i := range dfRes {
if dfRes[i].MountedOn == "/dev/shm" {
found = &dfRes[i]
break
}
}

s.Require().NotNil(found)
s.Require().Equal(int64(131072), found.TotalBlocks)
}
4 changes: 4 additions & 0 deletions integration/test_helpers.go
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,10 @@ attributes:
value: true
- key: capabilities/storage/2/class
value: beta2
- key: capabilities/storage/3/persistent
value: false
- key: capabilities/storage/3/class
value: ram
`
)

Expand Down
1 change: 1 addition & 0 deletions script/usd_pricing_oracle.sh
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ STORAGE_USD_SCALE[default]=0.02
STORAGE_USD_SCALE[beta1]=0.02
STORAGE_USD_SCALE[beta2]=0.03
STORAGE_USD_SCALE[beta3]=0.04
STORAGE_USD_SCALE[ram]=0.02 # ram storage class is for tmp disks like /dev/shm, making assumption for now pricing is same of for regular RAM

# used later for validation
MAX_INT64=9223372036854775807
Expand Down
41 changes: 41 additions & 0 deletions testdata/deployment/deployment-v2-storage-ram.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
---
version: "2.0"
services:
web:
image: ghcr.io/akash-network/e2e-test
expose:
- port: 8080
as: 80
to:
- global: true
accept:
- webdistest.localhost
params:
storage:
shm:
mount: /dev/shm
profiles:
compute:
web:
resources:
cpu:
units: "0.01"
memory:
size: "128Mi"
storage:
- size: "512Mi"
- name: shm
size: "256Mi"
attributes:
class: ram
placement:
global:
pricing:
web:
denom: uakt
amount: 10
deployment:
web:
global:
profile: web
count: 1

0 comments on commit 60335ab

Please sign in to comment.