Added tests for associations