Setup tests and add some unit cases