93 lines
2.7 KiB
Ruby
93 lines
2.7 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'spec_helper'
|
|
|
|
RSpec.describe Users::ValidateManualOtpService, feature_category: :user_profile do
|
|
let_it_be(:user) { create(:user) }
|
|
|
|
let(:otp_code) { 42 }
|
|
|
|
subject(:validate) { described_class.new(user).execute(otp_code) }
|
|
|
|
context 'Devise' do
|
|
it 'calls Devise strategy' do
|
|
expect_next_instance_of(::Gitlab::Auth::Otp::Strategies::Devise) do |strategy|
|
|
expect(strategy).to receive(:validate).with(otp_code).once
|
|
end
|
|
|
|
validate
|
|
end
|
|
end
|
|
|
|
context 'FortiAuthenticator' do
|
|
before do
|
|
stub_feature_flags(forti_authenticator: user)
|
|
allow(::Gitlab.config.forti_authenticator).to receive(:enabled).and_return(true)
|
|
end
|
|
|
|
it 'calls ManualOtp strategy' do
|
|
expect_next_instance_of(::Gitlab::Auth::Otp::Strategies::FortiAuthenticator::ManualOtp) do |strategy|
|
|
expect(strategy).to receive(:validate).with(otp_code).once
|
|
end
|
|
|
|
validate
|
|
end
|
|
|
|
it 'handles unexpected error' do
|
|
error_message = "boom!"
|
|
|
|
expect_next_instance_of(::Gitlab::Auth::Otp::Strategies::FortiAuthenticator::ManualOtp) do |strategy|
|
|
expect(strategy).to receive(:validate).with(otp_code).once.and_raise(StandardError, error_message)
|
|
end
|
|
expect(Gitlab::ErrorTracking).to receive(:log_exception)
|
|
|
|
result = validate
|
|
|
|
expect(result[:status]).to eq(:error)
|
|
expect(result[:message]).to eq(error_message)
|
|
end
|
|
end
|
|
|
|
context 'FortiTokenCloud' do
|
|
before do
|
|
stub_feature_flags(forti_token_cloud: user)
|
|
allow(::Gitlab.config.forti_token_cloud).to receive(:enabled).and_return(true)
|
|
end
|
|
|
|
it 'calls FortiTokenCloud strategy' do
|
|
expect_next_instance_of(::Gitlab::Auth::Otp::Strategies::FortiTokenCloud) do |strategy|
|
|
expect(strategy).to receive(:validate).with(otp_code).once
|
|
end
|
|
|
|
validate
|
|
end
|
|
end
|
|
|
|
context 'DuoAuth' do
|
|
before do
|
|
allow(::Gitlab.config.duo_auth).to receive(:enabled).and_return(true)
|
|
end
|
|
|
|
it 'calls DuoAuth strategy' do
|
|
expect_next_instance_of(::Gitlab::Auth::Otp::Strategies::DuoAuth::ManualOtp) do |strategy|
|
|
expect(strategy).to receive(:validate).with(otp_code).once
|
|
end
|
|
|
|
validate
|
|
end
|
|
|
|
it "handles unexpected error" do
|
|
error_message = "boom!"
|
|
|
|
expect_next_instance_of(::Gitlab::Auth::Otp::Strategies::DuoAuth::ManualOtp) do |strategy|
|
|
expect(strategy).to receive(:validate).with(otp_code).once.and_raise(StandardError, error_message)
|
|
end
|
|
expect(Gitlab::ErrorTracking).to receive(:log_exception)
|
|
|
|
result = validate
|
|
|
|
expect(result[:status]).to eq(:error)
|
|
expect(result[:message]).to eq(error_message)
|
|
end
|
|
end
|
|
end
|