Removed usages of non existent mock method assert_called and replaced with tests on called attribute