mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
910612d0b1
* Docker stats to Prometheus * Fix compilation error in docker_test * Refactor docker driver Run function to wait for the container to have stopped before stopping the colleciton of statistics * Fix go fmt errors * Updates to sending docker stats to Prometheus * remove new test TestWritResultImpl because we changes to support multiple waiters have been removed * Update docker.Run to use channels not contextrs to shut down stats collector
118 lines
3.0 KiB
Go
118 lines
3.0 KiB
Go
package docker
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fnproject/fn/api/agent/drivers"
|
|
"github.com/fsouza/go-dockerclient"
|
|
)
|
|
|
|
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 "fnproject/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(context.Context, drivers.Stat) { /* TODO */ }
|
|
func (f *taskDockerTest) Volumes() [][2]string { return [][2]string{} }
|
|
func (f *taskDockerTest) Memory() uint64 { return 256 * 1024 * 1024 }
|
|
func (f *taskDockerTest) WorkDir() string { return "" }
|
|
func (f *taskDockerTest) Close() {}
|
|
func (f *taskDockerTest) Input() io.Reader { return f.input }
|
|
|
|
func TestRunnerDocker(t *testing.T) {
|
|
dkr := NewDocker(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(ctx)
|
|
|
|
waiter, err := cookie.Run(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
result, err := waiter.Wait(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if result.Status() != "success" {
|
|
t.Fatal("Test should successfully run the image")
|
|
}
|
|
}
|
|
|
|
func TestRunnerDockerStdin(t *testing.T) {
|
|
dkr := NewDocker(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(ctx)
|
|
|
|
waiter, err := cookie.Run(ctx)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
result, err := waiter.Wait(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 TestRegistry(t *testing.T) {
|
|
image := "fnproject/hello"
|
|
|
|
sizer, err := CheckRegistry(context.Background(), image, docker.AuthConfiguration{})
|
|
if err != nil {
|
|
t.Fatal("expected registry check not to fail, got:", err)
|
|
}
|
|
|
|
size, err := sizer.Size()
|
|
if err != nil {
|
|
t.Fatal("expected sizer not to fail, got:", err)
|
|
}
|
|
|
|
if size <= 0 {
|
|
t.Fatal("expected positive size for image that exists, got size:", size)
|
|
}
|
|
}
|