Examples with DefaultBackupInput io.atomix.primitive.service.impl.DefaultBackupInput used on opensource projects

Example 16 with DefaultBackupInput

use of io.atomix.primitive.service.impl.DefaultBackupInput in project atomix by atomix.

the class DefaultAtomicMapServiceTest method testSnapshot.

@Test
@SuppressWarnings("unchecked")
public void testSnapshot() throws Exception {
    ServiceContext context = mock(ServiceContext.class);
    when(context.serviceType()).thenReturn(AtomicMapType.instance());
    when(context.serviceName()).thenReturn("test");
    when(context.serviceId()).thenReturn(PrimitiveId.from(1));
    when(context.wallClock()).thenReturn(new WallClock());
    Session session = mock(Session.class);
    when(session.sessionId()).thenReturn(SessionId.from(1));
    when(context.currentSession()).thenReturn(session);
    AbstractAtomicMapService service = new TestAtomicMapService();
    service.register(session);
    service.init(context);
    service.put("foo", "Hello world!".getBytes());
    service.lock("bar", 1, 0);
    Buffer buffer = HeapBuffer.allocate();
    service.backup(new DefaultBackupOutput(buffer, service.serializer()));
    service = new TestAtomicMapService();
    service.restore(new DefaultBackupInput(buffer.flip(), service.serializer()));
    Versioned<byte[]> value = service.get("foo");
    assertNotNull(value);
    assertArrayEquals("Hello world!".getBytes(), value.value());
    assertFalse(service.isLocked("foo", 0));
    assertTrue(service.isLocked("bar", 0));
}

Example 17 with DefaultBackupInput

use of io.atomix.primitive.service.impl.DefaultBackupInput in project atomix by atomix.

the class DefaultAtomicNavigableMapServiceTest method testSnapshot.

@Test
@SuppressWarnings("unchecked")
public void testSnapshot() throws Exception {
    ServiceContext context = mock(ServiceContext.class);
    when(context.serviceType()).thenReturn(AtomicMapType.instance());
    when(context.serviceName()).thenReturn("test");
    when(context.serviceId()).thenReturn(PrimitiveId.from(1));
    when(context.wallClock()).thenReturn(new WallClock());
    Session session = mock(Session.class);
    when(session.sessionId()).thenReturn(SessionId.from(1));
    AbstractAtomicNavigableMapService service = new TestAtomicNavigableMapService();
    service.init(context);
    service.put("foo", "Hello world!".getBytes());
    Buffer buffer = HeapBuffer.allocate();
    service.backup(new DefaultBackupOutput(buffer, service.serializer()));
    service = new TestAtomicNavigableMapService();
    service.restore(new DefaultBackupInput(buffer.flip(), service.serializer()));
    Versioned<byte[]> value = service.get("foo");
    assertNotNull(value);
    assertArrayEquals("Hello world!".getBytes(), value.value());
}

Example 18 with DefaultBackupInput

use of io.atomix.primitive.service.impl.DefaultBackupInput in project atomix by atomix.

the class DefaultAtomicCounterServiceTest method testSnapshot.

@Test
public void testSnapshot() throws Exception {
    DefaultAtomicCounterService service = new DefaultAtomicCounterService();
    service.set(1);
    Buffer buffer = HeapBuffer.allocate();
    service.backup(new DefaultBackupOutput(buffer, service.serializer()));
    service = new DefaultAtomicCounterService();
    service.restore(new DefaultBackupInput(buffer.flip(), service.serializer()));
    long value = service.get();
    assertEquals(1, value);
}