Added Unit Test using JustMock