almost finished unit tests / integration tests