Merge pull request #953 from ahmetalpbalkan/unsetenv

Use os.Unsetenv in options_test
This commit is contained in:
Victor Vieux 2015-07-06 15:32:06 -07:00
commit 8048703206
1 changed files with 20 additions and 11 deletions

View File

@ -11,9 +11,10 @@ import (
var opts = DriverOpts{"foo1=bar", "foo2=-5", "foo3=7", "foo4=0.6", "foo5=127.0.0.1"}
func TestString(t *testing.T) {
os.Setenv("FOO_4", "bar")
// FIXME: Use Unsetenv when switching to go 1.4
defer os.Setenv("FOO_4", "")
if err := os.Setenv("FOO_4", "bar"); err != nil {
t.Fatal(err)
}
defer os.Unsetenv("FOO_4")
val, ok := opts.String("foo1", "")
assert.True(t, ok)
@ -41,8 +42,10 @@ func TestString(t *testing.T) {
}
func TestInt(t *testing.T) {
os.Setenv("FOO_4", "bar")
defer os.Setenv("FOO_4", "")
if err := os.Setenv("FOO_4", "bar"); err != nil {
t.Fatal(err)
}
defer os.Unsetenv("FOO_4")
val, ok := opts.Int("foo1", "")
assert.True(t, ok)
@ -70,8 +73,10 @@ func TestInt(t *testing.T) {
}
func TestUint(t *testing.T) {
os.Setenv("FOO_4", "bar")
defer os.Setenv("FOO_4", "")
if err := os.Setenv("FOO_4", "bar"); err != nil {
t.Fatal(err)
}
defer os.Unsetenv("FOO_4")
val, ok := opts.Uint("foo1", "")
assert.True(t, ok)
@ -99,8 +104,10 @@ func TestUint(t *testing.T) {
}
func TestFloat(t *testing.T) {
os.Setenv("FOO_4", "0.2")
defer os.Setenv("FOO_4", "")
if err := os.Setenv("FOO_4", "0.2"); err != nil {
t.Fatal(err)
}
defer os.Unsetenv("FOO_4")
val, ok := opts.Float("foo1", "")
assert.True(t, ok)
@ -128,8 +135,10 @@ func TestFloat(t *testing.T) {
}
func TestIP(t *testing.T) {
os.Setenv("FOO_4", "0.0.0.0")
defer os.Setenv("FOO_4", "")
if err := os.Setenv("FOO_4", "0.0.0.0"); err != nil {
t.Fatal(err)
}
defer os.Unsetenv("FOO_4")
val, ok := opts.IP("foo5", "")
assert.True(t, ok)