From 8f40d15c5605ef83a6c73a33af90b468c0c4df7a Mon Sep 17 00:00:00 2001
From: Gustavo Bazan <gssbzn@gmail.com>
Date: Sun, 15 Dec 2019 19:48:06 +0000
Subject: [PATCH] Update viper_test.go

---
 viper_test.go | 54 +++++++++++++--------------------------------------
 1 file changed, 14 insertions(+), 40 deletions(-)

diff --git a/viper_test.go b/viper_test.go
index 429203d..4d35923 100644
--- a/viper_test.go
+++ b/viper_test.go
@@ -1261,46 +1261,6 @@ var hclWriteExpected = []byte(`"foos" = {
 
 "type" = "donut"`)
 
-func TestWriteConfigHCL(t *testing.T) {
-	v := New()
-	fs := afero.NewMemMapFs()
-	v.SetFs(fs)
-	v.SetConfigName("c")
-	v.SetConfigType("hcl")
-	err := v.ReadConfig(bytes.NewBuffer(hclExample))
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := v.WriteConfigAs("c.hcl"); err != nil {
-		t.Fatal(err)
-	}
-	read, err := afero.ReadFile(fs, "c.hcl")
-	if err != nil {
-		t.Fatal(err)
-	}
-	assert.Equal(t, hclWriteExpected, read)
-}
-
-func TestWriteConfigHCLWithoutFileExtension(t *testing.T) {
-	v := New()
-	fs := afero.NewMemMapFs()
-	v.SetFs(fs)
-	v.SetConfigName("c")
-	v.SetConfigType("hcl")
-	err := v.ReadConfig(bytes.NewBuffer(hclExample))
-	if err != nil {
-		t.Fatal(err)
-	}
-	if err := v.WriteConfigAs("c"); err != nil {
-		t.Fatal(err)
-	}
-	read, err := afero.ReadFile(fs, "c")
-	if err != nil {
-		t.Fatal(err)
-	}
-	assert.Equal(t, hclWriteExpected, read)
-}
-
 var jsonWriteExpected = []byte(`{
   "batters": {
     "batter": [
@@ -1356,6 +1316,20 @@ func TestWriteConfig(t *testing.T) {
 		input           []byte
 		expectedContent []byte
 	}{
+		"hcl with file extension": {
+			configName:      "c",
+			configType:      "hcl",
+			fileName:        "c.hcl",
+			input:           hclExample,
+			expectedContent: hclWriteExpected,
+		},
+		"hcl without file extension": {
+			configName:      "c",
+			configType:      "hcl",
+			fileName:        "c",
+			input:           hclExample,
+			expectedContent: hclWriteExpected,
+		},
 		"json with file extension": {
 			configName:      "c",
 			configType:      "json",