gistfile1.scala

package com.twitter.service.cachet.test.unit

import com.twitter.service.cachet._
import org.joda.time.DateTime
import javax.servlet.http._

import org.specs._
import org.specs.mock._
import org.specs.mock.JMocker._
import com.twitter.service.cachet.test.mock._

object ResponseSpec extends Specification with JMocker {
  "ResponseWrapper" should {
    "implement addDateHeader(x, y) such that" >> {
      var response: HttpServletResponse = null
      var responseWrapper: ResponseWrapper = null
      val millis = new DateTime().getMillis

      doBefore {
        response = mock(classOf[HttpServletResponse])
        responseWrapper = new ResponseWrapper(response)
      }

      "it delegates to the response" >> {
        expect { one(response).addDateHeader("Date", millis) }
        responseWrapper.addDateHeader("Date", millis)
      }

      "getDateHeader(x) returns y" >> {
        responseWrapper.addDateHeader("Date", millis)
        responseWrapper.getDateHeader("Date") must be(millis)
      }
    }
  }
}