fix basically every mock in tests :X

So, uh, remember that whole `assert mock.called_once_with` vs `mock.assert_called_once_with` thing? Ummm, yeah. Looks like I found out about that after writing these tests