[pluginize-solr] Moving a test and defining default value for extra_index variable

Rodrigo Souto
1 parent be886766
Showing 3 changed files with 7 additions and 6 deletions Show diff stats
plugins/solr/lib/ext/profile.rb
plugins/solr/test/unit/category_test.rb
test/unit/category_test.rb
@@ -41,6 +41,7 @@ class Profile
   end
  
   def self.solr_plugin_extra_data_for_index(sym = nil, &block)
+    self.solr_plugin_extra_index_methods ||= []
     self.solr_plugin_extra_index_methods.push(sym) if sym
     self.solr_plugin_extra_index_methods.push(block) if block_given?
   end
@@ -40,4 +40,10 @@ class CategoryTest &lt; ActiveSupport::TestCase
     c_name = Category.create!(:name => "science fiction", :environment_id => environment.id)
     assert_equal [c_name, c_abbr], Category.find_by_contents("science")[:results].docs
   end
+
+  should 'solr save' do
+    c = environment.categories.build(:name => 'my category');
+    c.expects(:solr_save)
+    c.save!
+  end
 end
@@ -513,10 +513,4 @@ class CategoryTest &lt; ActiveSupport::TestCase
     assert p.categories.reload.empty?
   end
  
-  should 'solr save' do
-    c = @env.categories.build(:name => 'my category');
-    c.expects(:solr_save)
-    c.save!
-  end
-
 end
...	...	@@ -41,6 +41,7 @@ class Profile
41	41	end
42	42
43	43	def self.solr_plugin_extra_data_for_index(sym = nil, &block)
	44	+ self.solr_plugin_extra_index_methods \|\|= []
44	45	self.solr_plugin_extra_index_methods.push(sym) if sym
45	46	self.solr_plugin_extra_index_methods.push(block) if block_given?
46	47	end
...	...
...	...	@@ -40,4 +40,10 @@ class CategoryTest < ActiveSupport::TestCase
40	40	c_name = Category.create!(:name => "science fiction", :environment_id => environment.id)
41	41	assert_equal [c_name, c_abbr], Category.find_by_contents("science")[:results].docs
42	42	end
	43	+
	44	+ should 'solr save' do
	45	+ c = environment.categories.build(:name => 'my category');
	46	+ c.expects(:solr_save)
	47	+ c.save!
	48	+ end
43	49	end
...	...
...	...	@@ -513,10 +513,4 @@ class CategoryTest < ActiveSupport::TestCase
513	513	assert p.categories.reload.empty?
514	514	end
515	515
516		- should 'solr save' do
517		- c = @env.categories.build(:name => 'my category');
518		- c.expects(:solr_save)
519		- c.save!
520		- end
521		-
522	516	end
...	...