Fix uses of Mock.assert_called_once in unit tests (fixes OPS-2077)

    make lint test