Add rake test task