WIP download test
This commit is contained in:
Shorssaud
parent
8130b0b898
commit
82160778d8
|
|
@ -0,0 +1,127 @@
|
|||
package codex
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/guseggert/clustertest/cluster"
|
||||
"github.com/guseggert/clustertest/cluster/basic"
|
||||
"github.com/guseggert/clustertest/cluster/docker"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"golang.org/x/sync/errgroup"
|
||||
)
|
||||
|
||||
func TestDownload1(t *testing.T) {
|
||||
run := func(t *testing.T, name string, impl cluster.Cluster) {
|
||||
t.Run(name, func(t *testing.T) {
|
||||
t.Parallel()
|
||||
|
||||
// Create the cluster.
|
||||
c := basic.New(impl)
|
||||
t.Cleanup(c.MustCleanup)
|
||||
|
||||
t.Logf("Launching %s nodes", name)
|
||||
nodes := c.MustNewNodes(2)
|
||||
|
||||
group, groupCtx := errgroup.WithContext(context.Background())
|
||||
addrs := "127.0.0.1"
|
||||
for i, node := range nodes {
|
||||
id := i
|
||||
node := node.Context(groupCtx)
|
||||
group.Go(func() error {
|
||||
// gets node ip
|
||||
ip, err := getIp(groupCtx, node)
|
||||
if err != nil {
|
||||
t.Error(Fatal("failed to get ip: %s", err))
|
||||
t.Error(Fatal("HOST EOutput: %s\n", ip))
|
||||
}
|
||||
t.Log(Info("ip: " + ip))
|
||||
|
||||
stdout := &bytes.Buffer{}
|
||||
stderr := &bytes.Buffer{}
|
||||
|
||||
_, err = node.StartProc(cluster.StartProcRequest{
|
||||
Command: "./build/codex",
|
||||
Args: []string{"--metrics", "--api-port=8090", "--data-dir=`pwd`/Codex1", "--disc-port=8070", "--log-level=TRACE"},
|
||||
Stdout: stdout,
|
||||
})
|
||||
if err != nil {
|
||||
t.Error(Fatal("HOST EOutput: %s\n", stderr.String()))
|
||||
return err
|
||||
}
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
if id == 1 {
|
||||
node.SendFile("tests/dog1.txt", bytes.NewBuffer([]byte("hello my dog")))
|
||||
for x := 0; x < 1; x++ {
|
||||
group.Go(func() error {
|
||||
runout := &bytes.Buffer{}
|
||||
runerr := &bytes.Buffer{}
|
||||
runout.Reset()
|
||||
runerr.Reset()
|
||||
proc, err := node.StartProc(cluster.StartProcRequest{
|
||||
Command: "curl",
|
||||
Args: []string{"-vvv", "-H", "content-type: application/octet-stream", "-H", "Expect:", "-T", "tests/dog1.txt", "http://" + addrs + ":8090/api/codex/v1/upload", "-X", "POST"},
|
||||
Stdout: runout,
|
||||
Stderr: runerr,
|
||||
})
|
||||
if err != nil {
|
||||
t.Error(Fatal("HOST EOutput: ", runout.String()))
|
||||
return err
|
||||
}
|
||||
_, err = proc.Wait()
|
||||
if err != nil {
|
||||
t.Error(Fatal("HOST EOutput: ", runout.String()))
|
||||
return err
|
||||
}
|
||||
t.Log(Info(runout.String()))
|
||||
t.Log(runerr.String())
|
||||
cid := runout.String()
|
||||
runout.Reset()
|
||||
runerr.Reset()
|
||||
proc, err = node.StartProc(cluster.StartProcRequest{
|
||||
Command: "curl",
|
||||
Args: []string{"-vvv", "http://" + addrs + ":8090/api/codex/v1/download<" + cid + ">", "--output", "tests/dog2.txt"},
|
||||
Stdout: runout,
|
||||
})
|
||||
_, _ = proc.Wait()
|
||||
t.Log(Info(runout.String()))
|
||||
t.Log(runerr.String())
|
||||
runout.Reset()
|
||||
runerr.Reset()
|
||||
_, err = node.Run(cluster.StartProcRequest{
|
||||
Command: "ls",
|
||||
Args: []string{"tests"},
|
||||
Stdout: runout,
|
||||
Stderr: runerr,
|
||||
})
|
||||
t.Log(Info(runout.String()))
|
||||
t.Log(runerr.String())
|
||||
runout.Reset()
|
||||
runerr.Reset()
|
||||
_, err = node.Run(cluster.StartProcRequest{
|
||||
Command: "cat",
|
||||
Args: []string{"tests/dog2.txt"},
|
||||
Stdout: runout,
|
||||
Stderr: runerr,
|
||||
})
|
||||
t.Log(Info(runout.String()))
|
||||
t.Log(runerr.String())
|
||||
assert.Equal(t, "hello my dog", runout.String())
|
||||
|
||||
return nil
|
||||
})
|
||||
}
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
group.Wait()
|
||||
})
|
||||
|
||||
}
|
||||
// run(t, "local cluster", local.MustNewCluster())
|
||||
run(t, "Docker cluster", docker.MustNewCluster().WithBaseImage("corbo12/nim-codex:v4"))
|
||||
}
|
||||
|
|
@ -58,18 +58,21 @@ func TestUpload1(t *testing.T) {
|
|||
for x := 0; x < 1; x++ {
|
||||
group.Go(func() error {
|
||||
runout := &bytes.Buffer{}
|
||||
proc, err := node.Run(cluster.StartProcRequest{
|
||||
runerr := &bytes.Buffer{}
|
||||
proc, err := node.StartProc(cluster.StartProcRequest{
|
||||
Command: "curl",
|
||||
Args: []string{"-vvv", "-H", "\"content-type: application/octet-stream\"", "-H", "Expect:", "-T", "tests/dog.txt", "http://" + addrs + ":8090/api/codex/v1/upload", "-X", "POST"},
|
||||
Args: []string{"-vvv", "-H", "content-type: application/octet-stream", "-H", "Expect:", "-T", "tests/dog1.txt", "http://" + addrs + ":8090/api/codex/v1/upload", "-X", "POST"},
|
||||
Stdout: runout,
|
||||
Stderr: runerr,
|
||||
})
|
||||
if err != nil {
|
||||
t.Error(Fatal("HOST EOutput: ", runout.String()))
|
||||
return err
|
||||
}
|
||||
time.Sleep(2 * time.Second)
|
||||
t.Log(Info(runout.String()))
|
||||
assert.Equal(t, 0, proc.ExitCode, "HOST EOutput: %s\n", err)
|
||||
code, _ := proc.Wait()
|
||||
t.Log(Info("CID: ", runout.String()))
|
||||
t.Log(runerr.String())
|
||||
assert.Equal(t, 0, code.ExitCode, "HOST EOutput: %s\n", err)
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue