From 43bf2dd0cfdc83a414b0fa56aae9fc63058e8325 Mon Sep 17 00:00:00 2001
From: Michael Mraka <michael.mraka@redhat.com>
Date: Fri, 31 May 2024 15:34:33 +0200
Subject: [PATCH] RHINENG-10439: update template test for environment id

---
 base/database/testing.go  |  3 ++-
 dev/test_data.sql         | 10 +++++-----
 listener/template_test.go | 13 ++++++++-----
 platform/platform.go      |  2 ++
 4 files changed, 17 insertions(+), 11 deletions(-)

diff --git a/base/database/testing.go b/base/database/testing.go
index dcab49007..cfeefc3a4 100644
--- a/base/database/testing.go
+++ b/base/database/testing.go
@@ -6,6 +6,7 @@ import (
 	"app/base/utils"
 	"fmt"
 	"strconv"
+	"strings"
 	"testing"
 	"time"
 
@@ -493,7 +494,7 @@ func DeleteSystem(t *testing.T, inventoryID string) {
 
 func CreateTemplate(t *testing.T, account int, uuid string, inventoryIDs []string) {
 	template := &models.Template{
-		RhAccountID: account, UUID: uuid, Name: uuid,
+		RhAccountID: account, UUID: uuid, Name: uuid, EnvironmentID: strings.ReplaceAll(uuid, "-", ""),
 	}
 
 	tx := DB.Begin()
diff --git a/dev/test_data.sql b/dev/test_data.sql
index 23a7b2752..f525dcc05 100644
--- a/dev/test_data.sql
+++ b/dev/test_data.sql
@@ -25,11 +25,11 @@ INSERT INTO baseline (id, rh_account_id, name, config, description) VALUES
 (3, 1, 'baseline_1-3', '{"to_time": "2000-01-01T00:00:00+00:00"}', NULL),
 (4, 3, 'baseline_3-4', '{"to_time": "2000-01-01T00:00:00+00:00"}', NULL);
 
-INSERT INTO template (id, rh_account_id, uuid, name, description, config, creator) VALUES
-(1, 1, '99900000-0000-0000-0000-000000000001', 'temp1-1', 'desc1', '{"to_time": "2010-09-22T00:00:00+00:00"}', 'user1'),
-(2, 1, '99900000-0000-0000-0000-000000000002', 'temp2-1', 'desc2', '{"to_time": "2021-01-01T00:00:00+00:00"}', 'user2'),
-(3, 1, '99900000-0000-0000-0000-000000000003', 'temp3-1',    NULL, '{"to_time": "2000-01-01T00:00:00+00:00"}', 'user3'),
-(4, 3, '99900000-0000-0000-0000-000000000004', 'temp4-3', 'desc4', '{"to_time": "2000-01-01T00:00:00+00:00"}', 'user4');
+INSERT INTO template (id, rh_account_id, uuid, environment_id, name, description, config, creator) VALUES
+(1, 1, '99900000-0000-0000-0000-000000000001', '99900000000000000000000000000001', 'temp1-1', 'desc1', '{"to_time": "2010-09-22T00:00:00+00:00"}', 'user1'),
+(2, 1, '99900000-0000-0000-0000-000000000002', '99900000000000000000000000000002', 'temp2-1', 'desc2', '{"to_time": "2021-01-01T00:00:00+00:00"}', 'user2'),
+(3, 1, '99900000-0000-0000-0000-000000000003', '99900000000000000000000000000003', 'temp3-1',    NULL, '{"to_time": "2000-01-01T00:00:00+00:00"}', 'user3'),
+(4, 3, '99900000-0000-0000-0000-000000000004', '99900000000000000000000000000004', 'temp4-3', 'desc4', '{"to_time": "2000-01-01T00:00:00+00:00"}', 'user4');
 
 INSERT INTO system_platform (id, inventory_id, display_name, rh_account_id, reporter_id, vmaas_json, json_checksum, last_evaluation, last_upload, packages_installed, packages_installable, packages_applicable, third_party, baseline_id, baseline_uptodate, template_id) VALUES
 (1, '00000000-0000-0000-0000-000000000001','00000000-0000-0000-0000-000000000001', 1, 1, '{ "package_list": [ "kernel-2.6.32-696.20.1.el6.x86_64" ], "repository_list": [ "rhel-6-server-rpms" ] }', '1', '2018-09-22 12:00:00-04', '2020-09-22 12:00:00-04',0,0,0, true, 1, true, 1),
diff --git a/listener/template_test.go b/listener/template_test.go
index ec8f87378..d21c2c949 100644
--- a/listener/template_test.go
+++ b/listener/template_test.go
@@ -9,6 +9,7 @@ import (
 	"encoding/json"
 	"errors"
 	"fmt"
+	"strings"
 	"testing"
 	"time"
 
@@ -24,12 +25,14 @@ func createTempateMsg(t *testing.T, eventName, orgID string, nTemplates int) mqu
 	templates := make([]mqueue.TemplateResponse, nTemplates)
 	for i := 0; i < nTemplates; i++ {
 		description := fmt.Sprintf("Template%d description", i)
+		uuid := fmt.Sprintf("77777777-0000-0000-0000-00000000000%1d", i)
 		templates[i] = mqueue.TemplateResponse{
-			UUID:        fmt.Sprintf("77777777-0000-0000-0000-00000000000%1d", i),
-			Name:        fmt.Sprintf("Template%d", i),
-			OrgID:       orgID,
-			Description: &description,
-			Date:        time.Now(),
+			UUID:          uuid,
+			EnvironmentID: strings.ReplaceAll(uuid, "-", ""),
+			Name:          fmt.Sprintf("Template%d", i),
+			OrgID:         orgID,
+			Description:   &description,
+			Date:          time.Now(),
 		}
 	}
 
diff --git a/platform/platform.go b/platform/platform.go
index 51e41a32b..97fb61244 100644
--- a/platform/platform.go
+++ b/platform/platform.go
@@ -164,6 +164,7 @@ var templates = []mqueue.TemplateResponse{
 		Version:         "1",
 		Date:            time.Now(),
 		RepositoryUUIDS: []string{"20000000-0000-0000-0000-000000000001"},
+		EnvironmentID:   "10000000000000000000000000000001",
 	},
 	{
 		UUID:            "10000000-0000-0000-0000-000000000002",
@@ -174,6 +175,7 @@ var templates = []mqueue.TemplateResponse{
 		Version:         "1",
 		Date:            time.Now(),
 		RepositoryUUIDS: []string{"20000000-0000-0000-0000-000000000001", "20000000-0000-0000-0000-000000000002"},
+		EnvironmentID:   "10000000000000000000000000000002",
 	},
 }