Add record tests

Author: DanWlker

internal/record/record.go

@@ -6,8 +6,6 @@ import (
 	"os"
 	"path/filepath"
 
-	"github.com/goccy/go-yaml"
-
 	"github.com/DanWlker/remind/internal/config"
 	"github.com/DanWlker/remind/internal/data"
 	i_error "github.com/DanWlker/remind/internal/error"
@@ -87,29 +85,38 @@ func SetFileContents(items []RecordEntity) (err error) {
 		}
 	}()
 
-	enc := yaml.NewEncoder(f)
-	if err := enc.Encode(items); err != nil {
-		return fmt.Errorf("enc.Encode: %w", err)
+	err = shared.FWriteStructToYaml[RecordEntity](f, items)
+	if err != nil {
+		return fmt.Errorf("shared.FWriteStructToYaml: %w", err)
 	}
 
 	return nil
 }
 
+func FGetRecordIdentityWithIdentifier(items []RecordEntity, identifier string) (RecordEntity, error) {
+	for _, record := range items {
+		if record.Path == identifier {
+			return record, nil
+		}
+	}
+
+	return RecordEntity{}, i_error.RecordDoesNotExistError{
+		ID: identifier,
+	}
+}
+
 func GetRecordEntityWithIdentifier(homeRemovedPath string) (RecordEntity, error) {
 	allRecords, err := GetFileContents()
 	if err != nil {
 		return RecordEntity{}, fmt.Errorf("GetFileContents: %w", err)
 	}
 
-	for _, record := range allRecords {
-		if record.Path == homeRemovedPath {
-			return record, nil
-		}
+	res, err := FGetRecordIdentityWithIdentifier(allRecords, homeRemovedPath)
+	if err != nil {
+		return RecordEntity{}, fmt.Errorf("FGetRecordIdentityWithIdentifier: %w", err)
 	}
 
-	return RecordEntity{}, i_error.RecordDoesNotExistError{
-		ID: homeRemovedPath,
-	}
+	return res, nil
 }
 
 func CreateNewRecord(pathIdentifier string) (RecordEntity, error) {

internal/record/record_test.go

@@ -0,0 +1,38 @@
+package record
+
+import (
+	"errors"
+	"testing"
+
+	i_error "github.com/DanWlker/remind/internal/error"
+)
+
+func TestFGetRecordIdentityWithIdentifier(t *testing.T) {
+	t.Run("Functionality test", func(t *testing.T) {
+		ex := RecordEntity{"up", "something"}
+		items := []RecordEntity{
+			{"never", "gonna"},
+			{"give", "you"},
+			ex,
+		}
+
+		got, err := FGetRecordIdentityWithIdentifier(items, "something")
+		if err != nil {
+			t.Errorf("Expectad no errors, got %v", err)
+		}
+
+		if got != ex {
+			t.Errorf("Wrong record found, expected %v, got %v", ex, got)
+		}
+	})
+
+	t.Run("Test record does not exist", func(t *testing.T) {
+		var items []RecordEntity
+		var ex i_error.RecordDoesNotExistError
+
+		_, err := FGetRecordIdentityWithIdentifier(items, "something")
+		if !errors.As(err, &ex) {
+			t.Errorf("Should return %v, got %v", ex, err)
+		}
+	})
+}