From 43b22d4e1441e159eaca431a9bef6b7f58f00356 Mon Sep 17 00:00:00 2001
From: findkim <kngo@hashicorp.com>
Date: Wed, 9 Jan 2019 15:20:33 -0600
Subject: [PATCH] Fix plugin protocol test

---
 plugin/discovery/get_test.go | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

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