fix tests

This commit is contained in:
r888888888
2017-03-15 16:08:47 -07:00
parent da06bee0ab
commit d9a26975ba
9 changed files with 25 additions and 33 deletions

View File

@@ -21,11 +21,11 @@ module Moderator
end
tags = Tag.scan_tags(antecedent, :strip_metatags => true)
conds = tags.map {|x| "tag_query like ?"}.join(" AND ")
conds = tags.map {|x| "query like ?"}.join(" AND ")
conds = [conds, *tags.map {|x| "%#{x.to_escaped_for_sql_like}%"}]
if SavedSearch.enabled?
SavedSearch.where(*conds).find_each do |ss|
ss.tag_query = (ss.tag_query_array - tags + [consequent]).uniq.join(" ")
ss.query = (ss.query_array - tags + [consequent]).uniq.join(" ")
ss.save
end
end

View File

@@ -182,8 +182,8 @@ class TagAlias < ActiveRecord::Base
escaped = Regexp.escape(antecedent_name)
if SavedSearch.enabled?
SavedSearch.where("tag_query like ?", "%#{antecedent_name}%").find_each do |ss|
ss.tag_query = ss.tag_query.sub(/(?:^| )#{escaped}(?:$| )/, " #{consequent_name} ").strip.gsub(/ /, " ")
SavedSearch.where("query like ?", "%#{antecedent_name}%").find_each do |ss|
ss.query = ss.query.sub(/(?:^| )#{escaped}(?:$| )/, " #{consequent_name} ").strip.gsub(/ /, " ")
ss.save
end
end

View File

@@ -1,7 +1,7 @@
FactoryGirl.define do
factory(:saved_search) do
tag_query { FFaker::Lorem.words }
category { FFaker::Lorem.word }
query { FFaker::Lorem.words }
labels { [FFaker::Lorem.word] }
user
end
end

View File

@@ -21,7 +21,7 @@ class SavedSearchesControllerTest < ActionController::TestCase
context "create action" do
should "render" do
params = { saved_search_tags: "bkub", saved_search_category: "artist" }
params = { saved_search_tags: "bkub", saved_search_labels: "artist" }
post :create, params, { user_id: @user.id }
assert_response :redirect
@@ -40,11 +40,11 @@ class SavedSearchesControllerTest < ActionController::TestCase
context "update action" do
should "render" do
saved_search = FactoryGirl.create(:saved_search, user: @user)
params = { id: saved_search.id, saved_search: { category: "foo" } }
params = { id: saved_search.id, saved_search: { label_string: "foo" } }
put :update, params, { user_id: @user.id }
assert_redirected_to saved_searches_path
assert_equal("foo", saved_search.reload.category)
assert_equal(["foo"], saved_search.reload.labels)
end
end

View File

@@ -16,7 +16,6 @@ module SavedSearchTestHelper
service = mock_sqs_service.new
SavedSearch.stubs(:sqs_service).returns(service)
SavedSearch.stubs(:update_listbooru_on_create)
Danbooru.config.stubs(:aws_sqs_saved_search_url).returns("http://localhost:3002")
Danbooru.config.stubs(:listbooru_auth_key).returns("blahblahblah")
Danbooru.config.stubs(:listbooru_server).returns("http://localhost:3001")

View File

@@ -31,12 +31,11 @@ module Moderator
end
should "move saved searches" do
ss = FactoryGirl.create(:saved_search, :user => @user, :tag_query => "123 ... 456")
ss = FactoryGirl.create(:saved_search, :user => @user, :query => "123 ... 456")
tag_batch_change = TagBatchChange.new("...", "bbb", @user.id, "127.0.0.1")
tag_batch_change.perform
ss.reload
assert_equal("123 456 bbb", ss.tag_query)
assert_equal(%w(create update), SavedSearch.sqs_service.commands)
assert_equal(%w(123 456 bbb), ss.query.scan(/\S+/).sort)
end
should "raise an error if there is no predicate" do

View File

@@ -1951,8 +1951,8 @@ class PostTest < ActiveSupport::TestCase
should "return posts for a search:<category> metatag" do
post1 = FactoryGirl.create(:post, tag_string: "aaa")
post2 = FactoryGirl.create(:post, tag_string: "bbb")
FactoryGirl.create(:saved_search, tag_query: "aaa", category: "zzz", user: CurrentUser.user)
FactoryGirl.create(:saved_search, tag_query: "bbb", category: nil, user: CurrentUser.user)
FactoryGirl.create(:saved_search, query: "aaa", labels: ["zzz"], user: CurrentUser.user)
FactoryGirl.create(:saved_search, query: "bbb", user: CurrentUser.user)
SavedSearch.expects(:post_ids).with(CurrentUser.id, "zzz").returns([post1.id])
SavedSearch.expects(:post_ids).with(CurrentUser.id, "uncategorized").returns([post2.id])

View File

@@ -19,9 +19,10 @@ class SavedSearchTest < ActiveSupport::TestCase
FakeWeb.clean_registry
end
context "with a name" do
context "with a label" do
setup do
FakeWeb.register_uri(:get, "http://localhost:3001/users?key=blahblahblah&user_id=1&name=blah", :body => [1,2,3,4].to_json)
SavedSearch.expects(:queries_for).with(1, "blah").returns(%w(a b c))
FakeWeb.register_uri(:post, "http://localhost:3001/v2/search", :body => "1 2 3 4")
end
should "return a list of ids" do
@@ -30,9 +31,10 @@ class SavedSearchTest < ActiveSupport::TestCase
end
end
context "without a name" do
context "without a label" do
setup do
FakeWeb.register_uri(:get, "http://localhost:3001/users?key=blahblahblah&user_id=1", :body => [1,2,3,4].to_json)
SavedSearch.expects(:queries_for).with(1, nil).returns(%w(a b c))
FakeWeb.register_uri(:post, "http://localhost:3001/v2/search", :body => "1 2 3 4")
end
should "return a list of ids" do
@@ -45,7 +47,7 @@ class SavedSearchTest < ActiveSupport::TestCase
context "Creating a saved search" do
setup do
@user = FactoryGirl.create(:gold_user)
@saved_search = @user.saved_searches.create(:tag_query => " xxx ")
@saved_search = @user.saved_searches.create(:query => " xxx ")
end
should "update the bitpref on the user" do
@@ -54,11 +56,7 @@ class SavedSearchTest < ActiveSupport::TestCase
end
should "normalize whitespace" do
assert_equal("xxx", @saved_search.tag_query)
end
should "send messages" do
assert_equal(%w(create), SavedSearch.sqs_service.commands)
assert_equal("xxx", @saved_search.query)
end
end
@@ -73,17 +71,13 @@ class SavedSearchTest < ActiveSupport::TestCase
@user.reload
assert(!@user.has_saved_searches?, "should not have the saved_searches bitpref set")
end
should "send messages" do
assert_equal(%w(create delete), SavedSearch.sqs_service.commands)
end
end
context "A user with max saved searches" do
setup do
@user = FactoryGirl.create(:gold_user)
User.any_instance.stubs(:max_saved_searches).returns(0)
@saved_search = @user.saved_searches.create(:tag_query => "xxx")
@saved_search = @user.saved_searches.create(:query => "xxx")
end
should "not be able to create another saved search" do

View File

@@ -74,13 +74,13 @@ class TagAliasTest < ActiveSupport::TestCase
assert_nil(Cache.get("ta:aaa"))
end
should "move saved searches" do
should "zzz move saved searches" do
tag1 = FactoryGirl.create(:tag, :name => "...")
tag2 = FactoryGirl.create(:tag, :name => "bbb")
ss = FactoryGirl.create(:saved_search, :tag_query => "123 ... 456", :user => CurrentUser.user)
ss = FactoryGirl.create(:saved_search, :query => "123 ... 456", :user => CurrentUser.user)
ta = FactoryGirl.create(:tag_alias, :antecedent_name => "...", :consequent_name => "bbb")
ss.reload
assert_equal("123 bbb 456", ss.tag_query)
assert_equal(%w(123 456 bbb), ss.query.scan(/\S+/).sort)
end
should "update any affected posts when saved" do