refactor: provider timezone and env (#5)

env/provider.go

@@ -0,0 +1,11 @@
+package env
+
+import "context"
+
+func Provider(env Env) func(ctx context.Context) error {
+	return func(ctx context.Context) error {
+		SetEnv(env)
+
+		return nil
+	}
+}

env/provider_test.go

@@ -0,0 +1,18 @@
+package env
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestProvider(t *testing.T) {
+	assert.True(t, Is(Prod))
+	assert.False(t, Is(Dev))
+
+	assert.NoError(t, Provider(Dev)(context.Background()))
+
+	assert.False(t, Is(Prod))
+	assert.True(t, Is(Dev))
+}

timezone/timezone.go → timezone/provider.go

@@ -17,7 +17,7 @@ func Local(name string) Option {
 	}
 }
 
-func Timezone(opts ...Option) func(ctx context.Context) error {
+func Provider(opts ...Option) func(ctx context.Context) error {
 	op := options{
 		local: "UTC",
 	}

timezone/timezone_test.go → timezone/provider_test.go

@@ -11,13 +11,13 @@ import (
 func TestTimezone(t *testing.T) {
 	// before
 	assert.Equal(t, "Local", time.Local.String())
-	assert.NoError(t, Timezone()(context.Background()))
+	assert.NoError(t, Provider()(context.Background()))
 	assert.Equal(t, "UTC", time.Local.String())
 
 	// after
-	assert.NoError(t, Timezone(Local("Asia/Shanghai"))(context.Background()))
+	assert.NoError(t, Provider(Local("Asia/Shanghai"))(context.Background()))
 	assert.Equal(t, "Asia/Shanghai", time.Local.String())
 
 	// err
-	assert.Error(t, Timezone(Local("Asia/Beijing"))(context.Background()))
+	assert.Error(t, Provider(Local("Asia/Beijing"))(context.Background()))
 }