2013-08-18 9 views
7

Mi piacerebbe essere in grado di prendere in giro un valore di ritorno di getOrElse metodo in modo che restituisca ciò che è passato come argomento orElse chiamata per nome con ScalaMockCome schernire un argomento call-by-name (come getOrElse) usando ScalaMock?

trait ToBeMocked { 
    def getOrElse(arg: Int)(orElse: => String): String 
} 

ScalaMock deve essere utilizzato a causa di accattivarsi. L'esecuzione di questo:

class CallByNameMockSpec extends Specification with MockFactory { 

    trait ToBeMocked { 
    def getOrElse(arg: Int)(orElse: => String): String 
    } 

    "Mocking functions with call-by-name arguments" should { 
    "work" in { 
     val m = mock[ToBeMocked] 

     (m.getOrElse (_: Int)(_: String)).expects(101, *).onCall((x, y) => y) 

     m.getOrElse(101)("result") must beEqualTo("result") 
    } 
    } 

} 

genera un'eccezione:

[error]  ClassCastException:   
test.service.CallByNameMockSpec$$anonfun$1$$anonfun$apply$1$$anonfun$apply$3$$anonfun$apply$4 cannot be cast to java.lang.String (CallByNameMockSpec.scala:16) 

risposta

5

Il trucco è quello di utilizzare Product versione di onCall metodo, convertire il secondo argomento di () => String e chiamarlo:

class CallByNameMockSpec extends Specification with MockFactory { 

    trait ToBeMocked { 
    def getOrElse(arg: Int)(orElse: => String): String 
    } 

    "Mocking functions with call-by-name arguments" should { 
    "work" in { 
     val m = mock[ToBeMocked] 

     (m.getOrElse (_: Int)(_: String)).expects(101, *).onCall(_.productElement(1).asInstanceOf[() => String]()) 

     m.getOrElse(101)("result") must beEqualTo("result") 
    } 
    } 

}