Commit 9ba5f586 authored by Angus Salkeld's avatar Angus Salkeld

Increase coverage in pkg/util/env to 100%

parent e7b92b9d
...@@ -50,6 +50,12 @@ func TestGetEnvAsIntOrFallback(t *testing.T) { ...@@ -50,6 +50,12 @@ func TestGetEnvAsIntOrFallback(t *testing.T) {
key = "FLOCKER_UNSET_VAR" key = "FLOCKER_UNSET_VAR"
returnVal, _ = GetEnvAsIntOrFallback(key, expected) returnVal, _ = GetEnvAsIntOrFallback(key, expected)
assert.Equal(expected, returnVal) assert.Equal(expected, returnVal)
key = "FLOCKER_SET_VAR"
os.Setenv(key, "not-an-int")
returnVal, err := GetEnvAsIntOrFallback(key, 1)
assert.Equal(expected, returnVal)
assert.EqualError(err, "strconv.ParseInt: parsing \"not-an-int\": invalid syntax")
} }
func TestGetEnvAsFloat64OrFallback(t *testing.T) { func TestGetEnvAsFloat64OrFallback(t *testing.T) {
...@@ -65,4 +71,10 @@ func TestGetEnvAsFloat64OrFallback(t *testing.T) { ...@@ -65,4 +71,10 @@ func TestGetEnvAsFloat64OrFallback(t *testing.T) {
key = "FLOCKER_UNSET_VAR" key = "FLOCKER_UNSET_VAR"
returnVal, _ = GetEnvAsFloat64OrFallback(key, 1.0) returnVal, _ = GetEnvAsFloat64OrFallback(key, 1.0)
assert.Equal(expected, returnVal) assert.Equal(expected, returnVal)
key = "FLOCKER_SET_VAR"
os.Setenv(key, "not-a-float")
returnVal, err := GetEnvAsFloat64OrFallback(key, 1.0)
assert.Equal(expected, returnVal)
assert.EqualError(err, "strconv.ParseFloat: parsing \"not-a-float\": invalid syntax")
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment