package archive

import (
	"io/ioutil"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestArchive(t *testing.T) {
	var assert = assert.New(t)
	folder, err := ioutil.TempDir("", "archivetest")
	assert.NoError(err)
	empty, err := os.Create(folder + "/empty.txt")
	assert.NoError(err)
	assert.NoError(os.Mkdir(folder+"/folder-inside", 0755))

	for _, archive := range []Archive{
		newArchive(folder, "tar.gz", t),
		newArchive(folder, "zip", t),
	} {
		assert.NoError(archive.Add("empty.txt", empty.Name()))
		assert.NoError(archive.Add("empty.txt", folder+"/folder-inside"))
		assert.Error(archive.Add("dont.txt", empty.Name()+"_nope"))
		assert.NoError(archive.Close())
	}
}

func newArchive(folder, format string, t *testing.T) Archive {
	file, err := os.Create(folder + "/folder." + format)
	assert.NoError(t, err)
	return New(file)
}