1 files changed, 97 insertions, 0 deletions
diff --git a/internal/domain/group_test.go b/internal/domain/group_test.go
new file mode 100644
index 00000000..2e41ef53
--- /dev/null
+++ b/internal/domain/group_test.go
@@ -0,0 +1,97 @@
+package domain
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestGroupDig(t *testing.T) {
+	matchingProject := &project{ID: 1}
+
+	tests := []struct {
+		name                string
+		g                   group
+		path                string
+		expectedProject     *project
+		expectedProjectPath string
+		expectedPath        string
+	}{
+		{
+			name: "empty group",
+			path: "projectb/demo/features.html",
+			g:    group{},
+		},
+		{
+			name: "group with project",
+			path: "projectb/demo/features.html",
+			g: group{
+				projects: projects{"projectb": matchingProject},
+			},
+			expectedProject:     matchingProject,
+			expectedProjectPath: "projectb",
+			expectedPath:        "demo/features.html",
+		},
+		{
+			name: "group with project and no path in URL",
+			path: "projectb",
+			g: group{
+				projects: projects{"projectb": matchingProject},
+			},
+			expectedProject:     matchingProject,
+			expectedProjectPath: "projectb",
+		},
+		{
+			name: "group with subgroup and project",
+			path: "projectb/demo/features.html",
+			g: group{
+				projects: projects{"projectb": matchingProject},
+				subgroups: subgroups{
+					"sub1": &group{
+						projects: projects{"another": &project{}},
+					},
+				},
+			},
+			expectedProject:     matchingProject,
+			expectedProjectPath: "projectb",
+			expectedPath:        "demo/features.html",
+		},
+		{
+			name: "group with project inside a subgroup",
+			path: "sub1/projectb/demo/features.html",
+			g: group{
+				subgroups: subgroups{
+					"sub1": &group{
+						projects: projects{"projectb": matchingProject},
+					},
+				},
+				projects: projects{"another": &project{}},
+			},
+			expectedProject:     matchingProject,
+			expectedProjectPath: "sub1/projectb",
+			expectedPath:        "demo/features.html",
+		},
+		{
+			name: "group with matching subgroup but no project",
+			path: "sub1/projectb/demo/features.html",
+			g: group{
+				subgroups: subgroups{
+					"sub1": &group{
+						projects: projects{"another": &project{}},
+					},
+				},
+			},
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			project, projectPath, urlPath := test.g.digProjectWithSubpath("", strings.Split(test.path, "/"))
+
+			assert.Equal(t, test.expectedProject, project)
+			assert.Equal(t, test.expectedProjectPath, projectPath)
+			assert.Equal(t, test.expectedPath, urlPath)
+		})
+	}
+}