diff options
| author | Dave Henderson <dhenderson@gmail.com> | 2022-01-09 17:12:13 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-01-09 22:12:13 +0000 |
| commit | 155803de7d5c1df5edd456a45a19008c9ae5ea7b (patch) | |
| tree | 7bdc7501e3450ea2a2fafa2521ff0d6a165a1a0e /data/datasource_env_test.go | |
| parent | d03538dcbc6f7d22e02c44269b63ff1d54020ed4 (diff) | |
Propagate context into datasource read functions (#1282)
Signed-off-by: Dave Henderson <dhenderson@gmail.com>
Diffstat (limited to 'data/datasource_env_test.go')
| -rw-r--r-- | data/datasource_env_test.go | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/data/datasource_env_test.go b/data/datasource_env_test.go index e368397a..c7723b3b 100644 --- a/data/datasource_env_test.go +++ b/data/datasource_env_test.go @@ -1,6 +1,7 @@ package data import ( + "context" "net/url" "os" "testing" @@ -14,6 +15,8 @@ func mustParseURL(in string) *url.URL { } func TestReadEnv(t *testing.T) { + ctx := context.Background() + content := []byte(`hello world`) os.Setenv("HELLO_WORLD", "hello world") defer os.Unsetenv("HELLO_WORLD") @@ -22,31 +25,31 @@ func TestReadEnv(t *testing.T) { source := &Source{Alias: "foo", URL: mustParseURL("env:HELLO_WORLD")} - actual, err := readEnv(source) + actual, err := readEnv(ctx, source) assert.NoError(t, err) assert.Equal(t, content, actual) source = &Source{Alias: "foo", URL: mustParseURL("env:/HELLO_WORLD")} - actual, err = readEnv(source) + actual, err = readEnv(ctx, source) assert.NoError(t, err) assert.Equal(t, content, actual) source = &Source{Alias: "foo", URL: mustParseURL("env:///HELLO_WORLD")} - actual, err = readEnv(source) + actual, err = readEnv(ctx, source) assert.NoError(t, err) assert.Equal(t, content, actual) source = &Source{Alias: "foo", URL: mustParseURL("env:HELLO_WORLD?foo=bar")} - actual, err = readEnv(source) + actual, err = readEnv(ctx, source) assert.NoError(t, err) assert.Equal(t, content, actual) source = &Source{Alias: "foo", URL: mustParseURL("env:///HELLO_WORLD?foo=bar")} - actual, err = readEnv(source) + actual, err = readEnv(ctx, source) assert.NoError(t, err) assert.Equal(t, content, actual) } |