1 files changed, 24 insertions, 7 deletions
diff --git a/spec/lib/bulk_imports/common/rest/get_badges_query_spec.rb b/spec/lib/bulk_imports/common/rest/get_badges_query_spec.rb
index 0a04c0a2243..fabef50af8b 100644
--- a/spec/lib/bulk_imports/common/rest/get_badges_query_spec.rb
+++ b/spec/lib/bulk_imports/common/rest/get_badges_query_spec.rb
@@ -9,15 +9,32 @@ RSpec.describe BulkImports::Common::Rest::GetBadgesQuery do
       let(:context) { BulkImports::Pipeline::Context.new(tracker) }
       let(:encoded_full_path) { ERB::Util.url_encode(entity.source_full_path) }
 
-      it 'returns correct query and page info' do
-        expected = {
-          resource: [entity.pluralized_name, encoded_full_path, 'badges'].join('/'),
-          query: {
-            page: context.tracker.next_page
+      context 'when source id is present' do
+        it 'returns correct query using source id and page info' do
+          expected = {
+            resource: [entity.base_resource_path, 'badges'].join('/'),
+            query: {
+              page: context.tracker.next_page
+            }
           }
-        }
 
-        expect(described_class.to_h(context)).to eq(expected)
+          expect(described_class.to_h(context)).to eq(expected)
+        end
+      end
+
+      context 'when source id is missing' do
+        it 'returns correct query using source full path' do
+          entity.update!(source_xid: nil)
+
+          expected = {
+            resource: ["/#{entity.pluralized_name}", encoded_full_path, 'badges'].join('/'),
+            query: {
+              page: context.tracker.next_page
+            }
+          }
+
+          expect(described_class.to_h(context)).to eq(expected)
+        end
       end
     end