parth-gr
63c66da813
few functions got change as they were deprecated
for ex: ioutil.Readfile change to os.Readfile
ioutil.TempFile change to os.CreateTemp
And fixed golang-ci-lint-issues
Signed-off-by: parth-gr <paarora@redhat.com>
(cherry picked from commit a84daf9bf0)
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
/*
|
|
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 (
|
|
"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 := os.ReadFile(binPath)
|
|
assert.NoError(t, err)
|
|
return string(b)
|
|
}
|
|
|
|
// set up the initial test directory tree without bins
|
|
// testRootDir/
|
|
// bin/
|
|
// copy-to-dir/
|
|
testRootDir := t.TempDir()
|
|
// create a test PATH="testRootDir/bin"
|
|
envPath := path.Join(testRootDir, "bin")
|
|
mkdir(envPath)
|
|
t.Setenv("PATH", envPath)
|
|
// 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"))
|
|
}
|