mirror of https://github.com/pulumi/pulumi.git
479 lines
11 KiB
Go
479 lines
11 KiB
Go
// Copyright 2016-2019, Pulumi Corporation.
|
|
//
|
|
// 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 workspace
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/blang/semver"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestPluginSelection_ExactMatch(t *testing.T) {
|
|
v1 := semver.MustParse("0.1.0")
|
|
v2 := semver.MustParse("0.2.0")
|
|
v3 := semver.MustParse("0.3.0")
|
|
candidatePlugins := []PluginInfo{
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v1,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v2,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "notmyplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: AnalyzerPlugin,
|
|
Version: &v3,
|
|
},
|
|
}
|
|
|
|
requested := semver.MustParseRange("0.2.0")
|
|
result, err := SelectCompatiblePlugin(candidatePlugins, ResourcePlugin, "myplugin", requested)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "myplugin", result.Name)
|
|
assert.Equal(t, "0.2.0", result.Version.String())
|
|
}
|
|
|
|
func TestPluginSelection_ExactMatchNotFound(t *testing.T) {
|
|
v1 := semver.MustParse("0.1.0")
|
|
v2 := semver.MustParse("0.2.1")
|
|
v3 := semver.MustParse("0.3.0")
|
|
candidatePlugins := []PluginInfo{
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v1,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v2,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "notmyplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: AnalyzerPlugin,
|
|
Version: &v3,
|
|
},
|
|
}
|
|
|
|
requested := semver.MustParseRange("0.2.0")
|
|
_, err := SelectCompatiblePlugin(candidatePlugins, ResourcePlugin, "myplugin", requested)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func TestPluginSelection_PatchVersionSlide(t *testing.T) {
|
|
v1 := semver.MustParse("0.1.0")
|
|
v2 := semver.MustParse("0.2.0")
|
|
v21 := semver.MustParse("0.2.1")
|
|
v3 := semver.MustParse("0.3.0")
|
|
candidatePlugins := []PluginInfo{
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v1,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v2,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v21,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "notmyplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: AnalyzerPlugin,
|
|
Version: &v3,
|
|
},
|
|
}
|
|
|
|
requested := semver.MustParseRange(">=0.2.0 <0.3.0")
|
|
result, err := SelectCompatiblePlugin(candidatePlugins, ResourcePlugin, "myplugin", requested)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "myplugin", result.Name)
|
|
assert.Equal(t, "0.2.1", result.Version.String())
|
|
}
|
|
|
|
func TestPluginSelection_EmptyVersionNoAlternatives(t *testing.T) {
|
|
v1 := semver.MustParse("0.1.0")
|
|
v2 := semver.MustParse("0.2.1")
|
|
v3 := semver.MustParse("0.3.0")
|
|
candidatePlugins := []PluginInfo{
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v1,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v2,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: nil,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "notmyplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: AnalyzerPlugin,
|
|
Version: &v3,
|
|
},
|
|
}
|
|
|
|
requested := semver.MustParseRange("0.2.0")
|
|
result, err := SelectCompatiblePlugin(candidatePlugins, ResourcePlugin, "myplugin", requested)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "myplugin", result.Name)
|
|
assert.Nil(t, result.Version)
|
|
}
|
|
|
|
func TestPluginSelection_EmptyVersionWithAlternatives(t *testing.T) {
|
|
v1 := semver.MustParse("0.1.0")
|
|
v2 := semver.MustParse("0.2.0")
|
|
v3 := semver.MustParse("0.3.0")
|
|
candidatePlugins := []PluginInfo{
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v1,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v2,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: nil,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: nil,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "notmyplugin",
|
|
Kind: ResourcePlugin,
|
|
Version: &v3,
|
|
},
|
|
{
|
|
Name: "myplugin",
|
|
Kind: AnalyzerPlugin,
|
|
Version: &v3,
|
|
},
|
|
}
|
|
|
|
requested := semver.MustParseRange("0.2.0")
|
|
result, err := SelectCompatiblePlugin(candidatePlugins, ResourcePlugin, "myplugin", requested)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "myplugin", result.Name)
|
|
assert.Equal(t, "0.2.0", result.Version.String())
|
|
}
|
|
|
|
func TestPluginDownloadUrl(t *testing.T) {
|
|
t.Run("Test Downloading From GitHub Releases", func(t *testing.T) {
|
|
version := semver.MustParse("4.32.0")
|
|
info := PluginInfo{
|
|
PluginDownloadURL: "",
|
|
Name: "aws",
|
|
Version: &version,
|
|
Kind: PluginKind("resource"),
|
|
}
|
|
serverURL := buildGitHubReleasesPluginURL(info.Kind, info.Name, info.Version, "darwin", "amd64")
|
|
assert.Equal(t,
|
|
"https://github.com/pulumi/pulumi-aws/releases/download/v4.32.0/pulumi-resource-aws-v4.32.0-darwin-amd64.tar.gz",
|
|
serverURL)
|
|
})
|
|
t.Run("Test Downloading From get.pulumi.com", func(t *testing.T) {
|
|
version := semver.MustParse("4.32.0")
|
|
info := PluginInfo{
|
|
PluginDownloadURL: "",
|
|
Name: "aws",
|
|
Version: &version,
|
|
Kind: PluginKind("resource"),
|
|
}
|
|
serverURL := buildPulumiHostedPluginURL(info.Kind, info.Name, info.Version, "darwin", "amd64")
|
|
assert.Equal(t,
|
|
"https://get.pulumi.com/releases/plugins/pulumi-resource-aws-v4.32.0-darwin-amd64.tar.gz", serverURL)
|
|
})
|
|
t.Run("Test Downloading From Custom Server URL", func(t *testing.T) {
|
|
version := semver.MustParse("4.32.0")
|
|
info := PluginInfo{
|
|
PluginDownloadURL: "https://customurl.jfrog.io/artifactory/pulumi-packages/package-name",
|
|
Name: "aws",
|
|
Version: &version,
|
|
Kind: PluginKind("resource"),
|
|
}
|
|
serverURL := buildUserSpecifiedPluginURL(info.PluginDownloadURL, info.Kind, info.Name, info.Version,
|
|
"darwin", "amd64")
|
|
assert.Equal(t,
|
|
"https://customurl.jfrog.io/artifactory/pulumi-packages/package-name"+
|
|
"/pulumi-resource-aws-v4.32.0-darwin-amd64.tar.gz",
|
|
serverURL)
|
|
})
|
|
t.Run("Test Searching GitHub Private Releases", func(t *testing.T) {
|
|
owner := "private"
|
|
token := "RaNd0m70K3n_"
|
|
os.Setenv("GITHUB_REPOSITORY_OWNER", owner)
|
|
os.Setenv("GITHUB_TOKEN", token)
|
|
version := semver.MustParse("4.32.0")
|
|
info := PluginInfo{
|
|
PluginDownloadURL: "",
|
|
Name: "private",
|
|
Version: &version,
|
|
Kind: PluginKind("resource"),
|
|
}
|
|
httpReq, err := buildPluginPrivateGitHubReleaseAPIRequest(info)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t,
|
|
"https://api.github.com/repos/private/pulumi-private/releases/tags/v4.32.0",
|
|
httpReq.URL.String())
|
|
assert.Equal(t, fmt.Sprintf("token %s", token), httpReq.Header.Get("Authentication"))
|
|
assert.Equal(t, "application/json", httpReq.Header.Get("Accept"))
|
|
})
|
|
t.Run("Test Downloading From GitHub Private Releases", func(t *testing.T) {
|
|
token := "RaNd0m70K3n_"
|
|
os.Setenv("GITHUB_TOKEN", token)
|
|
url := "https://api.github.com/repos/private/pulumi-private/releases/assets/123456"
|
|
httpReq, err := buildPluginFromPrivateGitHubRequest(url)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, url, httpReq.URL.String())
|
|
assert.Equal(t, fmt.Sprintf("token %s", token), httpReq.Header.Get("Authentication"))
|
|
assert.Equal(t, "application/octet-stream", httpReq.Header.Get("Accept"))
|
|
})
|
|
}
|
|
|
|
func TestInterpolateURL(t *testing.T) {
|
|
version := semver.MustParse("1.0.0")
|
|
const os = "linux"
|
|
const arch = "amd64"
|
|
assert.Equal(t, "", interpolateURL("", version, os, arch))
|
|
assert.Equal(t,
|
|
"https://get.pulumi.com/releases/plugins",
|
|
interpolateURL("https://get.pulumi.com/releases/plugins", version, os, arch))
|
|
assert.Equal(t,
|
|
"https://github.com/org/repo/releases/download/1.0.0",
|
|
interpolateURL("https://github.com/org/repo/releases/download/${VERSION}", version, os, arch))
|
|
assert.Equal(t,
|
|
"https://github.com/org/repo/releases/download/1.0.0/linux/amd64",
|
|
interpolateURL("https://github.com/org/repo/releases/download/${VERSION}/${OS}/${ARCH}", version, os, arch))
|
|
}
|
|
|
|
func TestParsePluginDownloadURLOverride(t *testing.T) {
|
|
type match struct {
|
|
name string
|
|
url string
|
|
ok bool
|
|
}
|
|
|
|
tests := []struct {
|
|
input string
|
|
expected pluginDownloadOverrideArray
|
|
matches []match
|
|
expectError bool
|
|
}{
|
|
{
|
|
input: "",
|
|
expected: nil,
|
|
},
|
|
{
|
|
input: "^foo.*=https://foo",
|
|
expected: pluginDownloadOverrideArray{
|
|
{
|
|
reg: regexp.MustCompile("^foo.*"),
|
|
url: "https://foo",
|
|
},
|
|
},
|
|
matches: []match{
|
|
{
|
|
name: "foo",
|
|
url: "https://foo",
|
|
ok: true,
|
|
},
|
|
{
|
|
name: "foo-bar",
|
|
url: "https://foo",
|
|
ok: true,
|
|
},
|
|
{
|
|
name: "fo",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
{
|
|
name: "",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
{
|
|
name: "nope",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
input: "^foo.*=https://foo,^bar.*=https://bar",
|
|
expected: pluginDownloadOverrideArray{
|
|
{
|
|
reg: regexp.MustCompile("^foo.*"),
|
|
url: "https://foo",
|
|
},
|
|
{
|
|
reg: regexp.MustCompile("^bar.*"),
|
|
url: "https://bar",
|
|
},
|
|
},
|
|
matches: []match{
|
|
{
|
|
name: "foo",
|
|
url: "https://foo",
|
|
ok: true,
|
|
},
|
|
{
|
|
name: "foo-bar",
|
|
url: "https://foo",
|
|
ok: true,
|
|
},
|
|
{
|
|
name: "fo",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
{
|
|
name: "",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
{
|
|
name: "bar",
|
|
url: "https://bar",
|
|
ok: true,
|
|
},
|
|
{
|
|
name: "barbaz",
|
|
url: "https://bar",
|
|
ok: true,
|
|
},
|
|
{
|
|
name: "ba",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
{
|
|
name: "nope",
|
|
url: "",
|
|
ok: false,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
input: "=", // missing regex and url
|
|
expectError: true,
|
|
},
|
|
{
|
|
input: "^foo.*=", // missing url
|
|
expectError: true,
|
|
},
|
|
{
|
|
input: "=https://foo", // missing regex
|
|
expectError: true,
|
|
},
|
|
{
|
|
input: "^foo.*=https://foo,", // trailing comma
|
|
expectError: true,
|
|
},
|
|
{
|
|
input: "[=https://foo", // invalid regex
|
|
expectError: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.input, func(t *testing.T) {
|
|
actual, err := parsePluginDownloadURLOverrides(tt.input)
|
|
if tt.expectError {
|
|
assert.Error(t, err)
|
|
} else {
|
|
assert.NoError(t, err)
|
|
}
|
|
assert.Equal(t, tt.expected, actual)
|
|
|
|
if len(tt.matches) > 0 {
|
|
for _, match := range tt.matches {
|
|
actualURL, actualOK := actual.get(match.name)
|
|
assert.Equal(t, match.url, actualURL)
|
|
assert.Equal(t, match.ok, actualOK)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|