From 420c1a6112b0c7d3093065eaadf04464b2062428 Mon Sep 17 00:00:00 2001
From: Andrei Markin <andrey.markin.ppc@gmail.com>
Date: Sat, 4 Oct 2025 23:26:32 +0400
Subject: [PATCH] tests: add tests for piece service

---
 services/piece_test.go | 40 ++++++++++++++++++++++++++++++++++++++++
 services/user_test.go  |  2 +-
 2 files changed, 41 insertions(+), 1 deletion(-)
 create mode 100644 services/piece_test.go

diff --git a/services/piece_test.go b/services/piece_test.go
new file mode 100644
index 0000000..0af6321
--- /dev/null
+++ b/services/piece_test.go
@@ -0,0 +1,40 @@
+package services
+
+import (
+	"testing"
+
+	"github.com/musgit-dev/musgit/models"
+)
+
+func TestAddPiece(t *testing.T) {
+	piece, err := pieceService.Add(
+		"piece",
+		"test_composer",
+		models.PieceComplexityEasy,
+	)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if id := piece.ID; id != 2 {
+		t.Fatalf("Unknown piece id, should be 2, got %d", id)
+	}
+
+	newPiece, err := pieceService.Get(2)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	if name := newPiece.Name; name != "piece" {
+		t.Fatalf("Unknown piece name, expected 'test', got %s", name)
+	}
+}
+
+func TestGetAllPieces(t *testing.T) {
+
+	pieces := pieceService.GetAll()
+
+	if nPieces := len(pieces); nPieces != 2 {
+		t.Fatalf("Unexpected number of pieces, got %d, expected 2", nPieces)
+	}
+}
diff --git a/services/user_test.go b/services/user_test.go
index b5a45aa..a11f0e2 100644
--- a/services/user_test.go
+++ b/services/user_test.go
@@ -6,7 +6,7 @@ import (
 
 var userService *UserService
 
-func TestAdd(t *testing.T) {
+func TestAddUser(t *testing.T) {
 	user, err := userService.Add("test_user")
 	if err != nil {
 		t.Fatal(err)