Fix String() for some option types

and add unit tests for them.

Signed-off-by: Daniel Nephin <dnephin@docker.com>
This commit is contained in:
Daniel Nephin 2016-06-15 13:11:23 -04:00
parent 6253c29494
commit c26e7d8c91
2 changed files with 143 additions and 5 deletions

View file

@ -28,7 +28,7 @@ type int64Value interface {
type memBytes int64
func (m *memBytes) String() string {
return strconv.FormatInt(m.Value(), 10)
return units.BytesSize(float64(m.Value()))
}
func (m *memBytes) Set(value string) error {
@ -48,7 +48,7 @@ func (m *memBytes) Value() int64 {
type nanoCPUs int64
func (c *nanoCPUs) String() string {
return strconv.FormatInt(c.Value(), 10)
return big.NewRat(c.Value(), 1e9).FloatString(3)
}
func (c *nanoCPUs) Set(value string) error {
@ -191,14 +191,14 @@ func (m *MountOpt) Set(value string) error {
case "writable":
mount.Writable, err = strconv.ParseBool(value)
if err != nil {
return fmt.Errorf("invalid value for writable: %s", err.Error())
return fmt.Errorf("invalid value for writable: %s", value)
}
case "bind-propagation":
mount.BindOptions.Propagation = swarm.MountPropagation(strings.ToUpper(value))
case "volume-populate":
volumeOptions().Populate, err = strconv.ParseBool(value)
if err != nil {
return fmt.Errorf("invalid value for populate: %s", err.Error())
return fmt.Errorf("invalid value for populate: %s", value)
}
case "volume-label":
setValueOnMap(volumeOptions().Labels, value)
@ -235,7 +235,8 @@ func (m *MountOpt) Type() string {
func (m *MountOpt) String() string {
mounts := []string{}
for _, mount := range m.values {
mounts = append(mounts, fmt.Sprintf("%v", mount))
repr := fmt.Sprintf("%s %s %s", mount.Type, mount.Source, mount.Target)
mounts = append(mounts, repr)
}
return strings.Join(mounts, ", ")
}

View file

@ -0,0 +1,137 @@
package service
import (
"strings"
"testing"
"time"
"github.com/docker/engine-api/types/swarm"
)
func assertEqual(t *testing.T, actual, expected interface{}) {
if expected != actual {
t.Fatalf("Expected '%v' (%T) got '%v' (%T)", expected, expected, actual, actual)
}
}
func assertNilError(t *testing.T, err error) {
if err != nil {
t.Fatalf("Expected no error, got: %s", err.Error())
}
}
func assertError(t *testing.T, err error, contains string) {
if err == nil {
t.Fatalf("Expected an error, but error was nil")
}
if !strings.Contains(err.Error(), contains) {
t.Fatalf("Expected error to contain '%s', got '%s'", contains, err.Error())
}
}
func TestMemBytesString(t *testing.T) {
var mem memBytes = 1048576
assertEqual(t, mem.String(), "1 MiB")
}
func TestMemBytesSetAndValue(t *testing.T) {
var mem memBytes
assertNilError(t, mem.Set("5kb"))
assertEqual(t, mem.Value(), int64(5120))
}
func TestNanoCPUsString(t *testing.T) {
var cpus nanoCPUs = 6100000000
assertEqual(t, cpus.String(), "6.100")
}
func TestNanoCPUsSetAndValue(t *testing.T) {
var cpus nanoCPUs
assertNilError(t, cpus.Set("0.35"))
assertEqual(t, cpus.Value(), int64(350000000))
}
func TestDurationOptString(t *testing.T) {
dur := time.Duration(300 * 10e8)
duration := DurationOpt{value: &dur}
assertEqual(t, duration.String(), "5m0s")
}
func TestDurationOptSetAndValue(t *testing.T) {
var duration DurationOpt
assertNilError(t, duration.Set("300s"))
assertEqual(t, *duration.Value(), time.Duration(300*10e8))
}
func TestUint64OptString(t *testing.T) {
value := uint64(2345678)
opt := Uint64Opt{value: &value}
assertEqual(t, opt.String(), "2345678")
opt = Uint64Opt{}
assertEqual(t, opt.String(), "none")
}
func TestUint64OptSetAndValue(t *testing.T) {
var opt Uint64Opt
assertNilError(t, opt.Set("14445"))
assertEqual(t, *opt.Value(), uint64(14445))
}
func TestMountOptString(t *testing.T) {
mount := MountOpt{
values: []swarm.Mount{
{
Type: swarm.MountType("BIND"),
Source: "/home/path",
Target: "/target",
},
{
Type: swarm.MountType("VOLUME"),
Source: "foo",
Target: "/target/foo",
},
},
}
expected := "BIND /home/path /target, VOLUME foo /target/foo"
assertEqual(t, mount.String(), expected)
}
func TestMountOptSetNoError(t *testing.T) {
var mount MountOpt
assertNilError(t, mount.Set("type=bind,target=/target,source=/foo"))
mounts := mount.Value()
assertEqual(t, len(mounts), 1)
assertEqual(t, mounts[0], swarm.Mount{
Type: swarm.MountType("BIND"),
Source: "/foo",
Target: "/target",
})
}
func TestMountOptSetErrorNoType(t *testing.T) {
var mount MountOpt
assertError(t, mount.Set("target=/target,source=/foo"), "type is required")
}
func TestMountOptSetErrorNoTarget(t *testing.T) {
var mount MountOpt
assertError(t, mount.Set("type=VOLUME,source=/foo"), "target is required")
}
func TestMountOptSetErrorInvalidKey(t *testing.T) {
var mount MountOpt
assertError(t, mount.Set("type=VOLUME,bogus=foo"), "unexpected key 'bogus'")
}
func TestMountOptSetErrorInvalidField(t *testing.T) {
var mount MountOpt
assertError(t, mount.Set("type=VOLUME,bogus"), "invalid field 'bogus'")
}
func TestMountOptSetErrorInvalidWritable(t *testing.T) {
var mount MountOpt
assertError(t, mount.Set("type=VOLUME,writable=yes"), "invalid value for writable: yes")
}