Fix plugin protocol test

This commit is contained in:
findkim 2019-01-09 15:20:33 -06:00
parent cdf7cc2449
commit 43b22d4e14
1 changed files with 16 additions and 4 deletions

View File

@ -203,6 +203,18 @@ func TestCheckProtocolVersions(t *testing.T) {
}, },
false, false,
}, },
{
&response.TerraformProviderVersion{
Protocols: []string{"4.0", "5.2"},
},
false,
},
{
&response.TerraformProviderVersion{
Protocols: []string{"5.0", "6.1"},
},
true,
},
} }
server := testReleaseServer() server := testReleaseServer()
@ -216,7 +228,6 @@ func TestCheckProtocolVersions(t *testing.T) {
if err == nil { if err == nil {
t.Fatal("succeeded; want error") t.Fatal("succeeded; want error")
} }
return
} else if err != nil { } else if err != nil {
t.Fatalf("unexpected error: %s", err) t.Fatalf("unexpected error: %s", err)
} }
@ -456,9 +467,10 @@ func TestProviderChecksum(t *testing.T) {
// newProviderInstaller returns a minimally-initialized ProviderInstaller // newProviderInstaller returns a minimally-initialized ProviderInstaller
func newProviderInstaller(s *httptest.Server) ProviderInstaller { func newProviderInstaller(s *httptest.Server) ProviderInstaller {
return ProviderInstaller{ return ProviderInstaller{
registry: registry.NewClient(Disco(s), nil), registry: registry.NewClient(Disco(s), nil),
OS: runtime.GOOS, OS: runtime.GOOS,
Arch: runtime.GOARCH, Arch: runtime.GOARCH,
PluginProtocolVersion: 4,
} }
} }