/*
Copyright 2019 The Rook Authors. All rights reserved.

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

	http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/

package util

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

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

func TestCopyBinary(t *testing.T) {
	createTestBinary := func(binPath string) {
		f, err := os.Create(binPath)
		assert.NoError(t, err)
		// the binary should just be the text showing where it was copied from so we can verify it
		_, err = f.WriteString(binPath)
		assert.NoError(t, err)
		err = os.Chmod(binPath, 0700)
		assert.NoError(t, err)
		f.Close()
	}

	mkdir := func(dir string) {
		err := os.MkdirAll(dir, 0700)
		assert.NoError(t, err)
	}

	cleanDir := func(dir string) {
		err := os.RemoveAll(dir)
		assert.NoError(t, err)
		mkdir(dir)
	}

	fileText := func(binPath string) string {
		b, err := ioutil.ReadFile(binPath)
		assert.NoError(t, err)
		return string(b)
	}

	// set up the initial test directory tree without bins
	// testRootDir/
	//   bin/
	//   copy-to-dir/
	testRootDir, err := ioutil.TempDir("", "rook-cmd-reporter-copy-binaries-test")
	assert.NoError(t, err)
	defer os.RemoveAll(testRootDir)
	// create a test PATH="testRootDir/bin"
	envPath := path.Join(testRootDir, "bin")
	mkdir(envPath)
	oPath := os.Getenv("PATH")
	defer os.Setenv("PATH", oPath)
	err = os.Setenv("PATH", envPath)
	assert.NoError(t, err)
	// create initial copy-to-dir
	copyToDir := path.Join(testRootDir, "copy-to-dir")
	mkdir(copyToDir)
	// set default rook dir for unit tests
	oDefaultRookDir := defaultRookDir
	defaultRookDir = path.Join(testRootDir, "/usr/local/bin")
	defer func() { defaultRookDir = oDefaultRookDir }()

	// expect success if binary is available in path
	// testRootDir/
	//   bin/
	//     rook
	//   copy-to-dir/
	createTestBinary(path.Join(envPath, "rook"))
	cleanDir(copyToDir)

	err = CopyBinaries(copyToDir)
	assert.NoError(t, err)
	r := fileText(path.Join(copyToDir, "rook"))
	assert.Contains(t, r, path.Join(testRootDir, "bin/rook"))

	// expect success if the binary is available in default locations AND in path
	// additionally expect that the binary will be taken from the default location in this case
	// testRootDir/
	//   usr/local/bin/
	//     rook
	//   bin/
	//     rook
	//   copy-to-dir/
	mkdir(defaultRookDir)
	createTestBinary(path.Join(defaultRookDir, "rook"))
	cleanDir(copyToDir)

	err = CopyBinaries(copyToDir)
	assert.NoError(t, err)
	r = fileText(path.Join(copyToDir, "rook"))
	assert.Contains(t, r, path.Join(testRootDir, "usr/local/bin/rook"))
}