require 'spec_helper'

class UserSpec < MiniTest::Spec
  # User and Bully are skeleton classes defined in the dummy application for
  # the sole purpose of running these tests. They're in spec/dummy/models/ but
  # really, there's nothing there of note.
  
  describe User do
    describe "before_destroy filters" do
      before :each do
        Recommendable.redis.flushdb
        @user1 = Factory(:user)
        @user2 = Factory(:user)
        @movie1 = Factory(:movie)
        @movie2 = Factory(:movie)
      end

      it "should no longer be involved in similarities or predictions" do
        @user1.like @movie1
        @user1.like @movie2
        @user2.like @movie1

        @user1.send :update_similarities
        @user2.send :update_similarities
        @user2.send :update_recommendations

        similarity_set = @user2.send :similarity_set
        predictions_set = @user2.send :predictions_set_for, Movie
        @user2.destroy

        @user1.similar_raters.wont_include @user2 
        Recommendable.redis.get(similarity_set).must_be_nil
        Recommendable.redis.get(predictions_set).must_be_nil
      end
    end

    describe "that does not act_as_recommended_to" do
      before :each do
        @user = Factory(:bully)
        @movie = Factory(:movie)
      end
      
      it "should not be able to rate, ignore, or stash an item, and trying should not be enough to get Redis keys" do
        proc { @user.like(@movie) }.must_raise    NoMethodError
        proc { @user.dislike(@movie) }.must_raise NoMethodError
        proc { @user.ignore(@movie) }.must_raise  NoMethodError
        proc { @user.stash(@movie) }.must_raise  NoMethodError
        
        @movie.liked_by.must_be_empty
        @movie.disliked_by.must_be_empty
        
        proc { @user.create_recommended_to_sets }.must_raise NoMethodError
      
        proc { @user.likes_set_for(Movie) }.must_raise NoMethodError
        proc { @user.dislikes_set_for(Movie) }.must_raise NoMethodError
        
        Recommendable.redis.smembers("Bully:#{@user.id}:likes:Movie").must_be_empty
        Recommendable.redis.smembers("Bully:#{@user.id}:dislikes:Movie").must_be_empty
        Recommendable.redis.smembers("Bully:#{@user.id}:similarities").must_be_empty
        Recommendable.redis.smembers("Bully:#{@user.id}:predictions:Movie").must_be_empty
      end
    end
    
    describe "that does act_as_recommendable" do
      before :each do
        @user = Factory(:user)
        @movie = Factory(:movie)
      end

      after :each do
        Recommendable.redis.flushdb
      end

      it "should be able to like a recommendable item" do
        @user.like(@movie).must_equal true
        @user.send(:create_recommended_to_sets)
        Recommendable.redis.smembers("User:#{@user.id}:likes:Movie").must_include @movie.id.to_s
        @user.send(:destroy_recommended_to_sets)
      end

      it "should not dislike an item after liking it" do
        @user.dislike(@movie)
        @user.like(@movie).must_equal true
        @user.dislikes?(@movie).must_equal false
        @user.likes?(@movie).must_equal true

        @user.send(:create_recommended_to_sets)
        Recommendable.redis.smembers("User:#{@user.id}:dislikes:Movie").wont_include @movie.id.to_s
        @user.send(:destroy_recommended_to_sets)
      end

      it "should not be ignoring an item after liking it" do
        @user.ignore(@movie)
        @user.like(@movie).must_equal true
        @user.ignored?(@movie).must_equal false
        @user.likes?(@movie).must_equal true
      end

      it "should not have an item stashed after liking it" do
        @user.stash(@movie)
        @user.like(@movie).must_equal true
        @user.stashed?(@movie).must_equal false
        @user.likes?(@movie).must_equal true
      end

      it "should be able to dislike a recommendable item" do
        @user.dislike(@movie).must_equal true
        @user.send(:create_recommended_to_sets)
        Recommendable.redis.smembers("User:#{@user.id}:dislikes:Movie").must_include @movie.id.to_s
        @user.send(:destroy_recommended_to_sets)
      end

      it "should not like an item after disliking it" do
        @user.like(@movie)
        @user.dislike(@movie).must_equal true
        @user.likes?(@movie).must_equal false
        @user.dislikes?(@movie).must_equal true

        @user.send(:create_recommended_to_sets)
        Recommendable.redis.smembers("User:#{@user.id}:dislikes:Movie").must_include @movie.id.to_s
        Recommendable.redis.smembers("User:#{@user.id}:likes:Movie").wont_include @movie.id.to_s
        @user.send(:destroy_recommended_to_sets)
      end

      it "should not be ignoring an item after disliking it" do
        @user.ignore(@movie)
        @user.dislike(@movie).must_equal true
        @user.ignored?(@movie).must_equal false
        @user.dislikes?(@movie).must_equal true
      end

      it "should not have an item stashed after disliking it" do
        @user.stash(@movie)
        @user.dislike(@movie).must_equal true
        @user.stashed?(@movie).must_equal false
        @user.dislikes?(@movie).must_equal true
      end

      it "should be able to ignore a recommendable item" do
        @user.ignore(@movie).must_equal true
      end

      it "should not like an item after ignoring it" do
        @user.like(@movie)
        @user.ignore(@movie).must_equal true
        @user.likes?(@movie).must_equal false
        @user.ignored?(@movie).must_equal true
      end

      it "should not dislike an item after ignoring it" do
        @user.dislike(@movie)
        @user.ignore(@movie).must_equal true
        @user.dislikes?(@movie).must_equal false
        @user.ignored?(@movie).must_equal true
      end

      it "should not have an item stashed after ignoring it" do
        @user.stash(@movie)
        @user.ignore(@movie).must_equal true
        @user.stashed?(@movie).must_equal false
        @user.ignored?(@movie).must_equal true
      end

      it "should be able to stash a recommendable item" do
        @user.stash(@movie).must_equal true
      end

      it "should not stash a liked item" do
        @user.like(@movie)
        @user.stash(@movie).must_be_nil
        @user.likes?(@movie).must_equal true
        @user.stashed?(@movie).must_equal false
      end

      it "should not stash a disliked item" do
        @user.dislike(@movie)
        @user.stash(@movie).must_be_nil
        @user.dislikes?(@movie).must_equal true
        @user.stashed?(@movie).must_equal false
      end

      it "should not have an item ignored after stashing it" do
        @user.ignore(@movie)
        @user.stash(@movie).must_equal true
        @user.ignored?(@movie).must_equal false
        @user.stashed?(@movie).must_equal true
      end
      
      it "should not be able to rate or ignore an item that is not recommendable." do
        @cakephp = Factory(:php_framework)
        
        proc { @user.like(@cakephp) }.must_raise    Recommendable::UnrecommendableError
        proc { @user.dislike(@cakephp) }.must_raise Recommendable::UnrecommendableError
        proc { @user.ignore(@cakephp) }.must_raise  Recommendable::UnrecommendableError
        proc { @user.stash(@cakephp) }.must_raise  Recommendable::UnrecommendableError
        
        proc { @cakephp.liked_by }.must_raise    NoMethodError
        proc { @cakephp.disliked_by }.must_raise NoMethodError
        
        @user.send(:create_recommended_to_sets)
      
        Recommendable.redis.smembers("User:#{@user.id}:likes:PhpFramework").must_be_empty
        Recommendable.redis.smembers("User:#{@user.id}:dislikes:PhpFramework").must_be_empty
        Recommendable.redis.smembers("User:#{@user.id}:similarities").must_be_empty
        Recommendable.redis.smembers("User:#{@user.id}:predictions:Movie").must_be_empty
        
        @user.send(:destroy_recommended_to_sets)
      end
      
      it "should not freak out when re-rating, re-ignoring, or re-stashing items" do
        @user.like(@movie)
        @user.like(@movie).must_be_nil
        
        @user.dislike(@movie)
        @user.dislike(@movie).must_be_nil
        
        @user.ignore(@movie)
        @user.ignore(@movie).must_be_nil

        @user.stash(@movie)
        @user.stash(@movie).must_be_nil
      end

      it "should be able to unrate, unignore or unstash items" do
        @user.like(@movie)
        @user.unlike(@movie).must_equal true

        @user.dislike(@movie)
        @user.undislike(@movie).must_equal true
        
        @user.ignore(@movie)
        @user.unignore(@movie).must_equal true

        @user.stash(@movie)
        @user.unstash(@movie).must_equal true
      end
    end

    describe "while using finders" do
      it "should return ActiveRecord::Relations" do
        @user = Factory(:user)
        @movie1 = Factory(:movie)
        @movie2 = Factory(:movie)
        @movie3 = Factory(:movie)
        @movie4 = Factory(:movie)
        @movie5 = Factory(:movie)

        @user.like @movie1
        @user.like @movie2
        @user.dislike @movie3

        @user.liked_movies.must_be_instance_of ActiveRecord::Relation
        @user.liked_movies.where(:title => @movie1.title).must_include @movie1
        @user.liked_movies.where(:title => @movie2.title).wont_include @movie1
        @user.liked_movies.limit(1).size.must_equal 1
        @user.disliked_movies.where(:title => @movie5.title).must_be_empty
      end
    end
    
    describe "while getting recommendations" do
      before :each do
        @dave   = Factory(:user)
        @frank  = Factory(:user)
        @hal    = Factory(:user)
        @movie1 = Factory(:movie)
        @movie2 = Factory(:movie)
        @movie3 = Factory(:movie)
        @movie4 = Factory(:movie)
        @movie5 = Factory(:movie)
      end
      
      after :each do
        Recommendable.redis.del "User:#{@dave.id}:similarities"
        Recommendable.redis.del "User:#{@dave.id}:predictions:Movie"
        Recommendable.redis.del "User:#{@frank.id}:similarities"
        Recommendable.redis.del "User:#{@frank.id}:predictions:Movie"
      end

      it "should respect passed counts" do
        @dave.like @movie1
        @frank.like @movie1
        @frank.like @movie2
        @frank.like @movie3
        @frank.like @movie4
        @frank.like @movie5

        @dave.send :update_similarities and @dave.send :update_recommendations

        @dave.recommendations.size.must_equal 4
        @dave.recommendations(2).size.must_equal 2
      end

      it "should return an ActiveRecord::Relation when using the dynamic finder" do
        @dave.like @movie1
        @frank.like @movie1
        @frank.like @movie2
        @frank.like @movie3
        @frank.like @movie4
        @frank.like @movie5

        @dave.send :update_similarities and @dave.send :update_recommendations

        @dave.recommended_movies.limit(2).size.must_equal 2
        @dave.recommended_movies.where(:title => @movie3.title).must_include @movie3
        @dave.recommended_movies.where(:title => @movie3.title).wont_include @movie2
        @dave.recommended_movies.where(:title => @movie3.title).wont_include @movie4
        @dave.recommended_movies.where(:title => @movie3.title).wont_include @movie5
      end

      it "should have common likes with a friend" do
        @dave.like @movie1
        @dave.like @movie2
        @dave.like @movie4

        @frank.like @movie2
        @frank.like @movie3
        @frank.like @movie4

        @dave.liked_movies_in_common_with(@frank).must_include @movie2
        @dave.liked_movies_in_common_with(@frank).must_include @movie4
        @dave.liked_movies_in_common_with(@frank).wont_include @movie1
        @dave.liked_movies_in_common_with(@frank).wont_include @movie3
      end

      it "should have common dislikes with a friend" do
        @dave.dislike @movie1
        @dave.dislike @movie3
        @dave.like    @movie4

        @frank.dislike @movie2
        @frank.dislike @movie3
        @frank.dislike @movie4

        @dave.disliked_movies_in_common_with(@frank).wont_include @movie2
        @dave.disliked_movies_in_common_with(@frank).wont_include @movie4
        @dave.disliked_movies_in_common_with(@frank).wont_include @movie1
        @dave.disliked_movies_in_common_with(@frank).must_include @movie3
      end
      
      it "should get populated sorted sets for similarities and recommendations" do
        @dave.like(@movie1)
        @frank.like(@movie1)
        @frank.like(@movie2)
        @dave.send :update_similarities
        @dave.send :update_recommendations
        
        @dave.similar_raters.must_include @frank
        @dave.recommended_movies.must_include @movie2
      end

      it "should return only the number of specified recommendations" do
        @dave.like(@movie1)
        @frank.like(@movie1)
        @frank.like(@movie2)
        @frank.like(@movie3)
        @frank.like(@movie4)
        @dave.send :update_similarities
        @dave.send :update_recommendations

        @dave.recommendations(2).size.must_equal 2
        @dave.recommended_movies(1).size.must_equal 1
      end
      
      it "should order similar users by similarity" do
        @dave.like(@movie1)
        @dave.like(@movie2)
        @frank.like(@movie1)
        @frank.dislike(@movie2)
        @hal.like(@movie1)
        @hal.like(@movie2)
        
        # hal should be more similar to dave than frank
        @dave.send :update_similarities
        @dave.send :update_recommendations
        
        @dave.similar_raters.must_equal [@hal, @frank]
      end
      
      it "should correctly recommend an awesome movie or two. In the correct order, of course." do
        @dave.like(@movie1)
        @dave.like(@movie2)
        @frank.like(@movie1)
        @frank.like(@movie2)
        @frank.like(@movie4)
        @hal.like(@movie1)
        @hal.like(@movie3)
        
        @dave.send :update_similarities
        @dave.send :update_recommendations
        
        @dave.recommendations.must_equal [@movie4, @movie3]
      end

      it "should not recommend rated items" do
        @dave.like(@movie1)
        @dave.dislike(@movie2)
        @frank.like(@movie1)
        @frank.like(@movie2)
        @frank.like(@movie4)
        @hal.like(@movie1)
        @hal.like(@movie3)
        
        @dave.send :update_similarities
        @dave.send :update_recommendations
        
        @dave.recommendations.wont_include @movie1
        @dave.recommendations.wont_include @movie2
      end

      it "should not recommend ignored items" do
        @dave.like(@movie1)
        @dave.like(@movie2)
        @dave.ignore(@movie4)
        @frank.like(@movie1)
        @frank.like(@movie2)
        @frank.like(@movie4)
        @hal.like(@movie1)
        @hal.like(@movie3)
        
        @dave.send :update_similarities
        @dave.send :update_recommendations
        
        @dave.recommendations.wont_include @movie4
      end

      it "should not recommend stashed items" do
        @dave.like(@movie1)
        @dave.like(@movie2)
        @dave.stash(@movie4)
        @frank.like(@movie1)
        @frank.like(@movie2)
        @frank.like(@movie4)
        @hal.like(@movie1)
        @hal.like(@movie3)
        
        @dave.send :update_similarities
        @dave.send :update_recommendations
        
        @dave.recommendations.wont_include @movie4
      end
    end
  end
end