From c79a47a40ce15f31ed69cdc5034eab032b8c0035 Mon Sep 17 00:00:00 2001 From: DanStough Date: Fri, 1 Sep 2023 18:02:25 -0400 Subject: [PATCH] test: fix snapshot save w/ append test for ENT --- command/snapshot/save/snapshot_save_test.go | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/command/snapshot/save/snapshot_save_test.go b/command/snapshot/save/snapshot_save_test.go index 8e339cd51f5..fb672b9993b 100644 --- a/command/snapshot/save/snapshot_save_test.go +++ b/command/snapshot/save/snapshot_save_test.go @@ -101,7 +101,20 @@ func TestSnapshotSaveCommandWithAppendFileNameFlag(t *testing.T) { status = "leader" } - newFilePath := filepath.Join(dir, "backup"+"-"+a.Config.Version+"-"+a.Config.Datacenter+ + // We need to use the self endpoint here for ENT, which returns the product suffix (+ent) + self, err := client.Agent().Self() + require.NoError(t, err) + + cfg, ok := self["Config"] + require.True(t, ok) + + versionAny, ok := cfg["Version"] + require.True(t, ok) + + version, ok := versionAny.(string) + require.True(t, ok) + + newFilePath := filepath.Join(dir, "backup"+"-"+version+"-"+a.Config.Datacenter+ "-"+a.Config.NodeName+"-"+status+".tgz") code := c.Run(args)