Use mock a lot better

Also, add missing tests in Base Transform