Right now you have a mixture of unit and integration tests. That's not good. It would be ❤ if you create unit tests for each class.