From 64201c83522d30a95bba7b5a3f02cb0b5848df09 Mon Sep 17 00:00:00 2001
From: jregan <jregan@google.com>
Date: Mon, 30 Nov 2020 19:17:15 -0800
Subject: [PATCH] Adjust SecretGeneratorTest to pass with --enable_kyaml

---
 .../secretgenerator/SecretGenerator_test.go      | 16 +++++++++++++---
 1 file changed, 13 insertions(+), 3 deletions(-)

diff --git a/plugin/builtin/secretgenerator/SecretGenerator_test.go b/plugin/builtin/secretgenerator/SecretGenerator_test.go
index 70fb7b4bd..7a418ec6c 100644
--- a/plugin/builtin/secretgenerator/SecretGenerator_test.go
+++ b/plugin/builtin/secretgenerator/SecretGenerator_test.go
@@ -4,8 +4,10 @@
 package main_test
 
 import (
+	"fmt"
 	"testing"
 
+	"sigs.k8s.io/kustomize/api/konfig"
 	kusttest_test "sigs.k8s.io/kustomize/api/testutils/kusttest"
 )
 
@@ -42,18 +44,26 @@ literals:
 - VEGETABLE=carrot
 `)
 
-	th.AssertActualEqualsExpected(rm, `
+	obscure := `obscure: CkxvcmVtIGlwc3VtIGRvbG9yIHNpdCBhbWV0LApjb25zZWN0ZXR1ciBhZGlwaXNjaW5nIGVsaXQuCg==`
+	if konfig.FlagEnableKyamlDefaultValue {
+		// The kyaml code does a better job of line wrapping.
+		obscure = `obscure: |
+    CkxvcmVtIGlwc3VtIGRvbG9yIHNpdCBhbWV0LApjb25zZWN0ZXR1ciBhZGlwaXNjaW5nIG
+    VsaXQuCg==`
+	}
+
+	th.AssertActualEqualsExpected(rm, fmt.Sprintf(`
 apiVersion: v1
 data:
   DB_PASSWORD: aWxvdmV5b3U=
   FRUIT: YXBwbGU=
   ROUTER_PASSWORD: YWRtaW4=
   VEGETABLE: Y2Fycm90
-  obscure: CkxvcmVtIGlwc3VtIGRvbG9yIHNpdCBhbWV0LApjb25zZWN0ZXR1ciBhZGlwaXNjaW5nIGVsaXQuCg==
+  %s
 kind: Secret
 metadata:
   name: mySecret
   namespace: whatever
 type: Opaque
-`)
+`, obscure))
 }