spec/routemaster/api_client_spec.rb in routemaster-drain-3.6.0 vs spec/routemaster/api_client_spec.rb in routemaster-drain-3.6.1

@@ -53,9 +53,35 @@
       subject.status
       assert_requested(:get, /example/) do |req|
         expect(req.headers).to include('X-Custom-Header')
       end
     end
+
+    describe 'source peer' do
+      context 'with a source peer' do
+        let(:options) { { source_peer: 'ServiceA' } }
+        let(:fetcher) { described_class.new(options) }
+
+        it 'should set the user agent header to the source peer' do
+          subject.status
+          assert_requested(:get, /example/) do |req|
+            expect(req.headers).to include('User-Agent' => 'ServiceA')
+          end
+        end
+      end
+
+      context 'without a source peer' do
+        let(:options) { { source_peer: nil } }
+        let(:fetcher) { described_class.new(options) }
+
+        it 'should set the user agent header to the faraday version' do
+          subject.status
+          assert_requested(:get, /example/) do |req|
+            expect(req.headers).to include('User-Agent' => "Faraday v#{Faraday::VERSION}" )
+          end
+        end
+      end
+    end
   end
 
   shared_examples 'a wrappable response' do
     context 'when response_class is present' do
       before do