package commands

import (
	"bytes"
	"github.com/bmizerany/assert"
	"io/ioutil"
	"os"
	"path/filepath"
	"testing"
)

func TestClean(t *testing.T) {
	repo := NewRepository(t, "empty")
	defer repo.Test()

	content := "HI\n"
	oid := "f712374589a4f37f0fd6b941a104c7ccf43f68b1fdecb4d5cd88b80acbf98fc2"
	prePushHookFile := filepath.Join(repo.Path, ".git", "hooks", "pre-push")

	cmd := repo.Command("clean")
	cmd.Input = bytes.NewBufferString(content)
	cmd.Output = `[git-media]
version=http://git-media.io/v/2
oid=sha256:` + oid + `
size=3`

	cmd.After(func() {
		// assert file gets queued
		queueDir := filepath.Join(repo.Path, ".git", "media", "queue", "upload")
		files, err := ioutil.ReadDir(queueDir)
		assert.Equal(t, nil, err)
		assert.Equal(t, 1, len(files))

		by, err := ioutil.ReadFile(filepath.Join(queueDir, files[0].Name()))
		assert.Equal(t, nil, err)
		assert.Equal(t, oid, string(by))

		// assert hooks
		stat, err := os.Stat(prePushHookFile)
		assert.Equal(t, nil, err)
		assert.Equal(t, false, stat.IsDir())
	})

	cmd = repo.Command("clean")
	cmd.Input = bytes.NewBufferString(content)
	cmd.Output = `[git-media]
version=http://git-media.io/v/2
oid=sha256:` + oid + `
size=3`
	customHook := []byte("echo 'yo'")
	cmd.Before(func() {
		err := ioutil.WriteFile(prePushHookFile, customHook, 0755)
		assert.Equal(t, nil, err)
	})

	cmd.After(func() {
		by, err := ioutil.ReadFile(prePushHookFile)
		assert.Equal(t, nil, err)
		assert.Equal(t, string(customHook), string(by))
	})
}