Add parsers package

Author: alexvitayu

src/gendiff.go

@@ -1,20 +1,16 @@
 package code
 
 import (
-	"encoding/json"
-	"errors"
+	"code/src/parsers"
 	"fmt"
-	"os"
 	"path/filepath"
 	"sort"
 	"strings"
-
-	"github.com/stretchr/testify/assert/yaml"
 )
 
 func GenDiff(path1, path2, format string) (string, error) {
 	abs1, abs2 := toAbsolutePath(path1, path2)
-	data1, data2, err := parseDataFromFiles(abs1, abs2)
+	data1, data2, err := parsers.ParseDataFromFiles(abs1, abs2)
 	if err != nil {
 		return "", fmt.Errorf("parseDataFromFiles: %w", err)
 	}
@@ -36,34 +32,6 @@ func toAbsolutePath(path1, path2 string) (p1, p2 string) {
 	return abs1, abs2
 }
 
-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":
-
-		if err := json.Unmarshal(dataFile1, &data11); err != nil {
-			return nil, nil, errors.New("не удалось преобразовать из json")
-		}
-		if err := json.Unmarshal(dataFile2, &data22); err != nil {
-			return nil, nil, errors.New("не удалось преобразовать из json")
-		}
-		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("неизвестный формат данных")
-	}
-}
-
 func genDiff(data1, data2 map[string]interface{}) string {
 
 	// Создаём множество ключей из обеих map

src/parsers/parsers.go

@@ -0,0 +1,50 @@
+package parsers
+
+import (
+	"encoding/json"
+	"errors"
+	"gopkg.in/yaml.v3"
+	"os"
+	"path/filepath"
+)
+
+func ParseDataFromFiles(path1, path2 string) (res1, res2 map[string]interface{}, err error) {
+	file1, err := os.ReadFile(path1)
+	if err != nil {
+		return nil, nil, errors.New("не удалось прочитать файл")
+	}
+	file2, err := os.ReadFile(path2)
+	if err != nil {
+		return nil, nil, errors.New("не удалось прочитать файл")
+	}
+	var data1 map[string]interface{}
+	var data2 map[string]interface{}
+	var ext1 = filepath.Ext(path1)
+	var ext2 = filepath.Ext(path2)
+
+	switch {
+	case ext1 == ".json" && ext2 == ".json":
+		if err := json.Unmarshal(file1, &data1); err != nil {
+			return nil, nil, errors.New("не удалось преобразовать из json")
+		}
+		if err := json.Unmarshal(file2, &data2); err != nil {
+			return nil, nil, errors.New("не удалось преобразовать из json")
+		}
+		return data1, data2, nil
+
+	case ext1 == ".yaml" || ext1 == ".yml" && ext2 == ".yaml" || ext2 == ".yml":
+		if err := yaml.Unmarshal(file1, &data1); err != nil {
+			return nil, nil, errors.New("не удалось преобразовать из yaml")
+		}
+		if err := yaml.Unmarshal(file2, &data2); err != nil {
+			return nil, nil, errors.New("не удалось преобразовать из yaml")
+		}
+		return data1, data2, nil
+
+	case ext1 != ext2:
+		return nil, nil, errors.New("разные форматы данных")
+
+	default:
+		return nil, nil, errors.New("неизвестный формат данных")
+	}
+}

src/parsers/parsers_test.go

@@ -0,0 +1,63 @@
+package parsers
+
+import (
+	"errors"
+	"path/filepath"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestParseDataFromFiles(t *testing.T) {
+	var testCases = []struct {
+		name      string
+		filename1 string
+		filename2 string
+		expected  [3]interface{}
+	}{
+		{
+			name:      "BothFilesAreJson",
+			filename1: filepath.Join("../../testdata/fixture/file1.json"),
+			filename2: filepath.Join("../../testdata/fixture/file2.json"),
+			expected: [3]interface{}{map[string]interface{}{"follow": false, "host": "hexlet.io", "proxy": "123.234.53.22", "timeout": float64(50)},
+				map[string]interface{}{"host": "hexlet.io", "timeout": float64(20), "verbose": true}, nil},
+		},
+		{
+			name:      "BothFilesAreYaml",
+			filename1: filepath.Join("../../testdata/fixture/file1.yml"),
+			filename2: filepath.Join("../../testdata/fixture/file2.yml"),
+			expected: [3]interface{}{map[string]interface{}{"follow": false, "host": "hexlet.io", "proxy": "123.234.53.22", "timeout": 50},
+				map[string]interface{}{"host": "hexlet.io", "timeout": 20, "verbose": true}, nil},
+		},
+		{
+			name:      "WrongFormatFiles",
+			filename1: filepath.Join("../../testdata/fixture/file1.pdf"),
+			filename2: filepath.Join("../../testdata/fixture/file2.pdf"),
+			expected:  [3]interface{}{nil, nil, errors.New("неизвестный формат данных")},
+		},
+		{
+			name:      "DifferentFileFormats",
+			filename1: filepath.Join("../../testdata/fixture/file1.json"),
+			filename2: filepath.Join("../../testdata/fixture/file2.pdf"),
+			expected:  [3]interface{}{nil, nil, errors.New("разные форматы данных")},
+		},
+	}
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			got1, got2, err := ParseDataFromFiles(tc.filename1, tc.filename2)
+			if got1 != nil && got2 != nil {
+				assert.Equal(t, got1, tc.expected[0])
+				assert.Equal(t, got2, tc.expected[1])
+			} else {
+				assert.Nil(t, got1)
+				assert.Nil(t, got2)
+			}
+			expectedErr := tc.expected[2]
+			if expectedErr == nil {
+				assert.NoError(t, err)
+			} else {
+				assert.ErrorAs(t, err, &expectedErr)
+			}
+		})
+	}
+}