Test return value of method with callbacks

This commit adds an expectation for the method that has all the
callbacks defined on it to test that it's return value is correct.