Added tests

Test for trainer controller