Make usage content spec the same for CE and EE

This changes the spec in question so that it is the same for both CE and
EE. This requires an `if Gitlab.ee?` since EE requires that we stub an
EE specific module, but doing so in CE will break the spec there.
This commit is contained in:
Yorick Peterse 2019-07-25 16:54:02 +02:00
parent 3d05243810
commit 3fa5ee8907
No known key found for this signature in database
GPG Key ID: EDD30D2BEB691AC9
1 changed files with 9 additions and 1 deletions

View File

@ -8,7 +8,15 @@ describe 'Usage stats consent' do
let(:message) { 'To help improve GitLab, we would like to periodically collect usage information.' }
before do
allow(user).to receive(:has_current_license?).and_return false
if Gitlab.ee?
allow_any_instance_of(EE::User)
.to receive(:has_current_license?)
.and_return(false)
else
allow(user)
.to receive(:has_current_license?)
.and_return(false)
end
gitlab_sign_in(user)
end