Replaced a huge test with the new mock method, and it makes me smile