RHINENG-20967: test candlepin environment assignment

Author: MichaelMraka

docker-compose.test.yml

@@ -43,12 +43,14 @@ services:
     env_file:
       - ./conf/common.env
       - ./conf/platform.env
+      - ./conf/gorun.env
     command: ./dev/scripts/docker-compose-entrypoint.sh platform
     restart: unless-stopped
     ports:
       - 9001:9001
       - 9005:9005
     volumes:
+      - ./:/go/src/app/
       - ./conf/cdappconfig.json:/go/src/app/conf/cdappconfig.json
     depends_on:
       - kafka

listener/rhsm_test.go

@@ -4,6 +4,7 @@ import (
 	"app/base/core"
 	"app/base/database"
 	"app/base/utils"
+	"context"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
@@ -48,3 +49,38 @@ func TestAssignTemplates(t *testing.T) {
 		})
 	}
 }
+
+func TestCallCandlepinEnvironment(t *testing.T) {
+	ctx := context.Background()
+	result, err := callCandlepinEnvironment(ctx, "00000000-0000-0000-0000-000000000001")
+
+	assert.NoError(t, err)
+	assert.NotNil(t, result)
+	assert.Len(t, result.Environments, 1)
+	assert.Equal(t, "99900000000000000000000000000001", result.Environments[0].ID)
+}
+
+func TestCallCandlepinEnvironmentError(t *testing.T) {
+	ctx := context.Background()
+	result, err := callCandlepinEnvironment(ctx, "return_404")
+
+	assert.Error(t, err)
+	assert.Contains(t, err.Error(), "candlepin /consumers call failed")
+	assert.Nil(t, result)
+}
+
+func TestCallCandlepinEnvNetworkError(t *testing.T) {
+	// Override the candlepin address to invalid URL
+	originalAddress := utils.CoreCfg.CandlepinAddress
+	utils.CoreCfg.CandlepinAddress = "http://invalid-host:12345"
+	defer func() {
+		utils.CoreCfg.CandlepinAddress = originalAddress
+	}()
+
+	ctx := context.Background()
+	result, err := callCandlepinEnvironment(ctx, "test-consumer")
+
+	assert.Error(t, err)
+	assert.Contains(t, err.Error(), "candlepin error")
+	assert.Nil(t, result)
+}

listener/upload_test.go

@@ -21,6 +21,8 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
+var accountID = int(1)
+
 func assertInLogs(t *testing.T, msg string, logs ...log.Entry) {
 	nLogs := len(logs)
 	i := 0
@@ -52,6 +54,20 @@ func createTestInvHost(t *testing.T) *Host {
 	return &host
 }
 
+func createTestHostWithEnv(reporter, consumer, baseURL string) *Host {
+	return &Host{
+		ID:       id,
+		Reporter: reporter,
+		SystemProfile: inventory.SystemProfile{
+			ConsumerID: consumer,
+			YumRepos: &[]inventory.YumRepo{{
+				ID:      "base",
+				Enabled: true,
+				BaseURL: baseURL}},
+		},
+	}
+}
+
 func TestUpdateSystemPlatform(t *testing.T) {
 	utils.SkipWithoutDB(t)
 	core.SetupTestEnvironment()
@@ -435,17 +451,9 @@ func TestHostTemplateRhsmReporter(t *testing.T) {
 	core.SetupTestEnvironment()
 	configure()
 
-	accountID := int(1)
-	host := &Host{
-		ID:       id,
-		Reporter: rhsmReporter,
-		SystemProfile: inventory.SystemProfile{
-			Rhsm: inventory.Rhsm{
-				Environments: []string{"99900000000000000000000000000001", "99900000000000000000000000000002"},
-			},
-		},
-	}
-
+	host := createTestHostWithEnv(rhsmReporter, "00000000-0000-0000-0000-000000000001",
+		"https://cert.console.example.com/api/pulp-content/abcdef/templates/"+
+			"12345678-90ab-cdef-1234-567890abcdef/content/dist/rhel9/$releasever/x86_64/baseos/os")
 	templateID := hostTemplate(database.DB, accountID, host)
 	assert.NotNil(t, templateID)
 	assert.Equal(t, int64(1), *templateID)
@@ -456,19 +464,33 @@ func TestHostTemplatePuptoo(t *testing.T) {
 	core.SetupTestEnvironment()
 	configure()
 
-	accountID := int(1)
-	host := &Host{
-		ID:       id,
-		Reporter: puptooReporter,
-		SystemProfile: inventory.SystemProfile{
-			ConsumerID: "00000000-0000-0000-0000-000000000002",
-			Rhsm: inventory.Rhsm{
-				Environments: []string{"99900000000000000000000000000002"},
-			},
-		},
-	}
-
+	host := createTestHostWithEnv(puptooReporter, "00000000-0000-0000-0000-000000000002",
+		"https://cert.console.example.com/api/pulp-content/abcdef/templates/"+
+			"12345678-90ab-cdef-1234-567890abcdef/content/dist/rhel9/$releasever/x86_64/baseos/os")
 	templateID := hostTemplate(database.DB, accountID, host)
 	assert.NotNil(t, templateID)
 	assert.Equal(t, int64(2), *templateID)
 }
+
+func TestNoHostTemplate(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+	configure()
+
+	host := createTestHostWithEnv(puptooReporter, "00000000-0000-0000-0000-000000000002",
+		"https://cdn.example.com/content/dist/rhel9/$releasever/x86_64/baseos/os")
+	templateID := hostTemplate(database.DB, accountID, host)
+	assert.Nil(t, templateID)
+}
+
+func TestHostTemplateCandlepinFailure(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+	configure()
+
+	host := createTestHostWithEnv(puptooReporter, "return_404",
+		"https://cert.console.example.com/api/pulp-content/abcdef/templates/"+
+			"12345678-90ab-cdef-1234-567890abcdef/content/dist/rhel9/$releasever/x86_64/baseos/os")
+	templateID := hostTemplate(database.DB, accountID, host)
+	assert.Nil(t, templateID)
+}