Complete unit test