More unit tests for business logic