Updated tests to follow the changed way of handling authentication context.