From 155803de7d5c1df5edd456a45a19008c9ae5ea7b Mon Sep 17 00:00:00 2001 From: Dave Henderson Date: Sun, 9 Jan 2022 17:12:13 -0500 Subject: Propagate context into datasource read functions (#1282) Signed-off-by: Dave Henderson --- data/datasource_env_test.go | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) (limited to 'data/datasource_env_test.go') 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) } -- cgit v1.2.3