Use unit tests instead of integration tests.

This provides a faster feedback cycle and the JUnit messages for failed
assertions are more meaningful.