mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package docker
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/vrischmann/envconfig"
|
|
"gitlab.oracledx.com/odx/functions/api/runner/common"
|
|
"gitlab.oracledx.com/odx/functions/api/runner/drivers"
|
|
)
|
|
|
|
type taskDockerTest struct {
|
|
id string
|
|
input io.Reader
|
|
output io.Writer
|
|
}
|
|
|
|
func (f *taskDockerTest) Command() string { return "" }
|
|
func (f *taskDockerTest) EnvVars() map[string]string {
|
|
return map[string]string{}
|
|
}
|
|
func (f *taskDockerTest) Labels() map[string]string { return nil }
|
|
func (f *taskDockerTest) Id() string { return f.id }
|
|
func (f *taskDockerTest) Group() string { return "" }
|
|
func (f *taskDockerTest) Image() string { return "funcy/hello" }
|
|
func (f *taskDockerTest) Timeout() time.Duration { return 30 * time.Second }
|
|
func (f *taskDockerTest) Logger() (stdout, stderr io.Writer) { return f.output, nil }
|
|
func (f *taskDockerTest) WriteStat(drivers.Stat) { /* TODO */ }
|
|
func (f *taskDockerTest) Volumes() [][2]string { return [][2]string{} }
|
|
func (f *taskDockerTest) WorkDir() string { return "" }
|
|
func (f *taskDockerTest) Close() {}
|
|
func (f *taskDockerTest) Input() io.Reader { return f.input }
|
|
|
|
func TestRunnerDocker(t *testing.T) {
|
|
env := common.NewEnvironment(func(e *common.Environment) {})
|
|
dkr := NewDocker(env, drivers.Config{})
|
|
ctx := context.Background()
|
|
|
|
task := &taskDockerTest{"test-docker", nil, nil}
|
|
|
|
cookie, err := dkr.Prepare(ctx, task)
|
|
if err != nil {
|
|
t.Fatal("Couldn't prepare task test")
|
|
}
|
|
defer cookie.Close()
|
|
|
|
result, err := cookie.Run(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if result.Status() != "success" {
|
|
t.Fatal("Test should successfully run the image")
|
|
}
|
|
}
|
|
|
|
func TestRunnerDockerStdin(t *testing.T) {
|
|
env := common.NewEnvironment(func(e *common.Environment) {})
|
|
dkr := NewDocker(env, drivers.Config{})
|
|
ctx := context.Background()
|
|
|
|
input := `{"name": "test"}`
|
|
var output bytes.Buffer
|
|
|
|
task := &taskDockerTest{"test-docker-stdin", bytes.NewBufferString(input), &output}
|
|
|
|
cookie, err := dkr.Prepare(ctx, task)
|
|
if err != nil {
|
|
t.Fatal("Couldn't prepare task test")
|
|
}
|
|
defer cookie.Close()
|
|
|
|
result, err := cookie.Run(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if result.Status() != "success" {
|
|
t.Error("Test should successfully run the image")
|
|
}
|
|
|
|
expect := "Hello test!"
|
|
got := output.String()
|
|
if !strings.Contains(got, expect) {
|
|
t.Errorf("Test expected output to contain '%s', got '%s'", expect, got)
|
|
}
|
|
}
|
|
|
|
func TestConfigLoadMemory(t *testing.T) {
|
|
if err := os.Setenv("MEMORY_PER_JOB", "128M"); err != nil {
|
|
t.Fatalf("Could not set MEMORY_PER_JOB: %v", err)
|
|
}
|
|
|
|
var conf drivers.Config
|
|
if err := envconfig.Init(&conf); err != nil {
|
|
t.Fatalf("Could not read config: %v", err)
|
|
}
|
|
|
|
if conf.Memory != 128*1024*1024 {
|
|
t.Fatalf("Memory read from config should match 128M, got %d", conf.Memory)
|
|
}
|
|
}
|