如何在rspec中调用模块方法,并设置当前模块

时间:2015-04-11 19:29:52

标签: ruby-on-rails ruby rspec module

所以我有以下模块来处理提取和保存推文。这一切都是在一个线程上完成的,我有兴趣在这个模块中测试两个方法:fetch_tweetssave_tweets

对于fetch_tweets,我想验证是否调用了save_tweets方法。我不在乎它的回归。

对于save_tweets,我想测试一条推文实际上已保存。

所以考虑到这一点,让我们看看一些代码。

Twitter :: TwitterHandler - 它位于lib/twitter/twitter_handler.rb

module Twitter
  module TwitterHandler

    def handle_tweets(twitter_client_object)
      Thread.new do
        begin
          twitter_lock(twitter_client_object)
        ensure
          ActiveRecord::Base.clear_active_connections!
        end
      end
    end

    def twitter_lock(twitter_client_object)
      TwitterTweet.with_advisory_lock('aisis-writer-tweets', 0) do
        fetch_tweets(twitter_client_object)
      end
    end

    def fetch_tweets(twitter_client_object)
      Rails.cache.fetch([:aisis_twitter_feed, Time.now.to_i/60/5], expires_in: 1.minute){
        twitter_client.user_timeline(user_id: '252157965', count: 100).map(&:as_json).select{ |h| h["text"].match(/aisiswriter/i) }.take(10).each do |tweet|
          save_tweets(tweet['id_str'], tweet['text'], tweet['created_at'])
        end
      }
    end

    def save_tweets(id, text, created_at)
      tweet = TwitterTweet.find_or_initialize_by(id: id)
      tweet.text = text
      tweet.created_at = created_at
      tweet.save
    end

  end
end

twitter_handler_spec.rb - 现存于spec / twitter / twitter_handler_spec.rb

require 'rails_helper'
require_relative '../../lib/twitter/twitter_handler'

describe 'TwitterHandler' do

  context "fetch tweets method" do
    it "should fetch some tweets" do
      t = Time.parse("01/01/2010 10:00")
      Time.should_receive(:now).and_return(t)
      expect(fetch_tweets(twitter_client)).to receive(:save_tweets).with('342', 'Something', t).and_return nil
    end
  end
end

当我运行bin/rspec spec/twitter/twitter_handler_spec.rb时,我得到:

Failures:

  1) TwitterHandler fetch tweets method should fetch some tweets
     Failure/Error: expect(fetch_tweets(twitter_client)).to receive(:save_tweets).with('342', 'Something', t).and_return nil
     NoMethodError:
       undefined method `fetch_tweets' for #<RSpec::ExampleGroups::TwitterHandler::FetchTweetsMethod:0x007ff1c724e300>
     # ./spec/twitter/twitter_handler_spec.rb:11:in `block (3 levels) in <top (required)>'

如何修复此问题,以便我可以在此模块中调用此方法和任何其他方法?

1 个答案:

答案 0 :(得分:1)

您无法调用模块的实例方法。您必须includeextend模块,然后调用包含或扩展模块的对象上的方法。您从rspec获得的错误是因为您在示例中调用fetch_tweets,但示例中未定义fetch_tweets

一旦解决了调用fetch_tweets的问题,那么你必须处理这样一个事实,即你expect(fetch_tweets(...))的期望是说fetch_tweets返回的对象将会收到:save_tweets的方法调用,这不是你想要的。根据您的描述,您希望设置一个期望,即包含或扩展的任何对象TwitterHandler将接收save_tweets然后您想要调用fetch_tweets

如果不清楚,我建议您阅读Ruby模块方法,然后阅读RSpec。