Skip to content

Commit 4db575c

Browse files
committed
Add tests to Service Job Controllers
1 parent 756ea85 commit 4db575c

File tree

5 files changed

+642
-6
lines changed

5 files changed

+642
-6
lines changed

server/src/main/java/org/apache/cloudstack/backup/BackupCompressionServiceJobController.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,7 @@ protected void searchAndDispatchJobs() {
105105
}
106106

107107
for (DataCenterVO zone : zones) {
108-
if (!BackupManager.BackupFrameworkEnabled.valueIn(zone.getId())) {
108+
if (!isFrameworkEnabledForZone(zone)) {
109109
logger.debug("Backup framework is not enabled for zone [{}], will not run the backup compression task for this zone.", zone.getUuid());
110110
continue;
111111
}
@@ -156,7 +156,7 @@ protected void submitQueuedJob(InternalBackupServiceJobVO job, long zoneId, Stri
156156
/**
157157
* Submit FinalizeCompression jobs, this should be called before submitStartJobsForExecution.
158158
* */
159-
private HashSet<Long> submitFinalizeJobsForExecution(List<InternalBackupServiceJobVO> jobsToExecute, List<Pair<HostVO, Long>> hostAndNumberOfJobsPairList, long zoneId) {
159+
protected HashSet<Long> submitFinalizeJobsForExecution(List<InternalBackupServiceJobVO> jobsToExecute, List<Pair<HostVO, Long>> hostAndNumberOfJobsPairList, long zoneId) {
160160
List<InternalBackupServiceJobVO> submittedJobs = new ArrayList<>();
161161
HashSet<Long> setOfInstancesWithExecutingCompressionJobs = new HashSet<>();
162162
for (InternalBackupServiceJobVO job : jobsToExecute) {
@@ -207,7 +207,7 @@ private void finalizeBackupCompression(InternalBackupServiceJobVO job, long zone
207207
}
208208
}
209209

210-
private List<InternalBackupServiceJobVO> filterJobsOfDomainsAndAccountsWithDisabledCompressionTask(List<InternalBackupServiceJobVO> jobsToFilter) {
210+
protected List<InternalBackupServiceJobVO> filterJobsOfDomainsAndAccountsWithDisabledCompressionTask(List<InternalBackupServiceJobVO> jobsToFilter) {
211211
ArrayList<InternalBackupServiceJobVO> filteredJobs = new ArrayList<>();
212212
for (InternalBackupServiceJobVO job : jobsToFilter) {
213213
if (backupCompressionTaskEnabled.valueIn(job.getAccountId())) {

server/src/main/java/org/apache/cloudstack/backup/BackupValidationServiceController.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,7 @@ protected void searchAndDispatchJobs() {
113113
}
114114

115115
for (DataCenterVO zone : zones) {
116-
if (!BackupManager.BackupFrameworkEnabled.valueIn(zone.getId())) {
116+
if (!isFrameworkEnabledForZone(zone)) {
117117
logger.debug("Backup framework is not enabled for zone [{}], will not run the backup validation task for this zone.", zone.getUuid());
118118
continue;
119119
}
@@ -179,7 +179,7 @@ private void startBackupValidation(InternalBackupServiceJobVO job, long zoneId,
179179
}
180180
}
181181

182-
private List<InternalBackupServiceJobVO> filterJobsOfDomainsAndAccountsWithDisabledValidationTask(List<InternalBackupServiceJobVO> jobsToFilter) {
182+
protected List<InternalBackupServiceJobVO> filterJobsOfDomainsAndAccountsWithDisabledValidationTask(List<InternalBackupServiceJobVO> jobsToFilter) {
183183
ArrayList<InternalBackupServiceJobVO> filteredJobs = new ArrayList<>();
184184
for (InternalBackupServiceJobVO job : jobsToFilter) {
185185
if (backupValidationTaskEnabled.valueIn(job.getAccountId())) {

server/src/main/java/org/apache/cloudstack/backup/InternalBackupServiceJobController.java

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ protected List<Pair<HostVO, Long>> filterHostsWithTooManyJobs(HashMap<HostVO, Lo
224224
return hostAndNumberOfJobsPairList;
225225
}
226226

227-
private Integer getMaxConcurrentCompressionsPerHost(ConfigKey<Integer> jobsPerHostConfiguration, HostVO host) {
227+
protected Integer getMaxConcurrentCompressionsPerHost(ConfigKey<Integer> jobsPerHostConfiguration, HostVO host) {
228228
if (host.getDetail(jobsPerHostConfiguration.key()) != null) {
229229
return Integer.valueOf(host.getDetail(jobsPerHostConfiguration.key()));
230230
} else {
@@ -271,6 +271,11 @@ protected void submitQueuedJobsForExecution(List<InternalBackupServiceJobVO> job
271271
}
272272
}
273273

274+
275+
protected Boolean isFrameworkEnabledForZone(DataCenterVO zone) {
276+
return BackupManager.BackupFrameworkEnabled.valueIn(zone.getId());
277+
}
278+
274279
protected void submitQueuedJob(InternalBackupServiceJobVO job, long zoneId, String logId) {
275280
}
276281

0 commit comments

Comments
 (0)