IntegrationTest+Webratで複数セッション扱う方法(仮)

実際に使っているコードより抜粋。

IntegrationTestの中に次の2つのメソッドを定義する。

159   def new_session(&block)
160     open_session do |sess|
161       webrat_session = ::Webrat.session_class.new(sess)
162       @_webrat_sessions.unshift webrat_session
163       yield sess
164       @_webrat_sessions.shift
165     end
166   end
167
168   def webrat_session
169     @_webrat_sessions ||= [::Webrat.session_class.new(self)]
170     @_webrat_sessions.first
171   end

使うときはこうする。

134   def test_easy_login
135     header('X_JPHONE_UID', "craccho")
136     header('USER_AGENT', "Vodafone")
137     visit "/itaku"
138     click_link "ログイン"
139     fill_in "ログインIDまたはメールアドレス", :with => "craccho"
140     fill_in "パスワード", :with => "asdfjkl;"
141     click_button "ログイン"
142     assert_match /ようこそ石倉さん/, response.body.toeuc
143
144     new_session do |sess|
145       visit "/itaku"
146       assert_match /ログイン/, sess.response.body.toeuc
147       header('X_JPHONE_UID', "craccho")
148       header('USER_AGENT', "Vodafone")
149       visit "/itaku"
150       assert_match /ようこそ石倉さん/, sess.response.body.toeuc
151       click_link "ログアウト"
152       assert_match /ログイン/, sess.response.body.toeuc
153     end
154
155     visit "/itaku"
156     assert_match /ようこそ石倉さん/, response.body.toeuc
157   end

test_easy_loginはテストメソッドで、その中で使ってるnew_sessionでwebrat用に新しいコンテキストを生成している。このメソッドに与えるブロック中では、webratが提供するメソッドはレシーバ無しでいいが、responseなどのIntegrationTestが提供するメソッドはブロックパラメータsessをレシーバにしないといけないというのがわかりづらい。webrat_sessionという非公開メソッドを再定義することで実現しているのでポータビリティにも欠けるが、これで一応うまく動いているようだ。公式にマルチセッションに対応してくれないかなぁ。
ちなみにIntegrationTestでwebratを使うには、gemでwebratいれてからtest/test_helper.rbに

require "webrat"

Webrat.configure do |config|
  config.mode = :rails
end

といれるだけでよいはず。

追記

上記だけではIntegrationTestでWebratは動かせないかもしれない。Railsが1.1.6だからだと思うが、test_helper.rbにさらに以下のコードが入っていた。

# for webrat
module ActionController
  class AbstractRequest
    def self.parse_query_parameters(query_string)
      return {} if query_string.blank?

      pairs = query_string.split('&').collect do |chunk|
        next if chunk.empty?
        key, value = chunk.split('=', 2)
        next if key.empty?
        value = value.nil? ? nil : CGI.unescape(value)
        [ CGI.unescape(key), value ]
      end.compact

      UrlEncodedPairParser.new(pairs).result
    end
  end

  class UrlEncodedPairParser < StringScanner #:nodoc:
    attr_reader :top, :parent, :result

    def initialize(pairs = [])
      super('')
      @result = {}
      pairs.each { |key, value| parse(key, value) }
    end

    KEY_REGEXP = %r{([^\[\]=&]+)}
    BRACKETED_KEY_REGEXP = %r{\[([^\[\]=&]+)\]}

    # Parse the query string
    def parse(key, value)
      self.string = key
      @top, @parent = result, nil

      # First scan the bare key
      key = scan(KEY_REGEXP) or return
      key = post_key_check(key)

      # Then scan as many nestings as present
      until eos?
        r = scan(BRACKETED_KEY_REGEXP) or return
        key = self[1]
        key = post_key_check(key)
      end

      bind(key, value)
    end

    private
      # After we see a key, we must look ahead to determine our next action. Cases:
      #
      #   [] follows the key. Then the value must be an array.
      #   = follows the key. (A value comes next)
      #   & or the end of string follows the key. Then the key is a flag.
      #   otherwise, a hash follows the key.
      def post_key_check(key)
        if scan(/\[\]/) # a[b][] indicates that b is an array
          container(key, Array)
          nil
        elsif check(/\[[^\]]/) # a[b] indicates that a is a hash
          container(key, Hash)
          nil
        else # End of key? We do nothing.
          key
        end
      end

      # Add a container to the stack.
      def container(key, klass)
        type_conflict! klass, top[key] if top.is_a?(Hash) && top.key?(key) && ! top[key].is_a?(klass)
        value = bind(key, klass.new)
        type_conflict! klass, value unless value.is_a?(klass)
        push(value)
      end

      # Push a value onto the 'stack', which is actually only the top 2 items.
      def push(value)
        @parent, @top = @top, value
      end

      # Bind a key (which may be nil for items in an array) to the provided value.
      def bind(key, value)
        if top.is_a? Array
          if key
            if top[-1].is_a?(Hash) && ! top[-1].key?(key)
              top[-1][key] = value
            else
              top << {key => value}.with_indifferent_access
              push top.last
              value = top[key]
            end
          else
            top << value
          end
        elsif top.is_a? Hash
          key = CGI.unescape(key)
          parent << (@top = {}) if top.key?(key) && parent.is_a?(Array)
          top[key] ||= value
          return top[key]
        else
          raise ArgumentError, "Don't know what to do: top is #{top.inspect}"
        end

        return value
      end

      def type_conflict!(klass, value)
        raise TypeError, "Conflicting types for parameter containers. Expected an instance of #{klass} but found an instance of #{value.class}. This can be caused by colliding Array and Hash parameters like qs[]=value&qs[key]=value. (The parameters received were #{value.inspect}.)"
      end
  end
end