Added methods on stub behaviors

Now the methods work on stub.onCall(...) as well