Examples with TestContext com.vmware.xenon.common.test.TestContext used on opensource projects

Example 6 with TestContext

use of com.vmware.xenon.common.test.TestContext in project photon-model by vmware.

the class BaseTestCase method getDocument.

@SuppressWarnings("unchecked")
protected <T> T getDocument(Class<T> type, URI uri) throws Throwable {
    TestContext ctx = testCreate(1);
    Object[] result = new Object[1];
    Operation get = Operation.createGet(uri).addPragmaDirective(Operation.PRAGMA_DIRECTIVE_QUEUE_FOR_SERVICE_AVAILABILITY).setReferer(this.host.getReferer()).setCompletion((o, e) -> {
        if (e != null) {
            this.host.log("Can't load document %s. Error: %s", uri, Utils.toString(e));
            ctx.failIteration(e);
        } else {
            result[0] = o.getBody(type);
            ctx.completeIteration();
        }
    });
    this.host.send(get);
    ctx.await();
    return (T) result[0];
}

Example 7 with TestContext

use of com.vmware.xenon.common.test.TestContext in project photon-model by vmware.

the class BaseModelTest method patchServiceSynchronously.

public <T extends ServiceDocument> int patchServiceSynchronously(String serviceUri, ComputeInstanceRequest patchBody) throws Throwable {
    TestContext ctx = this.host.testCreate(1);
    Operation patchOperation = Operation.createPatch(UriUtils.buildUri(this.host, serviceUri)).setBody(patchBody).setCompletion(ctx.getCompletion());
    this.host.send(patchOperation);
    this.testWait(ctx);
    return patchOperation.getStatusCode();
}

Example 8 with TestContext

use of com.vmware.xenon.common.test.TestContext in project photon-model by vmware.

the class BaseModelTest method waitForServiceAvailability.

protected void waitForServiceAvailability(ServiceHost h, String... serviceLinks) throws Throwable {
    if (serviceLinks == null || serviceLinks.length == 0) {
        throw new IllegalArgumentException("null or empty serviceLinks");
    }
    TestContext ctx = testCreate(serviceLinks.length);
    h.registerForServiceAvailability(ctx.getCompletion(), serviceLinks);
    ctx.await();
}

Example 9 with TestContext

use of com.vmware.xenon.common.test.TestContext in project photon-model by vmware.

the class BaseModelTest method getServiceSynchronously.

public <T extends ServiceDocument> T getServiceSynchronously(String serviceUri, Class<T> type) throws Throwable {
    AtomicReference<T> responseBody = new AtomicReference<>();
    TestContext ctx = this.host.testCreate(1);
    Operation getOperation = Operation.createGet(UriUtils.buildUri(this.host, serviceUri)).addPragmaDirective(Operation.PRAGMA_DIRECTIVE_QUEUE_FOR_SERVICE_AVAILABILITY).setCompletion((operation, throwable) -> {
        if (throwable != null) {
            ctx.failIteration(throwable);
            return;
        }
        responseBody.set(operation.getBody(type));
        ctx.completeIteration();
    });
    this.send(getOperation);
    this.testWait(ctx);
    return responseBody.get();
}

Example 10 with TestContext

use of com.vmware.xenon.common.test.TestContext in project photon-model by vmware.

the class BaseModelTest method deleteServiceSynchronously.

private <T extends ServiceDocument> void deleteServiceSynchronously(String serviceUri, boolean stopOnly) throws Throwable {
    TestContext ctx = this.host.testCreate(1);
    Operation deleteOperation = Operation.createDelete(UriUtils.buildUri(this.host, serviceUri)).setCompletion((operation, throwable) -> {
        if (throwable != null) {
            ctx.failIteration(throwable);
        }
        ctx.completeIteration();
    });
    if (stopOnly) {
        deleteOperation.addPragmaDirective(Operation.PRAGMA_DIRECTIVE_NO_INDEX_UPDATE);
    }
    this.send(deleteOperation);
    this.testWait(ctx);
}