spec/cinch-wikipedia_spec.rb in cinch-wikipedia-1.0.0 vs spec/cinch-wikipedia_spec.rb in cinch-wikipedia-1.0.1

@@ -1,41 +1,42 @@
 require 'spec_helper'
 
 describe Cinch::Plugins::Wikipedia do
+  include Cinch::Test
 
   before(:all) do
-    @plugin = Cinch::Plugins::Wikipedia.new
+    @bot = make_bot(Cinch::Plugins::Wikipedia)
   end
 
   # normal queries
   it 'should return a definition of a term' do
-    @plugin.send(:get_def, 'computer').
+    get_replies(make_message(@bot, '!wiki computer')).first.text.
       should include("A computer is a general purpose device")
   end
 
   it 'should not return multiple lined definitions' do
-    @plugin.send(:get_def, 'Teenager').
+    get_replies(make_message(@bot, '!wiki Teenager')).first.text.
       should_not include("\n")
   end
 
   it 'should not return definitions that are longer than 250 chars' do
-    @plugin.send(:get_def, 'Teenager').length.
+    get_replies(make_message(@bot, '!wiki Teenager')).first.text.length.
       should == 334
   end
 
   # Not found
   it 'should return an error message when a term is not found' do
-    @plugin.send(:get_def, 'dasdafasfasfasfasafsdfsdfsadf').
+    get_replies(make_message(@bot, '!wiki dasdafasfasfasfasafsdfsdfsadf')).first.text.
       should include("I couldn't find anything for that search, sorry!")
   end
 
   it 'should provide suggestions if one is listed on the page' do
-    @plugin.send(:get_def, 'smegama').
+    get_replies(make_message(@bot, '!wiki smegama')).first.text.
       should include("I couldn't find anything for that search, did you mean 'smegma'?")
   end
 
   # disambiguation
   it 'should return helful information when a disambuation page' do
-    @plugin.send(:get_def, 'hipster').
+    get_replies(make_message(@bot, '!wiki hacker')).first.text.
       should include('is too vague and lead to a disambiguation page')
   end
 end