Add association tests