DNS Prefetching + Preconnect of assets_host (CDN Domain)

1 files changed, 33 insertions, 0 deletions

diff --git a/spec/views/layouts/_head.html.haml_spec.rb b/spec/views/layouts/_head.html.haml_spec.rb
index e8e6d2e7a75..9d1efcabb80 100644
--- a/spec/views/layouts/_head.html.haml_spec.rb
+++ b/spec/views/layouts/_head.html.haml_spec.rb
@@ -29,6 +29,39 @@ describe 'layouts/_head' do
     expect(rendered).to match(%{content="foo" http-equiv="refresh"})
   end
 
+  context 'when an asset_host is set and feature is activated in the config it will' do
+    let(:asset_host) { 'http://assets' }
+
+    before do
+      stub_feature_flags(asset_host_prefetch: true)
+      allow(ActionController::Base).to receive(:asset_host).and_return(asset_host)
+    end
+
+    it 'add a link dns-prefetch tag' do
+      render
+      expect(rendered).to match('<link href="http://assets" rel="dns-prefetch">')
+    end
+
+    it 'add a link preconnect tag' do
+      render
+      expect(rendered).to match('<link crossorigin="" href="http://assets" rel="preconnnect">')
+    end
+  end
+
+  context 'when an asset_host is set and feature is not activated in the config it will' do
+    let(:asset_host) { 'http://assets' }
+
+    before do
+      stub_feature_flags(asset_host_prefetch: false)
+      allow(ActionController::Base).to receive(:asset_host).and_return(asset_host)
+    end
+
+    it 'not add a link dns-prefetch tag' do
+      render
+      expect(rendered).not_to match('<link href="http://assets" rel="dns-prefetch">')
+    end
+  end
+
   def stub_helper_with_safe_string(method)
     allow_any_instance_of(PageLayoutHelper).to receive(method)
       .and_return(%q{foo" http-equiv="refresh}.html_safe)