Converted the context class to be public so it will be available for tests
Added rest api tests