Refactor unit tests

Use Codeception to run unit tests