1 files changed, 14 insertions, 3 deletions
diff --git a/spec/graphql/types/ci/runner_upgrade_status_type_enum_spec.rb b/spec/graphql/types/ci/runner_upgrade_status_type_enum_spec.rb
index 81a852471b9..03c784dcbe7 100644
--- a/spec/graphql/types/ci/runner_upgrade_status_type_enum_spec.rb
+++ b/spec/graphql/types/ci/runner_upgrade_status_type_enum_spec.rb
@@ -3,11 +3,22 @@
 require 'spec_helper'
 
 RSpec.describe Types::Ci::RunnerUpgradeStatusTypeEnum do
+  let(:model_only_enum_values) { %w[not_processed] }
+  let(:expected_graphql_source_values) do
+    Ci::RunnerVersion.statuses.keys - model_only_enum_values
+  end
+
   specify { expect(described_class.graphql_name).to eq('CiRunnerUpgradeStatusType') }
 
-  it 'exposes all upgrade status values' do
-    expect(described_class.values.keys).to eq(
-      ['UNKNOWN'] + ::Gitlab::Ci::RunnerUpgradeCheck::STATUSES.map { |sym, _| sym.to_s.upcase }
+  it 'exposes all upgrade status values except not_processed' do
+    expect(described_class.values.keys).to match_array(
+      expected_graphql_source_values
+        .map(&:upcase)
+        .map { |v| v == 'INVALID_VERSION' ? 'INVALID' : v }
     )
   end
+
+  it 'exposes all upgrade status values except enum-only values' do
+    expect(described_class.values.values.map(&:value).map(&:to_s)).to match_array(expected_graphql_source_values)
+  end
 end