Updated the test cases

Tried using stub for testing but I need to learn more on how to use to
effectively.
So for now I wrote some known test cases