RHINENG-22325: inventory views tests

Author: MichaelMraka

base/inventory_views/inventory_views_test.go

@@ -0,0 +1,122 @@
+package inventory_views
+
+import (
+	"app/base/core"
+	"app/base/database"
+	"app/base/models"
+	"app/base/utils"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+)
+
+const rhAccountID = 1
+
+func TestMakeInventoryViewsEvent(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+	tx := database.DB.Begin()
+	defer tx.Rollback()
+
+	var rhAccount models.RhAccount
+	assert.NoError(t, tx.Where("id = ?", rhAccountID).First(&rhAccount).Error)
+	assert.NotEmpty(t, *rhAccount.OrgID)
+
+	orgID := *rhAccount.OrgID
+
+	var systems []models.SystemPlatform
+	assert.NoError(t, tx.Where("rh_account_id = ? AND id in (1,3)", rhAccountID).
+		Order("id").Find(&systems).Error)
+	assert.Equal(t, 2, len(systems))
+
+	// Test MakeInventoryViewsEvent
+	event, err := MakeInventoryViewsEvent(tx, orgID, systems)
+	assert.NoError(t, err)
+	assert.Equal(t, orgID, event.OrgID)
+	assert.NotEmpty(t, event.Timestamp)
+
+	// Verify timestamp is valid RFC3339 format
+	_, err = time.Parse(time.RFC3339, event.Timestamp)
+	assert.NoError(t, err)
+
+	// Verify hosts
+	assert.Equal(t, 2, len(event.Hosts))
+
+	// Check first system (with template)
+	host1 := event.Hosts[0]
+	assert.Equal(t, "00000000-0000-0000-0000-000000000001", host1.ID)
+	assert.Equal(t, 2, host1.Data.ApplicableRhsaCount)
+	assert.Equal(t, 3, host1.Data.ApplicableRhbaCount)
+	assert.Equal(t, 3, host1.Data.ApplicableRheaCount)
+	assert.Equal(t, 0, host1.Data.ApplicableOtherCount)
+	assert.Equal(t, 2, host1.Data.InstallableRhsaCount)
+	assert.Equal(t, 2, host1.Data.InstallableRhbaCount)
+	assert.Equal(t, 1, host1.Data.InstallableRheaCount)
+	assert.Equal(t, 0, host1.Data.InstallableOtherCount)
+	assert.Equal(t, 0, host1.Data.PackagesInstalled)
+	assert.Equal(t, 0, host1.Data.PackagesInstallable)
+	assert.Equal(t, 0, host1.Data.PackagesApplicable)
+	assert.Equal(t, "temp1-1", *host1.Data.TemplateName)
+	assert.Equal(t, "99900000-0000-0000-0000-000000000001", *host1.Data.TemplateUUID)
+
+	host2 := event.Hosts[1]
+	assert.Equal(t, "00000000-0000-0000-0000-000000000003", host2.ID)
+	assert.Equal(t, 0, host2.Data.ApplicableRhsaCount)
+	assert.Equal(t, 0, host2.Data.ApplicableRhbaCount)
+	assert.Equal(t, 1, host2.Data.ApplicableRheaCount)
+	assert.Equal(t, 0, host2.Data.ApplicableOtherCount) // 1 - 1 - 0 - 0
+	assert.Equal(t, 0, host2.Data.InstallableRhsaCount)
+	assert.Equal(t, 0, host2.Data.InstallableRhbaCount)
+	assert.Equal(t, 1, host2.Data.InstallableRheaCount)
+	assert.Equal(t, 0, host2.Data.InstallableOtherCount) // 1 - 1 - 0 - 0
+	assert.Equal(t, 0, host2.Data.PackagesInstalled)
+	assert.Equal(t, 0, host2.Data.PackagesInstallable)
+	assert.Equal(t, 0, host2.Data.PackagesApplicable)
+	assert.Equal(t, "temp2-1", *host2.Data.TemplateName)
+	assert.Equal(t, "99900000-0000-0000-0000-000000000002", *host2.Data.TemplateUUID)
+
+}
+
+func TestMakeInventoryViewsEventEmpty(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+
+	tx := database.DB.Begin()
+	defer tx.Rollback()
+
+	event, err := MakeInventoryViewsEvent(tx, "test-org", []models.SystemPlatform{})
+	assert.NoError(t, err)
+	assert.Equal(t, "test-org", event.OrgID)
+	assert.Equal(t, 0, len(event.Hosts))
+	assert.NotEmpty(t, event.Timestamp)
+}
+
+func TestMakeInventoryViewsEventNoTemplate(t *testing.T) {
+	utils.SkipWithoutDB(t)
+	core.SetupTestEnvironment()
+
+	tx := database.DB.Begin()
+	defer tx.Rollback()
+
+	var rhAccount models.RhAccount
+	assert.NoError(t, tx.Where("id = ?", rhAccountID).First(&rhAccount).Error)
+	assert.NotEmpty(t, *rhAccount.OrgID)
+	orgID := *rhAccount.OrgID
+
+	var systems []models.SystemPlatform
+	assert.NoError(t, tx.Where("rh_account_id = ? AND id in (4)", rhAccountID).
+		Order("id").Find(&systems).Error)
+	assert.Equal(t, 1, len(systems))
+
+	// Should not error, but template fields should be nil
+	event, err := MakeInventoryViewsEvent(tx, orgID, systems)
+	assert.NoError(t, err)
+	assert.Equal(t, 1, len(event.Hosts))
+
+	host := event.Hosts[0]
+	assert.Equal(t, "00000000-0000-0000-0000-000000000004", host.ID)
+	// Template fields should be nil when template is not found
+	assert.Nil(t, host.Data.TemplateName)
+	assert.Nil(t, host.Data.TemplateUUID)
+}

conf/test.env

@@ -10,7 +10,7 @@ DB_PASSWD=passwd
 LIMIT_PAGE_SIZE=false
 
 # don't put "" or '' around the text otherwise they'll be included into content
-POD_CONFIG=label=upload;vmaas_call_max_retries=100;template_change_eval=false;update_users;update_db_config;use_testing_db
+POD_CONFIG=label=upload;vmaas_call_max_retries=100;template_change_eval=false;update_users;update_db_config;use_testing_db;inventory_views
 
 KESSEL_URL=platform:9005
 KESSEL_INSECURE=true