1 files changed, 104 insertions, 0 deletions
diff --git a/HutchTests/RepositorySettingsViewModelTests.swift b/HutchTests/RepositorySettingsViewModelTests.swift
new file mode 100644
index 0000000..a2c88ce
--- /dev/null
+++ b/HutchTests/RepositorySettingsViewModelTests.swift
@@ -0,0 +1,104 @@
+import Foundation
+import Testing
+@testable import Hutch
+
+private struct UpdateRepositoryInfoEnvelope: Decodable {
+    let updateRepository: UpdateRepositoryInfoPayload
+}
+
+private struct UpdateRepositoryInfoPayload: Decodable {
+    let id: Int
+}
+
+struct RepositorySettingsViewModelTests {
+
+    @Test
+    func saveInfoResponseDecodesMinimalRepositoryPayload() throws {
+        let json = """
+        {
+            "data": {
+                "updateRepository": {
+                    "id": 42
+                }
+            }
+        }
+        """
+
+        let decoded = try JSONDecoder().decode(
+            GraphQLResponse<UpdateRepositoryInfoEnvelope>.self,
+            from: Data(json.utf8)
+        )
+
+        #expect(decoded.data?.updateRepository.id == 42)
+    }
+
+    @Test
+    func gitCanonicalEntityAddsMissingTilde() {
+        #expect(RepositorySettingsViewModel.gitCanonicalEntity(from: "alice") == "~alice")
+        #expect(RepositorySettingsViewModel.gitCanonicalEntity(from: "~alice") == "~alice")
+        #expect(RepositorySettingsViewModel.gitCanonicalEntity(from: "  alice  ") == "~alice")
+    }
+
+    @Test
+    func gitHeadReferenceUsesFullBranchRef() {
+        #expect(RepositorySettingsViewModel.gitHeadReference(from: "main") == "refs/heads/main")
+        #expect(RepositorySettingsViewModel.gitHeadReference(from: " refs/heads/dev ") == "refs/heads/dev")
+    }
+
+    @Test
+    @MainActor
+    func unchangedHeadIsOmittedFromSaveInput() {
+        let viewModel = RepositorySettingsViewModel(
+            repository: makeRepository(headName: "refs/heads/main"),
+            branches: [Reference(name: "refs/heads/main", target: nil)],
+            client: SRHTClient(token: "test-token")
+        )
+
+        #expect(viewModel.selectedHeadReferenceForSave() == nil)
+    }
+
+    @Test
+    @MainActor
+    func changedHeadUsesSelectedBranchReference() {
+        let viewModel = RepositorySettingsViewModel(
+            repository: makeRepository(headName: "refs/heads/main"),
+            branches: [
+                Reference(name: "refs/heads/main", target: nil),
+                Reference(name: "refs/heads/dev", target: nil)
+            ],
+            client: SRHTClient(token: "test-token")
+        )
+        viewModel.editedHead = "dev"
+
+        #expect(viewModel.selectedHeadReferenceForSave() == "refs/heads/dev")
+    }
+
+    @Test
+    @MainActor
+    func bareRepositoryOmitsHeadWhenNoBranchExists() {
+        let viewModel = RepositorySettingsViewModel(
+            repository: makeRepository(headName: nil),
+            branches: [],
+            client: SRHTClient(token: "test-token")
+        )
+
+        #expect(viewModel.selectedHeadReferenceForSave() == nil)
+        viewModel.editedHead = "main"
+        #expect(viewModel.selectedHeadReferenceForSave() == nil)
+    }
+
+    @MainActor
+    private func makeRepository(headName: String?) -> RepositorySummary {
+        RepositorySummary(
+            id: 1,
+            rid: "rid-1",
+            service: .git,
+            name: "repo",
+            description: "desc",
+            visibility: .public,
+            updated: .now,
+            owner: Entity(canonicalName: "~owner"),
+            head: headName.map { Reference(name: $0, target: nil) }
+        )
+    }
+}