diff --git a/lib/travis/api/v3/queries/cron.rb b/lib/travis/api/v3/queries/cron.rb index 25c38191..80ac2108 100644 --- a/lib/travis/api/v3/queries/cron.rb +++ b/lib/travis/api/v3/queries/cron.rb @@ -8,5 +8,9 @@ module Travis::API::V3 return Models::Cron.find_by_id(id) if id raise WrongParams, 'missing cron.id'.freeze end + + def find_for_branch(branch) + branch.cron + end end end diff --git a/lib/travis/api/v3/queries/crons.rb b/lib/travis/api/v3/queries/crons.rb index 723bb7e5..77c5dc8a 100644 --- a/lib/travis/api/v3/queries/crons.rb +++ b/lib/travis/api/v3/queries/crons.rb @@ -1,10 +1,6 @@ module Travis::API::V3 class Queries::Crons < Query - def find(branch) - branch.cron - end - def start(branch) raise ServerError, 'repository does not have a github_id'.freeze unless branch.repository.github_id diff --git a/lib/travis/api/v3/services/cron/for_branch.rb b/lib/travis/api/v3/services/cron/for_branch.rb index 478e4b2c..0290e112 100644 --- a/lib/travis/api/v3/services/cron/for_branch.rb +++ b/lib/travis/api/v3/services/cron/for_branch.rb @@ -2,7 +2,7 @@ module Travis::API::V3 class Services::Cron::ForBranch < Service def run! - Models::Cron.where(:branch_id => find(:branch, find(:repository))).first + query.find_for_branch(find(:branch, find(:repository))) end end end