mirror of
https://github.com/fnproject/fn.git
synced 2022-10-28 21:29:17 +03:00
780791da1c
the docker one actually sucks to use, as apparent here. had to add a lot of shit just for it to work, but the heroku registry client has gone stale and is causing issues with our dependencies. didn't expect this to be that hard, but it's over now. good thing we aren't even using this code ^_^. we need to at some point start limiting image sizes and remember this not being fun to figure out to begin with, so anyway, now this works and we don't have to wrestle with different versions of docker in deps (hopefully) anymore.
111 lines
3.0 KiB
Go
111 lines
3.0 KiB
Go
package docker
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/fnproject/fn/api/runner/common"
|
|
"github.com/fnproject/fn/api/runner/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 "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) 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) {
|
|
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(ctx)
|
|
|
|
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(ctx)
|
|
|
|
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 TestRegistry(t *testing.T) {
|
|
image := "funcy/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.Fatalf("expected positive size for image that exists, got size:", size)
|
|
}
|
|
}
|