package git

import (
	"os"
	"testing"
	"time"
)

func Test_Push_ToRemote(t *testing.T) {
	repo := createBareTestRepo(t)
	defer os.RemoveAll(repo.Path())
	repo2 := createTestRepo(t)
	defer os.RemoveAll(repo2.Workdir())

	remote, err := repo2.CreateRemote("test_push", repo.Path())
	checkFatal(t, err)

	index, err := repo2.Index()
	checkFatal(t, err)

	index.AddByPath("README")

	err = index.Write()
	checkFatal(t, err)

	newTreeId, err := index.WriteTree()
	checkFatal(t, err)

	tree, err := repo2.LookupTree(newTreeId)
	checkFatal(t, err)

	sig := &Signature{Name: "Rand Om Hacker", Email: "random@hacker.com", When: time.Now()}
	// this should cause master branch to be created if it does not already exist
	_, err = repo2.CreateCommit("HEAD", sig, sig, "message", tree)
	checkFatal(t, err)

	push, err := remote.NewPush()
	checkFatal(t, err)

	err = push.AddRefspec("refs/heads/master")
	checkFatal(t, err)

	err = push.Finish()
	checkFatal(t, err)

	err = push.StatusForeach(func(ref string, msg string) int {
		return 0
	})
	checkFatal(t, err)

	defer remote.Free()
	defer repo.Free()
}

func TestRemotePush(t *testing.T) {
	repo := createBareTestRepo(t)
	defer os.RemoveAll(repo.Path())
	localRepo := createTestRepo(t)
	defer os.RemoveAll(localRepo.Workdir())

	remote, err := localRepo.CreateRemote("test_push", repo.Path())
	checkFatal(t, err)

	seedTestRepo(t, localRepo)

	err = remote.Push([]string{"refs/heads/master"}, nil, nil, "")
	checkFatal(t, err)

	_, err = localRepo.LookupReference("refs/remotes/test_push/master")
	checkFatal(t, err)

	_, err = repo.LookupReference("refs/heads/master")
	checkFatal(t, err)
}