From 1dbe1fb1bc6c83a402a9f50142670ac2c4363452 Mon Sep 17 00:00:00 2001
From: Scott Albertson <scott@thoughtbot.com>
Date: Fri, 16 Aug 2013 11:40:30 -0700
Subject: [PATCH] Refactor User.find_by_username_or_email

* Improve test coverage
---
 app/models/user.rb       | 27 ++++++++----------
 spec/models/user_spec.rb | 60 +++++++++++++++++++++++++++++++++-------
 2 files changed, 61 insertions(+), 26 deletions(-)

diff --git a/app/models/user.rb b/app/models/user.rb
index f83c2edd17d..9278d90b9b9 100644
--- a/app/models/user.rb
+++ b/app/models/user.rb
@@ -127,23 +127,18 @@ class User < ActiveRecord::Base
   end
 
   def self.find_by_username_or_email(username_or_email)
-    lower_user = username_or_email.downcase
-    lower_email = Email.downcase(username_or_email)
-
-    users =
-      if username_or_email.include?('@')
-        User.where(email: lower_email)
-      else
-        User.where(username_lower: lower_user)
-      end
-        .to_a
-
-    if users.count > 1
-      raise Discourse::TooManyMatches
-    elsif users.count == 1
-      users[0]
+    conditions = if username_or_email.include?('@')
+      { email: Email.downcase(username_or_email) }
     else
-      nil
+      { username_lower: username_or_email.downcase }
+    end
+
+    users = User.where(conditions).all
+
+    if users.size > 1
+      raise Discourse::TooManyMatches
+    else
+      users.first
     end
   end
 
diff --git a/spec/models/user_spec.rb b/spec/models/user_spec.rb
index 9803528566c..971703399c4 100644
--- a/spec/models/user_spec.rb
+++ b/spec/models/user_spec.rb
@@ -850,19 +850,59 @@ describe User do
     end
   end
 
-  describe '#find_by_username_or_email' do
-    it 'works correctly' do
-      bob = Fabricate(:user, username: 'bob', name: 'bobs', email: 'bob@bob.com')
-      bob2 = Fabricate(:user, username: 'bob2', name: 'bobs', email: 'bob2@bob.com')
+  describe '.find_by_username_or_email' do
+    it 'finds user by username' do
+      bob = Fabricate(:user, username: 'bob')
 
-      expect(User.find_by_username_or_email('bob22@bob.com')).to eq(nil)
-      expect(User.find_by_username_or_email('bobs')).to eq(nil)
+      found_user = User.find_by_username_or_email('bob')
 
-      expect(User.find_by_username_or_email('bob2')).to eq(bob2)
-      expect(User.find_by_username_or_email('bob2@BOB.com')).to eq(bob2)
+      expect(found_user).to eq bob
+    end
 
-      expect(User.find_by_username_or_email('bob')).to eq(bob)
-      expect(User.find_by_username_or_email('bob@BOB.com')).to eq(bob)
+    it 'finds user by email' do
+      bob = Fabricate(:user, email: 'bob@example.com')
+
+      found_user = User.find_by_username_or_email('bob@example.com')
+
+      expect(found_user).to eq bob
+    end
+
+    context 'when user does not exist' do
+      it 'returns nil' do
+        found_user = User.find_by_username_or_email('doesnotexist@example.com') ||
+          User.find_by_username_or_email('doesnotexist')
+
+        expect(found_user).to be_nil
+      end
+    end
+
+    context 'when username case does not match' do
+      it 'finds user' do
+        bob = Fabricate(:user, username: 'bob')
+
+        found_user = User.find_by_username_or_email('Bob')
+
+        expect(found_user).to eq bob
+      end
+    end
+
+    context 'when email domain case does not match' do
+      it 'finds user' do
+        bob = Fabricate(:user, email: 'bob@example.com')
+
+        found_user = User.find_by_username_or_email('bob@Example.com')
+
+        expect(found_user).to eq bob
+      end
+    end
+
+    context 'when multiple users are found' do
+      it 'raises an exception' do
+        user_query = stub(all: [stub, stub])
+        User.stubs(:where).with(username_lower: 'bob').returns(user_query)
+
+        expect { User.find_by_username_or_email('bob') }.to raise_error(Discourse::TooManyMatches)
+      end
     end
   end
 end