From 1157f0dcfe4833c2db53642575993b0d8675ca9a Mon Sep 17 00:00:00 2001
From: Gennady Lipenkov <xgen@yandex-team.ru>
Date: Wed, 17 Jun 2020 17:06:05 +0300
Subject: [PATCH] Add acceptance test

---
 builder/yandex/builder_acc_test.go | 41 ++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/builder/yandex/builder_acc_test.go b/builder/yandex/builder_acc_test.go
index 44a3b0f9f..2fa759f7e 100644
--- a/builder/yandex/builder_acc_test.go
+++ b/builder/yandex/builder_acc_test.go
@@ -1,12 +1,17 @@
 package yandex
 
 import (
+	"fmt"
 	"os"
 	"testing"
 
+	"github.com/go-resty/resty/v2"
+
 	builderT "github.com/hashicorp/packer/helper/builder/testing"
 )
 
+const InstanceMetadataAddr = "169.254.169.254"
+
 func TestBuilderAcc_basic(t *testing.T) {
 	builderT.Test(t, builderT.TestCase{
 		PreCheck: func() { testAccPreCheck(t) },
@@ -15,6 +20,14 @@ func TestBuilderAcc_basic(t *testing.T) {
 	})
 }
 
+func TestBuilderAcc_instanceSA(t *testing.T) {
+	builderT.Test(t, builderT.TestCase{
+		PreCheck: func() { testAccPreCheckInstanceSA(t) },
+		Builder:  &Builder{},
+		Template: testBuilderAccInstanceSA,
+	})
+}
+
 func testAccPreCheck(t *testing.T) {
 	if v := os.Getenv("YC_TOKEN"); v == "" {
 		t.Fatal("YC_TOKEN must be set for acceptance tests")
@@ -24,6 +37,19 @@ func testAccPreCheck(t *testing.T) {
 	}
 }
 
+func testAccPreCheckInstanceSA(t *testing.T) {
+	client := resty.New()
+
+	_, err := client.R().SetHeader("Metadata-Flavor", "Google").Get(tokenUrl())
+	if err != nil {
+		t.Fatalf("error get Service Account token assignment", err)
+	}
+
+	if v := os.Getenv("YC_FOLDER_ID"); v == "" {
+		t.Fatal("YC_FOLDER_ID must be set for acceptance tests")
+	}
+}
+
 const testBuilderAccBasic = `
 {
 	"builders": [{
@@ -34,3 +60,18 @@ const testBuilderAccBasic = `
 	}]
 }
 `
+
+const testBuilderAccInstanceSA = `
+{
+	"builders": [{
+		"type": "test",
+        "source_image_family": "ubuntu-1804-lts",
+		"use_ipv4_nat": "true",
+		"ssh_username": "ubuntu"
+	}]
+}
+`
+
+func tokenUrl() string {
+	return fmt.Sprintf("http://%s/computeMetadata/v1/instance/service-accounts/default/token", InstanceMetadataAddr)
+}