Extract EE specific specs from environment_entity

ee/spec/serializers/environment_entity_spec.rb

+require 'spec_helper'
+
+describe EnvironmentEntity do
+  include KubernetesHelpers
+
+  let(:user) { create(:user) }
+  let(:environment) { create(:environment) }
+
+  before do
+    environment.project.add_maintainer(user)
+  end
+
+  let(:entity) do
+    described_class.new(environment, request: double(current_user: user))
+  end
+
+  describe '#as_json' do
+    subject { entity.as_json }
+
+    context 'when deploy_boards are available' do
+      before do
+        stub_licensed_features(deploy_board: true)
+      end
+
+      context 'with deployment service ready' do
+        before do
+          allow(environment).to receive(:has_terminals?).and_return(true)
+          allow(environment).to receive(:rollout_status).and_return(kube_deployment_rollout_status)
+        end
+
+        it 'exposes rollout_status' do
+          expect(subject).to include(:rollout_status)
+        end
+      end
+    end
+
+    context 'when deploy_boards are not available' do
+      before do
+        allow(environment).to receive(:has_terminals?).and_return(true)
+      end
+
+      it 'does not expose rollout_status' do
+        expect(subject).not_to include(:rollout_status)
+      end
+    end
+  end
+end

spec/serializers/environment_entity_spec.rb

 require 'spec_helper'
 
 describe EnvironmentEntity do
-  include KubernetesHelpers
-
-  let(:user) { create(:user) }
-  let(:environment) { create(:environment) }
-
+  let(:request) { double('request') }
   let(:entity) do
-    described_class.new(environment, request: double(current_user: user))
+    described_class.new(environment, request: spy('request'))
   end
 
+  let(:environment) { create(:environment) }
   subject { entity.as_json }
 
-  before do
-    environment.project.add_maintainer(user)
-  end
-
   it 'exposes latest deployment' do
     expect(subject).to include(:last_deployment)
   end
@@ -48,29 +41,6 @@ describe EnvironmentEntity do
     end
   end
 
-  context 'with deployment service ready' do
-    before do
-      stub_licensed_features(deploy_board: true)
-      allow(environment).to receive(:has_terminals?).and_return(true)
-      allow(environment).to receive(:rollout_status).and_return(kube_deployment_rollout_status)
-    end
-
-    it 'exposes rollout_status' do
-      expect(subject).to include(:rollout_status)
-    end
-  end
-
-  context 'when license does not has the GitLab_DeployBoard add-on' do
-    before do
-      stub_licensed_features(deploy_board: false)
-      allow(environment).to receive(:has_terminals?).and_return(true)
-    end
-
-    it 'does not expose rollout_status' do
-      expect(subject[:rollout_status_path]).to be_blank
-    end
-  end
-
   context 'with deployment platform' do
     let(:project) { create(:project, :repository) }
     let(:environment) { create(:environment, project: project) }