git-lfs/commands/smudge_test.go
2015-03-19 13:30:55 -06:00

130 lines
4.7 KiB
Go

package commands
import (
"bytes"
"github.com/bmizerany/assert"
"io/ioutil"
"os"
"path/filepath"
"testing"
)
func TestSmudge(t *testing.T) {
repo := NewRepository(t, "empty")
defer repo.Test()
prePushHookFile := filepath.Join(repo.Path, ".git", "hooks", "pre-push")
progressFile := filepath.Join(repo.Path, ".git", "progress")
// simple smudge example
cmd := repo.Command("smudge", "somefile")
cmd.Input = bytes.NewBufferString("version https://git-lfs.github.com/spec/v1\noid sha256:SOMEOID\nsize 9\n")
cmd.Output = "whatever"
cmd.Env = append(cmd.Env, "GIT_LFS_PROGRESS="+progressFile)
cmd.Before(func() {
path := filepath.Join(repo.Path, ".git", "lfs", "objects", "SO", "ME")
file := filepath.Join(path, "SOMEOID")
assert.Equal(t, nil, os.MkdirAll(path, 0755))
assert.Equal(t, nil, ioutil.WriteFile(file, []byte("whatever\n"), 0755))
})
cmd.After(func() {
// assert hook is created
stat, err := os.Stat(prePushHookFile)
assert.Equal(t, nil, err)
assert.Equal(t, false, stat.IsDir())
// assert progress file
progress, err := ioutil.ReadFile(progressFile)
assert.Equal(t, nil, err)
progLines := bytes.Split(progress, []byte("\n"))
assert.Equal(t, 2, len(progLines))
assert.Equal(t, "smudge 1/1 9/9 somefile", string(progLines[0]))
assert.Equal(t, "", string(progLines[1]))
})
// smudge with custom hook
cmd = repo.Command("smudge")
cmd.Input = bytes.NewBufferString("# git-media\n4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393")
cmd.Output = "whatever"
customHook := []byte("echo 'yo'")
cmd.Before(func() {
path := filepath.Join(repo.Path, ".git", "lfs", "objects", "4d", "7a")
file := filepath.Join(path, "4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393")
assert.Equal(t, nil, os.MkdirAll(path, 0755))
assert.Equal(t, nil, ioutil.WriteFile(file, []byte("whatever\n"), 0755))
assert.Equal(t, nil, ioutil.WriteFile(prePushHookFile, customHook, 0755))
})
cmd.After(func() {
// assert custom hook is not overwritten
by, err := ioutil.ReadFile(prePushHookFile)
assert.Equal(t, nil, err)
assert.Equal(t, string(customHook), string(by))
})
}
func TestSmudgeInfo(t *testing.T) {
repo := NewRepository(t, "empty")
defer repo.Test()
mediaPath := filepath.Join(repo.Path, ".git", "lfs", "objects", "4d", "7a")
mediaFile := filepath.Join(mediaPath, "4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393")
// smudge --info with old pointer format, without local file
cmd := repo.Command("smudge", "--info")
cmd.Input = bytes.NewBufferString("# git-media\n4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393")
cmd.Output = "0 --"
// smudge --info with old pointer format, with local file
cmd = repo.Command("smudge", "--info")
cmd.Input = bytes.NewBufferString("# git-media\n4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393")
cmd.Output = "9 " + mediaFile
cmd.Before(func() {
assert.Equal(t, nil, os.MkdirAll(mediaPath, 0755))
assert.Equal(t, nil, ioutil.WriteFile(mediaFile, []byte("whatever\n"), 0755))
})
// smudge --info with ini pointer format, without local file
cmd = repo.Command("smudge", "--info")
cmd.Input = bytes.NewBufferString("version http://git-media.io/v/2\noid sha256:4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393\nsize 123\n")
cmd.Output = "123 --"
// smudge --info with ini pointer format, with local file
cmd = repo.Command("smudge", "--info")
cmd.Input = bytes.NewBufferString("version http://git-media.io/v/2\noid sha256:4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393\nsize 123\n")
cmd.Output = "9 " + mediaFile
cmd.Before(func() {
assert.Equal(t, nil, os.MkdirAll(mediaPath, 0755))
assert.Equal(t, nil, ioutil.WriteFile(mediaFile, []byte("whatever\n"), 0755))
})
// smudge --info with Git LFS pointer format, without local file
cmd = repo.Command("smudge", "--info")
cmd.Input = bytes.NewBufferString("version https://git-lfs.github.com/spec/v1\noid sha256:4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393\nsize 123\n")
cmd.Output = "123 --"
// smudge --info with Git LFS pointer format, with local file
cmd = repo.Command("smudge", "--info")
cmd.Input = bytes.NewBufferString("version https://git-lfs.github.com/spec/v1\noid sha256:4d7a214614ab2935c943f9e0ff69d22eadbb8f32b1258daaa5e2ca24d17e2393\nsize 123\n")
cmd.Output = "9 " + mediaFile
cmd.Before(func() {
assert.Equal(t, nil, os.MkdirAll(mediaPath, 0755))
assert.Equal(t, nil, ioutil.WriteFile(mediaFile, []byte("whatever\n"), 0755))
})
}
func TestSmudgePassesInvalidData(t *testing.T) {
repo := NewRepository(t, "empty")
defer repo.Test()
cmd := repo.Command("smudge", "somefile")
cmd.Input = bytes.NewBufferString("this is not a pointer file\n")
cmd.Output = "this is not a pointer file"
}