Updated unit testing - checking for status codes