1 files changed, 59 insertions, 1 deletions
diff --git a/spec/lib/version_check_spec.rb b/spec/lib/version_check_spec.rb
index 23c381e241e..d7a772a3f7e 100644
--- a/spec/lib/version_check_spec.rb
+++ b/spec/lib/version_check_spec.rb
@@ -3,9 +3,67 @@
 require 'spec_helper'
 
 RSpec.describe VersionCheck do
+  describe '.image_url' do
+    it 'returns the correct URL' do
+      expect(described_class.image_url).to match(%r{\A#{Regexp.escape(described_class.host)}/check\.svg\?gitlab_info=\w+})
+    end
+  end
+
   describe '.url' do
     it 'returns the correct URL' do
-      expect(described_class.url).to match(%r{\A#{Regexp.escape(described_class.host)}/check\.svg\?gitlab_info=\w+})
+      expect(described_class.url).to match(%r{\A#{Regexp.escape(described_class.host)}/check\.json\?gitlab_info=\w+})
+    end
+  end
+
+  describe '#calculate_reactive_cache' do
+    context 'response code is 200' do
+      before do
+        stub_request(:get, described_class.url).to_return(status: 200, body: '{ "status": "success" }', headers: {})
+      end
+
+      it 'returns the response object' do
+        expect(described_class.new.calculate_reactive_cache).to eq("{ \"status\": \"success\" }")
+      end
+    end
+
+    context 'response code is not 200' do
+      before do
+        stub_request(:get, described_class.url).to_return(status: 500, body: nil, headers: {})
+      end
+
+      it 'returns nil' do
+        expect(described_class.new.calculate_reactive_cache).to be(nil)
+      end
+    end
+  end
+
+  describe '#response' do
+    context 'cache returns value' do
+      let(:response) { { "severity" => "success" }.to_json }
+
+      before do
+        allow_next_instance_of(described_class) do |instance|
+          allow(instance).to receive(:with_reactive_cache).and_return(response)
+        end
+      end
+
+      it 'returns the response object' do
+        expect(described_class.new.response).to be(response)
+      end
+    end
+
+    context 'cache returns nil' do
+      let(:response) { nil }
+
+      before do
+        allow_next_instance_of(described_class) do |instance|
+          allow(instance).to receive(:with_reactive_cache).and_return(response)
+        end
+      end
+
+      it 'returns nil' do
+        expect(described_class.new.response).to be(nil)
+      end
     end
   end
 end