Split the methods in two.
Fixed the mock to test inside callbacks.
Added a check to make sure that the where clause gets called.