added some tests with mock