add more test utilities

metio · Sep 9, 2022 · c94c346 · c94c346
1 parent 92bfe5a
commit c94c346
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/internal/testutils/files.go b/internal/testutils/files.go
@@ -22,7 +22,11 @@ func TemporaryDirectory(t *testing.T) string {
 }
 
 func WriteFile(t *testing.T, name string) {
-	err := os.WriteFile(name, []byte("hello world!"), 0644)
+	WriteFileContent(t, name, "hello world!")
+}
+
+func WriteFileContent(t *testing.T, name string, content string) {
+	err := os.WriteFile(name, []byte(content), 0644)
 	if err != nil {
 		t.Fatal(err)
 	}

diff --git a/internal/testutils/repository.go b/internal/testutils/repository.go
@@ -43,10 +43,14 @@ func GetRepositoryHead(t *testing.T, repository *git.Repository) *plumbing.Refer
 }
 
 func WriteFileInWorktree(t *testing.T, worktree *git.Worktree, name string) {
-	filename := filepath.Join(worktree.Filesystem.Root(), name)
+	filename := FileInWorktree(worktree, name)
 	WriteFile(t, filename)
 }
 
+func FileInWorktree(worktree *git.Worktree, name string) string {
+	return filepath.Join(worktree.Filesystem.Root(), name)
+}
+
 func AddAndCommitNewFile(t *testing.T, worktree *git.Worktree, name string) {
 	WriteFileInWorktree(t, worktree, name)
 	GitAdd(t, worktree, name)