Added tests for rake task