Merge branch 'mo-update-pipeline-artifacts-size-in-root-storage-statistics' into 'master'

Update root_storage_statistics with pipeline_artifacts_size

See merge request gitlab-org/gitlab!40539

app/models/namespace/root_storage_statistics.rb

@@ -2,7 +2,16 @@
 
 class Namespace::RootStorageStatistics < ApplicationRecord
   SNIPPETS_SIZE_STAT_NAME = 'snippets_size'.freeze
-  STATISTICS_ATTRIBUTES = %W(storage_size repository_size wiki_size lfs_objects_size build_artifacts_size packages_size #{SNIPPETS_SIZE_STAT_NAME}).freeze
+  STATISTICS_ATTRIBUTES = %W(
+    storage_size
+    repository_size
+    wiki_size
+    lfs_objects_size
+    build_artifacts_size
+    packages_size
+    #{SNIPPETS_SIZE_STAT_NAME}
+    pipeline_artifacts_size
+  ).freeze
 
   self.primary_key = :namespace_id
 
@@ -40,7 +49,8 @@ class Namespace::RootStorageStatistics < ApplicationRecord
         'COALESCE(SUM(ps.lfs_objects_size), 0) AS lfs_objects_size',
         'COALESCE(SUM(ps.build_artifacts_size), 0) AS build_artifacts_size',
         'COALESCE(SUM(ps.packages_size), 0) AS packages_size',
-        "COALESCE(SUM(ps.snippets_size), 0) AS #{SNIPPETS_SIZE_STAT_NAME}"
+        "COALESCE(SUM(ps.snippets_size), 0) AS #{SNIPPETS_SIZE_STAT_NAME}",
+        'COALESCE(SUM(ps.pipeline_artifacts_size), 0) AS pipeline_artifacts_size'
       )
   end
 

spec/models/namespace/root_storage_statistics_spec.rb

@@ -44,6 +44,7 @@ RSpec.describe Namespace::RootStorageStatistics, type: :model do
         total_packages_size = stat1.packages_size + stat2.packages_size
         total_storage_size = stat1.storage_size + stat2.storage_size
         total_snippets_size = stat1.snippets_size + stat2.snippets_size
+        total_pipeline_artifacts_size = stat1.pipeline_artifacts_size + stat2.pipeline_artifacts_size
 
         expect(root_storage_statistics.repository_size).to eq(total_repository_size)
         expect(root_storage_statistics.wiki_size).to eq(total_wiki_size)
@@ -52,6 +53,7 @@ RSpec.describe Namespace::RootStorageStatistics, type: :model do
         expect(root_storage_statistics.packages_size).to eq(total_packages_size)
         expect(root_storage_statistics.storage_size).to eq(total_storage_size)
         expect(root_storage_statistics.snippets_size).to eq(total_snippets_size)
+        expect(root_storage_statistics.pipeline_artifacts_size).to eq(total_pipeline_artifacts_size)
       end
 
       it 'works when there are no projects' do
@@ -67,6 +69,7 @@ RSpec.describe Namespace::RootStorageStatistics, type: :model do
         expect(root_storage_statistics.packages_size).to eq(0)
         expect(root_storage_statistics.storage_size).to eq(0)
         expect(root_storage_statistics.snippets_size).to eq(0)
+        expect(root_storage_statistics.pipeline_artifacts_size).to eq(0)
       end
     end