add tests for associations