diff options
author | Michael Muré <batolettre@gmail.com> | 2019-05-27 21:14:55 +0200 |
---|---|---|
committer | Michael Muré <batolettre@gmail.com> | 2019-05-27 21:14:55 +0200 |
commit | c7abac388aadd274d4f23f996a15f8bba90f2a92 (patch) | |
tree | ced1116de500346eb340f72b7bfc672136e40c1e /repository | |
parent | 9865dfcdf0810e288283f2ef4eba450041db32f9 (diff) | |
download | git-bug-c7abac388aadd274d4f23f996a15f8bba90f2a92.tar.gz |
repo: refactor how test repo are created/cleaned
Diffstat (limited to 'repository')
-rw-r--r-- | repository/git_testing.go | 80 |
1 files changed, 80 insertions, 0 deletions
diff --git a/repository/git_testing.go b/repository/git_testing.go new file mode 100644 index 00000000..dd597a8e --- /dev/null +++ b/repository/git_testing.go @@ -0,0 +1,80 @@ +package repository + +import ( + "io/ioutil" + "log" + "os" + "testing" +) + +// This is intended for testing only + +func CreateTestRepo(bare bool) *GitRepo { + dir, err := ioutil.TempDir("", "") + if err != nil { + log.Fatal(err) + } + + // fmt.Println("Creating repo:", dir) + + var creator func(string) (*GitRepo, error) + + if bare { + creator = InitBareGitRepo + } else { + creator = InitGitRepo + } + + repo, err := creator(dir) + if err != nil { + log.Fatal(err) + } + + if err := repo.StoreConfig("user.name", "testuser"); err != nil { + log.Fatal("failed to set user.name for test repository: ", err) + } + if err := repo.StoreConfig("user.email", "testuser@example.com"); err != nil { + log.Fatal("failed to set user.email for test repository: ", err) + } + + return repo +} + +func CleanupTestRepos(t testing.TB, repos ...Repo) { + var firstErr error + for _, repo := range repos { + path := repo.GetPath() + // fmt.Println("Cleaning repo:", path) + err := os.RemoveAll(path) + if err != nil { + log.Println(err) + if firstErr == nil { + firstErr = err + } + } + } + + if firstErr != nil { + t.Fatal(firstErr) + } +} + +func SetupReposAndRemote(t testing.TB) (repoA, repoB, remote *GitRepo) { + repoA = CreateTestRepo(false) + repoB = CreateTestRepo(false) + remote = CreateTestRepo(true) + + remoteAddr := "file://" + remote.GetPath() + + err := repoA.AddRemote("origin", remoteAddr) + if err != nil { + t.Fatal(err) + } + + err = repoB.AddRemote("origin", remoteAddr) + if err != nil { + t.Fatal(err) + } + + return repoA, repoB, remote +} |