spec/routemaster/api_client_spec.rb in routemaster-drain-3.6.3 vs spec/routemaster/api_client_spec.rb in routemaster-drain-3.6.4

@@ -50,11 +50,12 @@
 
     it 'passes headers' do
       headers['x-custom-header'] = 'why do you even'
       subject.status
       assert_requested(:get, /example/) do |req|
-        expect(req.headers).to include('X-Custom-Header')
+        expect(req.headers.keys).to include('X-Custom-Header')
+        expect(req.headers['User-Agent']).to eql "RoutemasterDrain - Faraday v0.15.3"
       end
     end
 
     describe 'source peer' do
       context 'with a source peer' do
@@ -74,10 +75,10 @@
         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}" )
+            expect(req.headers).to include('User-Agent' => "RoutemasterDrain - Faraday v#{Faraday::VERSION}" )
           end
         end
       end
     end
   end