Add basic unit tests

As of yet, we only have integration tests.