RHINENG-17345: add tests for candlepin env check

Author: MichaelMraka

listener/upload_test.go

@@ -429,3 +429,48 @@ func TestGetRepoPath(t *testing.T) {
 	assert.Nil(t, err)
 	assert.Equal(t, "/content/dist/rhel8/rhui/8.4/x86_64/baseos/os", repoPath)
 }
+
+func TestHostTemplateRhsmReporter(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+	configure()
+
+	accountID := int(1)
+	host := &Host{
+		ID:       id,
+		Reporter: rhsmReporter,
+		SystemProfile: inventory.SystemProfile{
+			Rhsm: inventory.Rhsm{
+				Environments: []string{"99900000000000000000000000000001", "99900000000000000000000000000002"},
+			},
+		},
+	}
+
+	templateID := hostTemplate(database.DB, accountID, host)
+	assert.NotNil(t, templateID)
+	assert.Equal(t, int64(1), *templateID)
+}
+
+func TestHostTemplatePuptoo(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+	configure()
+
+	accountID := int(1)
+	host := &Host{
+		ID:       id,
+		Reporter: puptooReporter,
+		SystemProfile: inventory.SystemProfile{
+			ConsumerID: "00000000-0000-0000-0000-000000000002",
+			YumRepos: &[]inventory.YumRepo{{
+				ID:      "base",
+				Enabled: true,
+				BaseURL: "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)
+}

platform/candlepin.go

@@ -1,10 +1,12 @@
 package platform
 
 import (
+	"app/base/candlepin"
 	"app/base/utils"
 	"fmt"
 	"io"
 	"net/http"
+	"strings"
 
 	"github.com/gin-gonic/gin"
 )
@@ -30,18 +32,32 @@ func candlepinEnvHandler(c *gin.Context) {
 	c.Data(http.StatusOK, gin.MIMEJSON, []byte(data))
 }
 
-func candlepinConsumersHandler(c *gin.Context) {
+func candlepinConsumersPutHandler(c *gin.Context) {
 	consumer := c.Param("consumer")
 	jsonData, _ := io.ReadAll(c.Request.Body)
-	utils.LogInfo("consumer", consumer, "body", string(jsonData))
+	utils.LogInfo("PUT consumer", consumer, "body", string(jsonData))
 	if consumer == "return_404" {
 		c.Data(http.StatusNotFound, gin.MIMEJSON, []byte{})
 		return
 	}
 	c.Data(http.StatusOK, gin.MIMEJSON, []byte{})
 }
 
+func candlepinConsumersGetHandler(c *gin.Context) {
+	consumer := c.Param("consumer")
+	utils.LogInfo("GET consumer", consumer, "body")
+	env := strings.ReplaceAll(consumer, "-", "")
+	env = strings.Replace(env, "000", "999", 1)
+	response := candlepin.ConsumersDetailResponse{
+		Environments: []candlepin.ConsumersEnvironment{
+			{ID: env},
+		},
+	}
+	c.JSON(http.StatusOK, response)
+}
+
 func initCandlepin(app *gin.Engine) {
 	app.POST("/candlepin/environments/:envid/consumers", candlepinEnvHandler)
-	app.PUT("/candlepin/consumers/:consumer", candlepinConsumersHandler)
+	app.PUT("/candlepin/consumers/:consumer", candlepinConsumersPutHandler)
+	app.GET("/candlepin/consumers/:consumer", candlepinConsumersGetHandler)
 }