danbooru/test/unit/artist_test.rb

require_relative '../test_helper'

class ArtistTest < ActiveSupport::TestCase
  context "An artist" do
    setup do
      user = Factory.create(:user)
      CurrentUser.user = user
      CurrentUser.ip_addr = "127.0.0.1"
      MEMCACHE.flush_all
    end

    teardown do
      CurrentUser.user = nil
      CurrentUser.ip_addr = nil
    end
    
    should "create a new wiki page to store any note information" do
      artist = nil
      assert_difference("WikiPage.count") do
        artist = Factory.create(:artist, :name => "aaa", :notes => "testing")
      end
      assert_equal("testing", artist.notes)
      assert_equal("testing", artist.wiki_page.body)
      assert_equal(artist.name, artist.wiki_page.title)
    end
    
    should "update the wiki page when notes are assigned" do
      artist = Factory.create(:artist, :name => "aaa", :notes => "testing")
      artist.update_attribute(:notes, "kokoko")
      artist.reload
      assert_equal("kokoko", artist.notes)
      assert_equal("kokoko", artist.wiki_page.body)
    end
    
    should "normalize its name" do
      artist = Factory.create(:artist, :name => "  AAA BBB  ")
      assert_equal("aaa_bbb", artist.name)
    end
    
    should "resolve ambiguous urls" do
      bobross = Factory.create(:artist, :name => "bob_ross", :url_string => "http://artists.com/bobross/image.jpg")
      bob = Factory.create(:artist, :name => "bob", :url_string => "http://artists.com/bob/image.jpg")
      matches = Artist.find_all_by_url("http://artists.com/bob/test.jpg")
      assert_equal(1, matches.size)
      assert_equal("bob", matches.first.name)
    end
    
    should "parse urls" do
      artist = Factory.create(:artist, :name => "rembrandt", :url_string => "http://rembrandt.com/test.jpg http://aaa.com")
      artist.reload
      assert_equal(["http://aaa.com", "http://rembrandt.com/test.jpg"], artist.urls.map(&:to_s).sort)
    end
    
    should "make sure old urls are deleted" do
      artist = Factory.create(:artist, :name => "rembrandt", :url_string => "http://rembrandt.com/test.jpg")
      artist.url_string = "http://not.rembrandt.com/test.jpg"
      artist.save
      artist.reload
      assert_equal(["http://not.rembrandt.com/test.jpg"], artist.urls.map(&:to_s).sort)
    end

    should "find matches by url" do
      a1 = Factory.create(:artist, :name => "rembrandt", :url_string => "http://rembrandt.com/test.jpg")
      a2 = Factory.create(:artist, :name => "subway", :url_string => "http://subway.com/test.jpg")

      assert_equal(["rembrandt"], Artist.find_all_by_url("http://rembrandt.com/test.jpg").map(&:name))
      assert_equal(["rembrandt"], Artist.find_all_by_url("http://rembrandt.com/another.jpg").map(&:name))    
      assert_equal([], Artist.find_all_by_url("http://nonexistent.com/test.jpg").map(&:name))
    end
    
    should "not allow duplicates" do
      Factory.create(:artist, :name => "warhol", :url_string => "http://warhol.com/a/image.jpg\nhttp://warhol.com/b/image.jpg")
      assert_equal(["warhol"], Artist.find_all_by_url("http://warhol.com/test.jpg").map(&:name))
    end
    
    should "hide deleted artists" do
      Factory.create(:artist, :name => "warhol", :url_string => "http://warhol.com/a/image.jpg", :is_active => false)
      assert_equal([], Artist.find_all_by_url("http://warhol.com/a/image.jpg").map(&:name))
    end
    
    should "normalize its other names" do
      artist = Factory.create(:artist, :name => "a1", :other_names => "aaa, bbb, ccc ddd")
      assert_equal("aaa bbb ccc_ddd", artist.other_names)
    end
    
    should "search on other names should return matches" do
      artist = Factory.create(:artist, :name => "artist", :other_names => "aaa, ccc ddd")
      assert_not_nil(Artist.find_by_any_name("name:artist"))
      assert_nil(Artist.find_by_any_name("name:aaa"))
      assert_nil(Artist.find_by_any_name("name:ccc_ddd"))
      assert_nil(Artist.find_by_any_name("other:artist"))
      assert_not_nil(Artist.find_by_any_name("other:aaa"))
      assert_not_nil(Artist.find_by_any_name("other:ccc_ddd"))
    end
    
    should "search on group name and return matches" do
      cat_or_fish = Factory.create(:artist, :name => "cat_or_fish")
      yuu = Factory.create(:artist, :name => "yuu", :group_name => "cat_or_fish")
      cat_or_fish.reload
      assert_equal("yuu", cat_or_fish.member_names)
      assert_not_nil(Artist.find_by_any_name("group:cat_or_fish"))
    end
    
    should "have an associated wiki" do
      user = Factory.create(:user)
      artist = Factory.create(:artist, :name => "max", :wiki_page_attributes => {:body => "this is max"})
      assert_not_nil(artist.wiki_page)
      assert_equal("this is max", artist.wiki_page.body)
    
      artist.update_attributes(:wiki_page_attributes => {:id => artist.wiki_page.id, :body => "this is hoge mark ii", :creator_id => user.id})
      assert_equal("this is hoge mark ii", artist.wiki_page(true).body)
    end
    
    should "revert to prior versions" do
      user = Factory.create(:user)
      reverter = Factory.create(:user)
      artist = nil
      assert_difference("ArtistVersion.count") do
        artist = Factory.create(:artist, :other_names => "yyy")
      end
      
      assert_difference("ArtistVersion.count") do
        artist.other_names = "xxx"
        artist.save
      end
      
      first_version = ArtistVersion.first
      assert_equal("yyy", first_version.other_names)
      artist.revert_to!(first_version)
      artist.reload
      assert_equal("yyy", artist.other_names)
    end
  end
end