1 files changed, 10 insertions, 2 deletions
diff --git a/spec/requests/api/graphql/mutations/user_preferences/update_spec.rb b/spec/requests/api/graphql/mutations/user_preferences/update_spec.rb
index 967ad75c906..65b8083c74f 100644
--- a/spec/requests/api/graphql/mutations/user_preferences/update_spec.rb
+++ b/spec/requests/api/graphql/mutations/user_preferences/update_spec.rb
@@ -11,7 +11,8 @@ RSpec.describe Mutations::UserPreferences::Update, feature_category: :user_profi
 
   let(:input) do
     {
-      'issuesSort' => sort_value
+      'issuesSort' => sort_value,
+      'visibilityPipelineIdType' => 'IID'
     }
   end
 
@@ -24,15 +25,20 @@ RSpec.describe Mutations::UserPreferences::Update, feature_category: :user_profi
 
       expect(response).to have_gitlab_http_status(:success)
       expect(mutation_response['userPreferences']['issuesSort']).to eq(sort_value)
+      expect(mutation_response['userPreferences']['visibilityPipelineIdType']).to eq('IID')
 
       expect(current_user.user_preference.persisted?).to eq(true)
       expect(current_user.user_preference.issues_sort).to eq(Types::IssueSortEnum.values[sort_value].value.to_s)
+      expect(current_user.user_preference.visibility_pipeline_id_type).to eq('iid')
     end
   end
 
   context 'when user has existing preference' do
     before do
-      current_user.create_user_preference!(issues_sort: Types::IssueSortEnum.values['TITLE_DESC'].value)
+      current_user.create_user_preference!(
+        issues_sort: Types::IssueSortEnum.values['TITLE_DESC'].value,
+        visibility_pipeline_id_type: 'id'
+      )
     end
 
     it 'updates the existing value' do
@@ -42,8 +48,10 @@ RSpec.describe Mutations::UserPreferences::Update, feature_category: :user_profi
 
       expect(response).to have_gitlab_http_status(:success)
       expect(mutation_response['userPreferences']['issuesSort']).to eq(sort_value)
+      expect(mutation_response['userPreferences']['visibilityPipelineIdType']).to eq('IID')
 
       expect(current_user.user_preference.issues_sort).to eq(Types::IssueSortEnum.values[sort_value].value.to_s)
+      expect(current_user.user_preference.visibility_pipeline_id_type).to eq('iid')
     end
   end
 end