private/protocol: Fix benchmarks (#329)

Fixes the restxml/restjson benchmarks.
aws · Jun 25, 2019 · 4f5e4b5 · 4f5e4b5
1 parent 76aec52
commit 4f5e4b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/private/protocol/restjson/build_bench_test.go b/private/protocol/restjson/build_bench_test.go
@@ -1,5 +1,3 @@
-// +build bench
-
 package restjson_test
 
 import (
@@ -15,7 +13,7 @@ import (
 )
 
 var (
-	elastictranscoderSvc *elastictranscoder.ElasticTranscoder
+	elastictranscoderSvc *elastictranscoder.Client
 )
 
 func TestMain(m *testing.M) {
@@ -119,7 +117,7 @@ func elastictranscoderCreateJobInput() *elastictranscoder.CreateJobInput {
 			Resolution: aws.String("Resolution"),
 		},
 		PipelineId: aws.String("Id"), // Required
-		Output: &elastictranscoder.CreateJobOutput{
+		Output: &elastictranscoder.CreateJobOutputResult{
 			AlbumArt: &elastictranscoder.JobAlbumArt{
 				Artwork: []elastictranscoder.Artwork{
 					{ // Required
@@ -212,7 +210,7 @@ func elastictranscoderCreateJobInput() *elastictranscoder.CreateJobInput {
 			},
 		},
 		OutputKeyPrefix: aws.String("Key"),
-		Outputs: []elastictranscoder.CreateJobOutput{
+		Outputs: []elastictranscoder.CreateJobOutputResult{
 			{ // Required
 				AlbumArt: &elastictranscoder.JobAlbumArt{
 					Artwork: []elastictranscoder.Artwork{

diff --git a/private/protocol/restxml/build_bench_test.go b/private/protocol/restxml/build_bench_test.go
@@ -1,5 +1,3 @@
-// +build bench
-
 package restxml_test
 
 import (
@@ -20,8 +18,8 @@ import (
 )
 
 var (
-	cloudfrontSvc *cloudfront.CloudFront
-	s3Svc         *s3.S3
+	cloudfrontSvc *cloudfront.Client
+	s3Svc         *s3.Client
 )
 
 func TestMain(m *testing.M) {