Remove mock expected calls from edit handler tests

'Expected calls' works differently in Python 2 and 3