executor_test.go

// Copyright 2017 GRAIL, Inc. All rights reserved.
// Use of this source code is governed by the Apache 2.0
// license that can be found in the LICENSE file.

package reflow_test

import (
	"testing"
	"time"

	"docker.io/go-docker/api/types"
	"github.com/grailbio/reflow"
)

func TestResources(t *testing.T) {
	r1 := reflow.Resources{"mem": 10, "cpu": 5, "disk": 1}
	r2 := reflow.Resources{"mem": 5, "cpu": 2, "disk": 3}
	var got, want reflow.Resources
	got.Sub(r1, r2)
	if want := (reflow.Resources{"mem": 5, "cpu": 3, "disk": -2}); !got.Equal(want) {
		t.Errorf("got %v, want %v", got, want)
	}
	got.Add(r1, r2)
	if want := (reflow.Resources{"mem": 15, "cpu": 7, "disk": 4}); !got.Equal(want) {
		t.Errorf("got %v, want %v", got, want)
	}
	got.Add(r1, r2)
	want.Add(r2, r1)
	if !got.Equal(want) {
		t.Errorf("got %v, want %v", got, want)
	}
	if r1.Available(r2) {
		t.Errorf("expected %v to be unavailable in %v", r2, r1)
	}
	r3 := reflow.Resources{"mem": 3, "cpu": 1, "disk": 1}
	if !r1.Available(r3) {
		t.Errorf("expected %v to be available in %v", r3, r1)
	}
	got.Min(r1, r2)
	if want := (reflow.Resources{"mem": 5, "cpu": 2, "disk": 1}); !got.Equal(want) {
		t.Errorf("got %v, want %v", got, want)
	}
	got.Max(r1, r2)
	if want := (reflow.Resources{"mem": 10, "cpu": 5, "disk": 3}); !got.Equal(want) {
		t.Errorf("got %v, want %v", got, want)
	}
}

func TestRequirements(t *testing.T) {
	var (
		req  reflow.Requirements
		res1 = reflow.Resources{"mem": 10, "cpu": 5, "disk": 1}
		res2 = reflow.Resources{"mem": 20, "cpu": 3, "disk": 1}
	)
	req.AddSerial(res1)
	req.AddSerial(res2)
	if got, want := req.Min, (reflow.Resources{"mem": 20, "cpu": 5, "disk": 1}); !got.Equal(want) {
		t.Errorf("got %v, want %v", got, want)
	}
}

func TestRuntime(t *testing.T) {
	var (
		e = new(reflow.ExecInspect)
		b = new(types.ContainerJSONBase)
		d = new(types.ContainerJSON)
		s = new(types.ContainerState)
	)
	b.State = s
	d.ContainerJSONBase = b
	e.Docker = *d

	for _, tt := range []struct {
		sd, ed string
		want   time.Duration
	}{
		// End is around 0.01s before start. In this case, the runtime should be 0s.
		{"2019-05-30T22:09:34.945074271Z", "2019-05-30T22:09:34.910391085Z", time.Duration(0)},
		// End is 10 seconds after start, rounded to the nearest second.
		{"2019-05-30T22:09:34.945074271Z", "2019-05-30T22:09:44.910391085Z", 10 * time.Second},
		// End is 10 seconds after start, rounded to the nearest second. End is also missing a digit (8 decimals instead of 9).
		{"2019-05-30T22:09:34.945074271Z", "2019-05-30T22:09:44.91039105Z", 10 * time.Second},
	} {
		s.StartedAt, s.FinishedAt = tt.sd, tt.ed
		if got, want := e.Runtime().Round(time.Second), tt.want; got != want {
			t.Errorf("got %v, want %v", got, want)
		}
	}
}