Add SetRawConfig method (#5)

ory · Jul 17, 2019 · 8d638da · 8d638da
1 parent 3c5a66b
commit 8d638da
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/viper.go b/viper.go
@@ -1400,6 +1400,14 @@ func (v *Viper) ReadInConfig() error {
 	return nil
 }
 
+// SetRawConfig overwrites the raw config.
+func SetRawConfig(config map[string]interface{}) { v.SetRawConfig(config) }
+func (v *Viper) SetRawConfig(config map[string]interface{}) {
+	v.lock.Lock()
+	defer v.lock.Unlock()
+	v.config = config
+}
+
 // MergeInConfig merges a new configuration with an existing config.
 func MergeInConfig() error { return v.MergeInConfig() }
 func (v *Viper) MergeInConfig() error {

diff --git a/viper_test.go b/viper_test.go
@@ -1836,6 +1836,12 @@ bar:
 	wg.Wait()
 }
 
+func TestSetRawConfig(t *testing.T) {
+	Reset()
+	SetRawConfig(map[string]interface{}{"foo": "bar"})
+	assert.Equal(t, "bar", Get("foo"))
+}
+
 func BenchmarkGetBool(b *testing.B) {
 	key := "BenchmarkGetBool"
 	v = New()