RHINENG-16453: gather table sizes from all used schemas, not only public

Author: TenSt

README.md

@@ -58,8 +58,14 @@ We cover a large part of the application functionality with tests; this requires
 podman-compose -f docker-compose.test.yml up --build --abort-on-container-exit
 ~~~
 
-### Run single test
-After running all test suit, testing platform components are still running (kafka, platform, db). This is especially useful when fixing some test or adding a new one. You need to have golang installed.
+### Run single test or set of tests in containers
+- Open `./scripts/go_test.sh` file
+- Comment line that runs all tests
+- Uncomment and modify the last line to run one or a set of tests
+- Run the same command as for running all tests (from above)
+
+### Run single test locally
+After running all test suite, testing platform components are still running (kafka, platform, db). This is especially useful when fixing some test or adding a new one. You need to have golang installed.
 ~~~bash
 podman-compose -f docker-compose.test.yml up --build --no-start # build images
 podman-compose -f docker-compose.test.yml start db platform kafka # start containers

scripts/go_test.sh

@@ -9,3 +9,6 @@ TEST_DIRS=${1:-./...}
 # Run go test and colorize output (PASS - green, FAIL - red).
 # Set "-p 1" to run test sequentially to avoid parallel changes in testing database.
 gotestsum --format=standard-verbose -- -v -p 1 -coverprofile=coverage.txt -covermode=atomic $TEST_DIRS
+
+# To run one test uncomment and modify last string and comment the one above:
+# go test -count=1 -v ./tasks/vmaas_sync -run TestTableSizes

tasks/vmaas_sync/metrics_db.go

@@ -51,8 +51,9 @@ func updateMetricsWithState(items []keyValue, metrics *prometheus.GaugeVec) {
 
 func getTableSizes() []keyValue {
 	var tableSizes []keyValue
-	err := tasks.CancelableDB().Raw(`select tablename as key, pg_total_relation_size(quote_ident(tablename)) as value
-        from (select * from pg_catalog.pg_tables where schemaname = 'public') t;`).
+	err := tasks.CancelableDB().Raw(`select concat(schemaname, '.', tablename) as key, 
+	    pg_total_relation_size(schemaname || '.' || tablename) as value
+        from (select * from pg_catalog.pg_tables where schemaname in ('public', 'inventory', 'repack')) t;`).
 		Find(&tableSizes).Error
 	if err != nil {
 		utils.LogError("err", err.Error(), "unable to get database table sizes")

tasks/vmaas_sync/metrics_db_test.go

@@ -17,11 +17,12 @@ func TestTableSizes(t *testing.T) {
 	for _, item := range tableSizes {
 		uniqueTables[item.Key] = true
 	}
-	assert.Equal(t, 213, len(tableSizes))
-	assert.Equal(t, 213, len(uniqueTables))
-	assert.True(t, uniqueTables["system_platform"]) // check whether table names were loaded
-	assert.True(t, uniqueTables["package"])
-	assert.True(t, uniqueTables["repo"])
+	assert.Equal(t, 214, len(tableSizes))
+	assert.Equal(t, 214, len(uniqueTables))
+	assert.True(t, uniqueTables["public.system_platform"]) // check whether table names were loaded
+	assert.True(t, uniqueTables["public.package"])
+	assert.True(t, uniqueTables["public.repo"])
+	assert.True(t, uniqueTables["inventory.hosts_v1_0"])
 }
 
 func TestDatabaseSize(t *testing.T) {