fix: allow other dirs than vendor (#80)

This commit is contained in:
Hang Xie 2020-02-08 05:46:51 -08:00 committed by GitHub
parent 0ba0ff5522
commit 1d729c9517
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 36 additions and 6 deletions

View file

@ -48,6 +48,18 @@ const want = `
`
func TestRewrite(t *testing.T) {
testRewriteWithJsonnetHome(t, "vendor")
}
func TestRewriteCustomJsonnetHome(t *testing.T) {
testRewriteWithJsonnetHome(t, "custom-vendor-dir")
}
func TestRewriteDeepCustomJsonnetHome(t *testing.T) {
testRewriteWithJsonnetHome(t, "custom/vendor/dir")
}
func testRewriteWithJsonnetHome(t *testing.T, jsonnetHome string) {
dir, err := ioutil.TempDir("", "jbrewrite")
require.Nil(t, err)
defer os.RemoveAll(dir)
@ -56,11 +68,11 @@ func TestRewrite(t *testing.T) {
err = ioutil.WriteFile(name, []byte(sample), 0644)
require.Nil(t, err)
vendorDir := filepath.Join(dir, "vendor")
vendorDir := filepath.Join(dir, jsonnetHome)
err = os.MkdirAll(vendorDir, os.ModePerm)
require.Nil(t, err)
err = Rewrite(dir, "vendor", locks)
err = Rewrite(dir, jsonnetHome, locks)
require.Nil(t, err)
content, err := ioutil.ReadFile(name)