Small refactor and syntax fixes.

2 files changed, 3 insertions, 0 deletions

diff --git a/spec/requests/api/oauth_tokens_spec.rb b/spec/requests/api/oauth_tokens_spec.rb
index dbe30f264dd..7e2cc50e591 100644
--- a/spec/requests/api/oauth_tokens_spec.rb
+++ b/spec/requests/api/oauth_tokens_spec.rb
@@ -11,6 +11,7 @@ describe API::API, api: true  do
     context 'when user has 2FA enabled' do
       it 'does not create an access token' do
         user = create(:user, :two_factor)
+
         request_oauth_token(user)
 
         expect(response).to have_http_status(401)
@@ -21,6 +22,7 @@ describe API::API, api: true  do
     context 'when user does not have 2FA enabled' do
       it 'creates an access token' do
         user = create(:user)
+
         request_oauth_token(user)
 
         expect(response).to have_http_status(200)
diff --git a/spec/requests/api/session_spec.rb b/spec/requests/api/session_spec.rb
index 09f9192e7a8..acad1365ace 100644
--- a/spec/requests/api/session_spec.rb
+++ b/spec/requests/api/session_spec.rb
@@ -25,6 +25,7 @@ describe API::API, api: true  do
           post api('/session'), email: user.email, password: user.password
 
           expect(response).to have_http_status(401)
+          expect(response.body).to include('You have 2FA enabled.')
         end
       end
     end