mirror of
https://github.com/grdl/git-get.git
synced 2026-02-09 06:49:26 +00:00
Add status checking function and basic unit tests
This commit is contained in:
54
status.go
Normal file
54
status.go
Normal file
@@ -0,0 +1,54 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
git "github.com/libgit2/git2go/v30"
|
||||||
|
"github.com/pkg/errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
type RepoStatus int
|
||||||
|
|
||||||
|
const (
|
||||||
|
StatusOk RepoStatus = iota
|
||||||
|
StatusUntrackedFiles
|
||||||
|
StatusUncommittedChanges
|
||||||
|
StatusUnknown
|
||||||
|
)
|
||||||
|
|
||||||
|
func GetStatus(path string) (RepoStatus, error) {
|
||||||
|
repo, err := git.OpenRepository(path)
|
||||||
|
if err != nil {
|
||||||
|
return StatusUnknown, errors.Wrap(err, "failed opening repository")
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = statusEntries(repo)
|
||||||
|
return StatusOk, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func statusEntries(repo *git.Repository) ([]git.StatusEntry, error) {
|
||||||
|
opts := &git.StatusOptions{
|
||||||
|
Show: git.StatusShowIndexAndWorkdir,
|
||||||
|
Flags: git.StatusOptIncludeUntracked,
|
||||||
|
}
|
||||||
|
|
||||||
|
status, err := repo.StatusList(opts)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "failed getting repository status")
|
||||||
|
}
|
||||||
|
|
||||||
|
entryCount, err := status.EntryCount()
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "failed getting repository status count")
|
||||||
|
}
|
||||||
|
|
||||||
|
var entries []git.StatusEntry
|
||||||
|
for i := 0; i < entryCount; i++ {
|
||||||
|
entry, err := status.ByIndex(i)
|
||||||
|
if err != nil {
|
||||||
|
return nil, errors.Wrap(err, "failed getting repository status entry")
|
||||||
|
}
|
||||||
|
|
||||||
|
entries = append(entries, entry)
|
||||||
|
}
|
||||||
|
|
||||||
|
return entries, nil
|
||||||
|
}
|
||||||
109
status_test.go
Normal file
109
status_test.go
Normal file
@@ -0,0 +1,109 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
git "github.com/libgit2/git2go/v30"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestStatusEntriesWithEmptyRepo(t *testing.T) {
|
||||||
|
repo, err := newTestRepo()
|
||||||
|
checkFatal(t, err)
|
||||||
|
defer cleanupRepo(t, repo)
|
||||||
|
|
||||||
|
entries, err := statusEntries(repo)
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
if len(entries) != 0 {
|
||||||
|
t.Errorf("Empty repo should have no status entries")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStatusEntriesWithSingleUnstagedFile(t *testing.T) {
|
||||||
|
repo, err := newTestRepo()
|
||||||
|
checkFatal(t, err)
|
||||||
|
defer cleanupRepo(t, repo)
|
||||||
|
|
||||||
|
err = createFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
entries, err := statusEntries(repo)
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
if len(entries) != 1 {
|
||||||
|
t.Errorf("Repo with single unstaged file should have only one status entry")
|
||||||
|
}
|
||||||
|
|
||||||
|
if entries[0].Status != git.StatusWtNew {
|
||||||
|
t.Errorf("Invalid status, got %d; want %d", entries[0].Status, git.StatusWtNew)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStatusEntriesWithSingleStagedFile(t *testing.T) {
|
||||||
|
repo, err := newTestRepo()
|
||||||
|
checkFatal(t, err)
|
||||||
|
defer cleanupRepo(t, repo)
|
||||||
|
|
||||||
|
err = createFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = stageFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
entries, err := statusEntries(repo)
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
if len(entries) != 1 {
|
||||||
|
t.Errorf("Repo with single staged file should have only one status entry")
|
||||||
|
}
|
||||||
|
|
||||||
|
if entries[0].Status != git.StatusIndexNew {
|
||||||
|
t.Errorf("Invalid status, got %d; want %d", entries[0].Status, git.StatusIndexNew)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStatusEntriesWithSingleCommit(t *testing.T) {
|
||||||
|
repo, err := newTestRepo()
|
||||||
|
checkFatal(t, err)
|
||||||
|
defer cleanupRepo(t, repo)
|
||||||
|
|
||||||
|
err = createFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = stageFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = createCommit(repo, "Initial commit")
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
entries, err := statusEntries(repo)
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
if len(entries) != 0 {
|
||||||
|
t.Errorf("Repo with no uncommitted files should have no status entries")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStatusEntriesWithMultipleCommits(t *testing.T) {
|
||||||
|
repo, err := newTestRepo()
|
||||||
|
checkFatal(t, err)
|
||||||
|
defer cleanupRepo(t, repo)
|
||||||
|
|
||||||
|
err = createFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = stageFile(repo, "SomeFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = createCommit(repo, "Initial commit")
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
err = createFile(repo, "AnotherFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = stageFile(repo, "AnotherFile")
|
||||||
|
checkFatal(t, err)
|
||||||
|
err = createCommit(repo, "Second commit")
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
entries, err := statusEntries(repo)
|
||||||
|
checkFatal(t, err)
|
||||||
|
|
||||||
|
if len(entries) != 0 {
|
||||||
|
t.Errorf("Repo with no uncommitted files should have no status entries")
|
||||||
|
}
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user