Added test with the mock