Make use of assert_called_with and assert_has_calls

Signed-off-by: Fernando Rocha <fernandogrd@gmail.com>