Add TestGenDiffWithFixturesYaml

Author: alexvitayu

src/gendiff.go

@@ -8,6 +8,8 @@ import (
 	"path/filepath"
 	"sort"
 	"strings"
+
+	"github.com/stretchr/testify/assert/yaml"
 )
 
 func GenDiff(path1, path2, format string) (string, error) {
@@ -35,19 +37,28 @@ func toAbsolutePath(path1, path2 string) (p1, p2 string) {
 }
 
 func parseDataFromFiles(path1, path2 string) (data1, data2 map[string]interface{}, err error) {
+	dataFile1, _ := os.ReadFile(path1)
+	dataFile2, _ := os.ReadFile(path2)
+	var data11 map[string]interface{}
+	var data22 map[string]interface{}
 	switch {
 	case filepath.Ext(path1) == ".json" && filepath.Ext(path2) == ".json":
-		var data1 map[string]interface{}
-		var data2 map[string]interface{}
-		dataFile1, _ := os.ReadFile(path1)
-		dataFile2, _ := os.ReadFile(path2)
-		if err := json.Unmarshal(dataFile1, &data1); err != nil {
+
+		if err := json.Unmarshal(dataFile1, &data11); err != nil {
 			return nil, nil, errors.New("не удалось преобразовать из json")
 		}
-		if err := json.Unmarshal(dataFile2, &data2); err != nil {
+		if err := json.Unmarshal(dataFile2, &data22); err != nil {
 			return nil, nil, errors.New("не удалось преобразовать из json")
 		}
-		return data1, data2, nil
+		return data11, data22, nil
+	case filepath.Ext(path1) == ".yml" && filepath.Ext(path2) == ".yml":
+		if err := yaml.Unmarshal(dataFile1, &data11); err != nil {
+			return nil, nil, errors.New("не удалось преобразовать из yaml")
+		}
+		if err := yaml.Unmarshal(dataFile2, &data22); err != nil {
+			return nil, nil, errors.New("не удалось преобразовать из yaml")
+		}
+		return data11, data22, nil
 	default:
 		return nil, nil, errors.New("неизвестный формат данных")
 	}

src/gendiff_test.go

@@ -11,7 +11,7 @@ import (
 )
 
 // мой тест с использованием фикстур
-func TestGenDiffWithFixtures(t *testing.T) {
+func TestGenDiffWithFixturesJson(t *testing.T) {
 	path1 := filepath.Join("../testdata/fixture/file1.json") //фикстура №1
 	path2 := filepath.Join("../testdata/fixture/file2.json") //фикстура №2
 	path3 := filepath.Join("../testdata/fixture/result.txt") //фикстура №3
@@ -24,6 +24,19 @@ func TestGenDiffWithFixtures(t *testing.T) {
 	assert.Equal(t, string(bytes), str) // утверждаем, что ожидаемая строка равна полученной строке
 }
 
+func TestGenDiffWithFixturesYaml(t *testing.T) {
+	path1 := filepath.Join("../testdata/fixture/file1.yml")  //фикстура №1
+	path2 := filepath.Join("../testdata/fixture/file2.yml")  //фикстура №2
+	path3 := filepath.Join("../testdata/fixture/result.txt") //фикстура №3
+
+	str, err := GenDiff(path1, path2, "stylish")
+	require.NoError(t, err) // если произошла ошибка в основном коде, то и нет смысла двигаться дальше
+
+	bytes, err := os.ReadFile(path3)
+	require.NoError(t, err)             // если произошла ошибка чтения, то нет смысла идти дальше
+	assert.Equal(t, string(bytes), str) // утверждаем, что ожидаемая строка равна полученной строке
+}
+
 // мой тест до использовния фикстур
 func TestGenDiff(t *testing.T) {
 	var testCases = []struct {

testdata/fixture/file1.yml

@@ -0,0 +1,4 @@
+host: hexlet.io
+timeout: 50
+proxy: 123.234.53.22
+follow: false

testdata/fixture/file2.yml

@@ -0,0 +1,3 @@
+timeout: 20
+verbose: true
+host: hexlet.io