From 98a75e593180dc08c464b3e3af000841d6eebc3a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Steffen=20Ko=CC=88tte?= Date: Mon, 1 Feb 2016 15:23:15 +0100 Subject: [PATCH] move find cron for branch to correct query --- lib/travis/api/v3/queries/cron.rb | 4 ++++ lib/travis/api/v3/queries/crons.rb | 4 ---- lib/travis/api/v3/services/cron/for_branch.rb | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) 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