Add test mock class

Enough for today :D