container_test.go

package model

import "testing"

func TestStepContainerID(t *testing.T) {
	s := inittests(t)
	container := s.Steps[0].Component.Container
	if container.ID != "16fd2a16-3ac6-11e5-a25d-2fa4b0893ef1" {
		t.Errorf("The step's container ID was '%s' when it should have been '16fd2a16-3ac6-11e5-a25d-2fa4b0893ef1'", container.ID)
	}
}

func TestStepContainerImageID(t *testing.T) {
	s := inittests(t)
	image := s.Steps[0].Component.Container.Image
	if image.ID != "fc210a84-f7cd-4067-939c-a68ec3e3bd2b" {
		t.Errorf("The container image ID was '%s' when it should have been 'fc210a84-f7cd-4067-939c-a68ec3e3bd2b'", image.ID)
	}
}

func TestStepContainerImageURL(t *testing.T) {
	s := inittests(t)
	image := s.Steps[0].Component.Container.Image
	if image.URL != "https://registry.hub.docker.com/u/discoenv/backwards-compat" {
		t.Errorf("The container image URL was '%s' when it should have been 'https://registry.hub.docker.com/u/discoenv/backwards-compat'", image.URL)
	}
}

func TestStepContainerImageTag(t *testing.T) {
	s := inittests(t)
	image := s.Steps[0].Component.Container.Image
	if image.Tag != "latest" {
		t.Errorf("The container image tag was '%s' when it should have been 'latest'", image.Tag)
	}
}

func TestStepContainerImageName(t *testing.T) {
	s := inittests(t)
	image := s.Steps[0].Component.Container.Image
	if image.Name != "gims.iplantcollaborative.org:5000/backwards-compat" {
		t.Errorf("The container image name was '%s' when it should have been 'gims.iplantcollaborative.org:5000/backwards-compat'", image.Name)
	}
}

func TestStepContainerMissingOSGImagePath(t *testing.T) {
	s := inittests(t)
	image := s.Steps[0].Component.Container.Image
	if image.OSGImagePath != "" {
		t.Errorf("The OSG image path of the container was '%s' when it should have been empty", image.OSGImagePath)
	}
}

func TestStepContainerOSGImagePath(t *testing.T) {
	s := inittestsFile(t, "test/test_submission_osg.json")
	image := s.Steps[0].Component.Container.Image
	if image.OSGImagePath != "/path/to/image" {
		t.Errorf("The OSG image path of the container was '%s' when it should have been '/path/to/image'", image.OSGImagePath)
	}
}

func TestStepContainerVolumesFrom1(t *testing.T) {
	s := inittests(t)
	vfs := s.Steps[0].Component.Container.VolumesFrom
	vfsLength := len(vfs)
	if vfsLength != 2 {
		t.Errorf("The number of VolumesFrom entries was '%d' when it should have been '2'", vfsLength)
	}
}

func TestStepContainerVolumesFrom2(t *testing.T) {
	s := inittests(t)
	vfs := s.Steps[0].Component.Container.VolumesFrom[0]
	if vfs.Name != "vf-name1" {
		t.Errorf("The VolumesFrom name was '%s' when it should have been 'vf-name1'", vfs.Name)
	}
	if vfs.NamePrefix != "vf-prefix1" {
		t.Errorf("The VolumesFrom prefix was '%s' when it should have been 'vf-prefix1'", vfs.NamePrefix)
	}
	if vfs.Tag != "vf-tag1" {
		t.Errorf("The VolumesFrom tag was '%s' when it should have been 'vf-tag1'", vfs.Tag)
	}
	if vfs.URL != "vf-url1" {
		t.Errorf("The VolumesFrom url was '%s' when it should have been 'vf-url1'", vfs.URL)
	}
	if vfs.HostPath != "/host/path1" {
		t.Errorf("The VolumesFrom host path was '%s' when it should have been '/host/path1'", vfs.HostPath)
	}
	if vfs.ContainerPath != "/container/path1" {
		t.Errorf("The VolumesFrom container path was '%s' when it should have been '/container/path1'", vfs.ContainerPath)
	}
	if !vfs.ReadOnly {
		t.Error("The VolumesFrom read-only field was false when it should have been true.")
	}
}

func TestStepContainerVolumesFrom3(t *testing.T) {
	s := inittests(t)
	vfs := s.Steps[0].Component.Container.VolumesFrom[1]
	if vfs.Name != "vf-name2" {
		t.Errorf("The VolumesFrom name was '%s' when it should have been 'vf-name2'", vfs.Name)
	}
	if vfs.NamePrefix != "vf-prefix2" {
		t.Errorf("The VolumesFrom prefix was '%s' when it should have been 'vf-prefix2'", vfs.NamePrefix)
	}
	if vfs.Tag != "vf-tag2" {
		t.Errorf("The VolumesFrom tag was '%s' when it should have been 'vf-tag2'", vfs.Tag)
	}
	if vfs.URL != "vf-url2" {
		t.Errorf("The VolumesFrom url was '%s' when it should have been 'vf-url2'", vfs.URL)
	}
	if vfs.HostPath != "/host/path2" {
		t.Errorf("The VolumesFrom host path was '%s' when it should have been '/host/path2'", vfs.HostPath)
	}
	if vfs.ContainerPath != "/container/path2" {
		t.Errorf("The VolumesFrom container path was '%s' when it should have been '/container/path2'", vfs.ContainerPath)
	}
	if !vfs.ReadOnly {
		t.Error("The VolumesFrom read-only field was false when it should have been true.")
	}
}

func TestStepContainerVolumes(t *testing.T) {
	s := inittests(t)
	vols := s.Steps[0].Component.Container.Volumes
	volslen := len(vols)
	if volslen != 2 {
		t.Errorf("The number of volumes was %d when it should have been 2", volslen)
	}
	vol := vols[0]
	if vol.HostPath != "/host/path1" {
		t.Errorf("The volume host path was set to '%s' instead of '/host/path1'", vol.HostPath)
	}
	if vol.ContainerPath != "/container/path1" {
		t.Errorf("The volume container path was set to '%s' instead of '/container/path1'", vol.ContainerPath)
	}
	vol = vols[1]
	if vol.HostPath != "" {
		t.Errorf("The volume host path was set to '%s' instead of an empty string", vol.HostPath)
	}
	if vol.ContainerPath != "/container/path2" {
		t.Errorf("The volume container path was set to '%s' instead of '/container/path2'", vol.ContainerPath)
	}
}

func TestUsesVolumes(t *testing.T) {
	s := inittests(t)
	container := s.Steps[0].Component.Container
	if !container.UsesVolumes() {
		t.Error("The container UsesVolumes method was false when it should have been true.")
	}

	step := s.Steps[0]
	if !step.UsesVolumes() {
		t.Error("The step UsesVolumes method was false when it should have been true.")
	}

	if !s.UsesVolumes() {
		t.Error("The job UsesVolumes method was false when it should have been true.")
	}
}

func TestNotUsesVolumes(t *testing.T) {
	s := inittestsFile(t, "test/no_volumes_submission.json")
	container := s.Steps[0].Component.Container
	if container.UsesVolumes() {
		t.Error("The container UsesVolumes method was false when it should have been true.")
	}

	step := s.Steps[0]
	if step.UsesVolumes() {
		t.Error("The step UsesVolumes method was false when it should have been true.")
	}

	if s.UsesVolumes() {
		t.Error("The job UsesVolumes method was false when it should have been true.")
	}
}

func TestStepContainerDevices(t *testing.T) {
	s := inittests(t)
	devices := s.Steps[0].Component.Container.Devices
	numdevices := len(devices)
	if numdevices != 2 {
		t.Errorf("The number of devices was %d when it should have been 2", numdevices)
	}
	device := devices[0]
	if device.HostPath != "/host/path1" {
		t.Errorf("The volume host path was set to '%s' instead of '/host/path1'", device.HostPath)
	}
	if device.ContainerPath != "/container/path1" {
		t.Errorf("The volume container path was set to '%s' instead of '/container/path1'", device.ContainerPath)
	}
	device = devices[1]
	if device.HostPath != "/host/path2" {
		t.Errorf("The volume host path was set to '%s' instead of '/host/path2'", device.HostPath)
	}
	if device.ContainerPath != "/container/path2" {
		t.Errorf("The volume container path was set to '%s' instead of '/container/path2'", device.ContainerPath)
	}
}

func TestStepContainerWorkingDir(t *testing.T) {
	s := inittests(t)
	w := s.Steps[0].Component.Container.WorkingDir
	if w != "/work" {
		t.Errorf("The working directory for the container was '%s' instead of '/work'", w)
	}
}

func TestStepContainerWorkingDirectory(t *testing.T) {
	s := _inittests(t, false)
	w := s.Steps[0].Component.Container.WorkingDirectory()
	if w != "/work" {
		t.Errorf("The return value of WorkingDirectory() was '%s' instead of '/work'", w)
	}
	s.Steps[0].Component.Container.WorkingDir = ""
	w = s.Steps[0].Component.Container.WorkingDirectory()
	if w != "/de-app-work" {
		t.Errorf("The return value of WorkingDirectory was '%s' instead of '/de-app-work'", w)
	}
}

func TestDevices(t *testing.T) {
	s := inittests(t)
	numdevices := len(s.Steps[0].Component.Container.Devices)
	if numdevices != 2 {
		t.Errorf("The number of devices was '%d' rather than '2'", numdevices)
	}
	d1 := s.Steps[0].Component.Container.Devices[0]
	if d1.HostPath != "/host/path1" {
		t.Errorf("The first device's host path was '%s' instead of '/host/path1'", d1.HostPath)
	}
	if d1.ContainerPath != "/container/path1" {
		t.Errorf("The first device's container path was '%s' instead of '/container/path1'", d1.ContainerPath)
	}
	d2 := s.Steps[0].Component.Container.Devices[1]
	if d2.HostPath != "/host/path2" {
		t.Errorf("The second device's host path was '%s' instead of '/host/path2'", d2.HostPath)
	}
	if d2.ContainerPath != "/container/path2" {
		t.Errorf("The second device's container path was '%s' instead of '/container/path1'", d2.ContainerPath)
	}
}

func TestPorts(t *testing.T) {
	s := inittests(t)

	numports := len(s.Steps[0].Component.Container.Ports)
	if numports != 2 {
		t.Errorf("The number of port mappings was '%d' instead of 2", numports)
	}

	p1 := s.Steps[0].Component.Container.Ports[0]
	if p1.ContainerPort != 1000 {
		t.Errorf("container port was %d instead of 1000", p1.ContainerPort)
	}
	if p1.HostPort != 1001 {
		t.Errorf("host port was %d instead of 1001", p1.HostPort)
	}
	if p1.BindToHost {
		t.Error("bind to host was true")
	}

	p2 := s.Steps[0].Component.Container.Ports[1]
	if p2.ContainerPort != 1002 {
		t.Errorf("container port was %d instead of 1002", p2.ContainerPort)
	}
	if p2.HostPort != 1003 {
		t.Errorf("host port was %d instead of 1003", p2.HostPort)
	}
	if !p2.BindToHost {
		t.Error("bind to host was false")
	}
}