Corrected the failing unit test and added 2 tests for the http stage