use of io.vertx.core.Context in project hono by eclipse.
the class FileBasedRegistrationServiceTest method testDoStartIgnoresMalformedJson.
/**
* Verifies that the registration service successfully starts up even if
* the file to read device information from contains malformed JSON.
*
* @param ctx The vert.x context.
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testDoStartIgnoresMalformedJson(final TestContext ctx) {
// GIVEN a registration service configured to read data from a file
// that contains malformed JSON
when(fileSystem.existsBlocking(props.getFilename())).thenReturn(Boolean.TRUE);
doAnswer(invocation -> {
final Buffer data = mock(Buffer.class);
when(data.getBytes()).thenReturn("NO JSON".getBytes(StandardCharsets.UTF_8));
Handler handler = invocation.getArgument(1);
handler.handle(Future.succeededFuture(data));
return null;
}).when(fileSystem).readFile(eq(props.getFilename()), any(Handler.class));
// WHEN starting the service
Async startup = ctx.async();
Future<Void> startupTracker = Future.future();
startupTracker.setHandler(ctx.asyncAssertSuccess(started -> {
startup.complete();
}));
registrationService.doStart(startupTracker);
// THEN startup succeeds
startup.await();
}
use of io.vertx.core.Context in project hono by eclipse.
the class FileBasedRegistrationServiceTest method testDoStartCreatesFile.
/**
* Verifies that the registration service creates a file for persisting device registration
* data if it does not exist yet during startup.
*
* @param ctx The vert.x context.
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testDoStartCreatesFile(final TestContext ctx) {
// GIVEN a registration service configured to persist data to a not yet existing file
props.setSaveToFile(true);
when(fileSystem.existsBlocking(props.getFilename())).thenReturn(Boolean.FALSE);
doAnswer(invocation -> {
Handler handler = invocation.getArgument(1);
handler.handle(Future.succeededFuture());
return null;
}).when(fileSystem).createFile(eq(props.getFilename()), any(Handler.class));
doAnswer(invocation -> {
Handler handler = invocation.getArgument(1);
handler.handle(Future.failedFuture("malformed file"));
return null;
}).when(fileSystem).readFile(eq(props.getFilename()), any(Handler.class));
// WHEN starting the service
Async startup = ctx.async();
Future<Void> startupTracker = Future.future();
startupTracker.setHandler(ctx.asyncAssertSuccess(started -> {
startup.complete();
}));
registrationService.doStart(startupTracker);
// THEN the file gets created
startup.await();
verify(fileSystem).createFile(eq(props.getFilename()), any(Handler.class));
}
use of io.vertx.core.Context in project hono by eclipse.
the class FileBasedRegistrationServiceTest method testDoStartLoadsDeviceIdentities.
/**
* Verifies that device identities are successfully loaded from file during startup.
*
* @param ctx The test context.
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testDoStartLoadsDeviceIdentities(final TestContext ctx) {
// GIVEN a service configured with a file name
when(fileSystem.existsBlocking(props.getFilename())).thenReturn(Boolean.TRUE);
doAnswer(invocation -> {
final Buffer data = DeviceRegistryTestUtils.readFile(FILE_NAME);
Handler handler = invocation.getArgument(1);
handler.handle(Future.succeededFuture(data));
return null;
}).when(fileSystem).readFile(eq(props.getFilename()), any(Handler.class));
// WHEN the service is started
Async startup = ctx.async();
Future<Void> startFuture = Future.future();
startFuture.setHandler(ctx.asyncAssertSuccess(s -> {
startup.complete();
}));
registrationService.doStart(startFuture);
// THEN the device identities from the file are loaded
startup.await();
registrationService.getDevice(TENANT, DEVICE, ctx.asyncAssertSuccess());
registrationService.getDevice(TENANT, "4712", ctx.asyncAssertSuccess(result -> {
final JsonObject data = result.getPayload().getJsonObject(RegistrationConstants.FIELD_DATA);
ctx.assertEquals(data.getString(FileBasedRegistrationService.PROPERTY_VIA), GW);
}));
registrationService.getDevice(TENANT, GW, ctx.asyncAssertSuccess());
}
use of io.vertx.core.Context in project hono by eclipse.
the class FileBasedRegistrationServiceTest method testContentsNotSavedOnShutdownIfSavingIfDisabled.
/**
* Verifies that setting the <em>saveToFile</em> configuration property to <em>false</em> prevents
* the registration service to write its content to the file system during shutdown.
*
* @param ctx The vert.x test context.
*/
@SuppressWarnings({ "unchecked", "rawtypes" })
@Test
public void testContentsNotSavedOnShutdownIfSavingIfDisabled(final TestContext ctx) {
// GIVEN a registration service configured to not persist data
props.setSaveToFile(false);
when(fileSystem.existsBlocking(props.getFilename())).thenReturn(Boolean.TRUE);
doAnswer(invocation -> {
Handler handler = invocation.getArgument(1);
handler.handle(Future.failedFuture("malformed data"));
return null;
}).when(fileSystem).readFile(eq(props.getFilename()), any(Handler.class));
Async startup = ctx.async();
Future<Void> startupTracker = Future.future();
startupTracker.setHandler(ctx.asyncAssertSuccess(started -> {
startup.complete();
}));
registrationService.doStart(startupTracker);
startup.await();
// WHEN adding a device
registrationService.addDevice(TENANT, DEVICE, new JsonObject());
// and shutting down the service
Async shutdown = ctx.async();
Future<Void> shutdownTracker = Future.future();
shutdownTracker.setHandler(ctx.asyncAssertSuccess(done -> {
shutdown.complete();
}));
registrationService.doStop(shutdownTracker);
// THEN no data has been written to the file system
shutdown.await();
verify(fileSystem, never()).createFile(eq(props.getFilename()), any(Handler.class));
}
use of io.vertx.core.Context in project hono by eclipse.
the class FileBasedRegistrationServiceTest method testDuplicateRegistrationFails.
/**
* Verifies that the registry returns 409 when trying to register a device twice.
*
* @param ctx The vert.x test context.
*/
@Test
public void testDuplicateRegistrationFails(final TestContext ctx) {
final EventBusMessage createRequest = newRequest(RegistrationConstants.ACTION_REGISTER, TENANT);
registrationService.processRequest(createRequest).map(response -> {
ctx.assertEquals(HttpURLConnection.HTTP_CREATED, response.getStatus());
return response;
}).compose(ok -> registrationService.processRequest(createRequest)).setHandler(ctx.asyncAssertSuccess(response -> {
ctx.assertEquals(HttpURLConnection.HTTP_CONFLICT, response.getStatus());
}));
}
Aggregations