Adds tests and mock lib