Add rspec traits controller tests