Skip to content

Commit

Permalink
Add RemoteComputeHelper and integration tests
Browse files Browse the repository at this point in the history
  • Loading branch information
mziccard committed Mar 1, 2016
1 parent 8073b20 commit 3af1497
Show file tree
Hide file tree
Showing 3 changed files with 526 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,117 @@
/*
* Copyright 2016 Google Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

package com.google.gcloud.compute.testing;

import com.google.gcloud.AuthCredentials;
import com.google.gcloud.RetryParams;
import com.google.gcloud.compute.ComputeOptions;

import java.io.IOException;
import java.io.InputStream;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
* Utility to create a remote Compute configuration for testing. Compute options can be obtained
* via the {@link #options()} method. Returned options have custom
* {@link ComputeOptions#retryParams()}: {@link RetryParams#retryMaxAttempts()} is {@code 10},
* {@link RetryParams#retryMinAttempts()} is {@code 6}, {@link RetryParams#maxRetryDelayMillis()} is
* {@code 30000}, {@link RetryParams#totalRetryPeriodMillis()} is {@code 120000} and
* {@link RetryParams#initialRetryDelayMillis()} is {@code 250}.
* {@link ComputeOptions#connectTimeout()} and {@link ComputeOptions#readTimeout()} are both set to
* {@code 60000}.
*/
public class RemoteComputeHelper {

private static final Logger log = Logger.getLogger(RemoteComputeHelper.class.getName());
private final ComputeOptions options;

private RemoteComputeHelper(ComputeOptions options) {
this.options = options;
}

/**
* Returns a {@link ComputeOptions} object to be used for testing.
*/
public ComputeOptions options() {
return options;
}

/**
* Creates a {@code RemoteComputeHelper} object for the given project id and JSON key input
* stream.
*
* @param projectId id of the project to be used for running the tests
* @param keyStream input stream for a JSON key
* @return A {@code RemoteComputeHelper} object for the provided options
* @throws ComputeHelperException if {@code keyStream} is not a valid JSON key stream
*/
public static RemoteComputeHelper create(String projectId, InputStream keyStream)
throws ComputeHelperException {
try {
ComputeOptions computeOptions = ComputeOptions.builder()
.authCredentials(AuthCredentials.createForJson(keyStream))
.projectId(projectId)
.retryParams(retryParams())
.connectTimeout(60000)
.readTimeout(60000)
.build();
return new RemoteComputeHelper(computeOptions);
} catch (IOException ex) {
if (log.isLoggable(Level.WARNING)) {
log.log(Level.WARNING, ex.getMessage());
}
throw ComputeHelperException.translate(ex);
}
}

/**
* Creates a {@code RemoteComputeHelper} object using default project id and authentication
* credentials.
*/
public static RemoteComputeHelper create() {
ComputeOptions computeOptions = ComputeOptions.builder()
.retryParams(retryParams())
.connectTimeout(60000)
.readTimeout(60000)
.build();
return new RemoteComputeHelper(computeOptions);
}

private static RetryParams retryParams() {
return RetryParams.builder()
.retryMaxAttempts(10)
.retryMinAttempts(6)
.maxRetryDelayMillis(30000)
.totalRetryPeriodMillis(120000)
.initialRetryDelayMillis(250)
.build();
}

public static class ComputeHelperException extends RuntimeException {

private static final long serialVersionUID = -5747977015007639912L;

public ComputeHelperException(String message, Throwable cause) {
super(message, cause);
}

public static ComputeHelperException translate(Exception ex) {
return new ComputeHelperException(ex.getMessage(), ex);
}
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
/*
* Copyright 2016 Google Inc. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/

/**
* A testing helper for Google Compute Engine.
*
* <p>A simple usage example:
*
* <p>Before the test:
* <pre> {@code
* RemoteComputeHelper computeHelper = RemoteComputeHelper.create();
* Compute compute = computeHelper.options().service();
* } </pre>
*
* @see <a href="https://github.com/GoogleCloudPlatform/gcloud-java/blob/master/TESTING.md#testing-code-that-uses-compute">
* gcloud-java tools for testing</a>
*/
package com.google.gcloud.compute.testing;
Loading

0 comments on commit 3af1497

Please sign in to comment.