Split up unit test and integration test