Search in sources :

Example 41 with ByteArray

use of org.apache.ignite.lang.ByteArray in project ignite-3 by apache.

the class WatchAggregatorTest method testOneCriterionInference.

@Test
public void testOneCriterionInference() {
    var watchAggregator = new WatchAggregator();
    watchAggregator.add(new ByteArray("key"), null);
    var keyCriterion = watchAggregator.watch(0, null).get().keyCriterion();
    assertEquals(new KeyCriterion.ExactCriterion(new ByteArray("key")), keyCriterion);
}
Also used : KeyCriterion(org.apache.ignite.internal.metastorage.watch.KeyCriterion) ByteArray(org.apache.ignite.lang.ByteArray) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Test(org.junit.jupiter.api.Test)

Example 42 with ByteArray

use of org.apache.ignite.lang.ByteArray in project ignite-3 by apache.

the class WatchAggregatorTest method testCancel.

@Test
public void testCancel() {
    var watchAggregator = new WatchAggregator();
    var lsnr1 = mock(WatchListener.class);
    when(lsnr1.onUpdate(any())).thenReturn(true);
    var lsnr2 = mock(WatchListener.class);
    when(lsnr2.onUpdate(any())).thenReturn(true);
    final var id1 = watchAggregator.add(new ByteArray("1"), lsnr1);
    var id2 = watchAggregator.add(new ByteArray("2"), lsnr2);
    var entryEvt1 = new EntryEvent(entry("1", "value1", 1, 1), entry("1", "value1n", 1, 1));
    var entryEvt2 = new EntryEvent(entry("2", "value2", 1, 1), entry("2", "value2n", 1, 1));
    watchAggregator.watch(1, (v1, v2) -> {
    }).get().listener().onUpdate(new WatchEvent(List.of(entryEvt1, entryEvt2)));
    verify(lsnr1, times(1)).onUpdate(any());
    verify(lsnr2, times(1)).onUpdate(any());
    watchAggregator.cancel(id1);
    watchAggregator.watch(1, (v1, v2) -> {
    }).get().listener().onUpdate(new WatchEvent(List.of(entryEvt1, entryEvt2)));
    verify(lsnr1, times(1)).onUpdate(any());
    verify(lsnr2, times(2)).onUpdate(any());
}
Also used : ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) Arrays(java.util.Arrays) WatchEvent(org.apache.ignite.internal.metastorage.client.WatchEvent) WatchListener(org.apache.ignite.internal.metastorage.client.WatchListener) ByteArray(org.apache.ignite.lang.ByteArray) Mockito.times(org.mockito.Mockito.times) Mockito.when(org.mockito.Mockito.when) StandardCharsets(java.nio.charset.StandardCharsets) Mockito.verify(org.mockito.Mockito.verify) HashSet(java.util.HashSet) Test(org.junit.jupiter.api.Test) Nullable(org.jetbrains.annotations.Nullable) List(java.util.List) ArgumentCaptor(org.mockito.ArgumentCaptor) KeyCriterion(org.apache.ignite.internal.metastorage.watch.KeyCriterion) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) EntryEvent(org.apache.ignite.internal.metastorage.client.EntryEvent) NotNull(org.jetbrains.annotations.NotNull) Mockito.mock(org.mockito.Mockito.mock) Entry(org.apache.ignite.internal.metastorage.client.Entry) EntryEvent(org.apache.ignite.internal.metastorage.client.EntryEvent) ByteArray(org.apache.ignite.lang.ByteArray) WatchEvent(org.apache.ignite.internal.metastorage.client.WatchEvent) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Test(org.junit.jupiter.api.Test)

Example 43 with ByteArray

use of org.apache.ignite.lang.ByteArray in project ignite-3 by apache.

the class WatchAggregatorTest method testNullKeyAsEndOfRangeCriterion.

@Test
public void testNullKeyAsEndOfRangeCriterion() {
    var watchAggregator = new WatchAggregator();
    watchAggregator.add(new ByteArray("key3"), null);
    watchAggregator.add(new ByteArray("key1"), null, null);
    var keyCriterion = watchAggregator.watch(0, null).get().keyCriterion();
    var expKeyCriterion = new KeyCriterion.RangeCriterion(new ByteArray("key1"), null);
    assertEquals(expKeyCriterion, keyCriterion);
}
Also used : ByteArray(org.apache.ignite.lang.ByteArray) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Test(org.junit.jupiter.api.Test)

Example 44 with ByteArray

use of org.apache.ignite.lang.ByteArray in project ignite-3 by apache.

the class WatchAggregatorTest method testExactInTheMiddleAndRangeCriteriaOnTheEdgesUnion.

@Test
public void testExactInTheMiddleAndRangeCriteriaOnTheEdgesUnion() {
    var watchAggregator = new WatchAggregator();
    watchAggregator.add(new ByteArray("key1"), null);
    watchAggregator.add(new ByteArray("key0"), new ByteArray("key2"), null);
    var keyCriterion = watchAggregator.watch(0, null).get().keyCriterion();
    var expKeyCriterion = new KeyCriterion.RangeCriterion(new ByteArray("key0"), new ByteArray("key2"));
    assertEquals(expKeyCriterion, keyCriterion);
}
Also used : ByteArray(org.apache.ignite.lang.ByteArray) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Test(org.junit.jupiter.api.Test)

Example 45 with ByteArray

use of org.apache.ignite.lang.ByteArray in project ignite-3 by apache.

the class WatchAggregatorTest method testCancelByFalseFromListener.

@Test
public void testCancelByFalseFromListener() {
    var watchAggregator = new WatchAggregator();
    var lsnr1 = mock(WatchListener.class);
    when(lsnr1.onUpdate(any())).thenReturn(false);
    var lsnr2 = mock(WatchListener.class);
    when(lsnr2.onUpdate(any())).thenReturn(true);
    var id1 = watchAggregator.add(new ByteArray("1"), lsnr1);
    var id2 = watchAggregator.add(new ByteArray("2"), lsnr2);
    var entryEvt1 = new EntryEvent(entry("1", "value1", 1, 1), entry("1", "value1n", 1, 1));
    var entryEvt2 = new EntryEvent(entry("2", "value2", 1, 1), entry("2", "value2n", 1, 1));
    watchAggregator.watch(1, (v1, v2) -> {
    }).get().listener().onUpdate(new WatchEvent(List.of(entryEvt1, entryEvt2)));
    verify(lsnr1, times(1)).onUpdate(any());
    verify(lsnr2, times(1)).onUpdate(any());
    watchAggregator.watch(1, (v1, v2) -> {
    }).get().listener().onUpdate(new WatchEvent(List.of(entryEvt1, entryEvt2)));
    verify(lsnr1, times(1)).onUpdate(any());
    verify(lsnr2, times(2)).onUpdate(any());
}
Also used : ArgumentMatchers.any(org.mockito.ArgumentMatchers.any) Arrays(java.util.Arrays) WatchEvent(org.apache.ignite.internal.metastorage.client.WatchEvent) WatchListener(org.apache.ignite.internal.metastorage.client.WatchListener) ByteArray(org.apache.ignite.lang.ByteArray) Mockito.times(org.mockito.Mockito.times) Mockito.when(org.mockito.Mockito.when) StandardCharsets(java.nio.charset.StandardCharsets) Mockito.verify(org.mockito.Mockito.verify) HashSet(java.util.HashSet) Test(org.junit.jupiter.api.Test) Nullable(org.jetbrains.annotations.Nullable) List(java.util.List) ArgumentCaptor(org.mockito.ArgumentCaptor) KeyCriterion(org.apache.ignite.internal.metastorage.watch.KeyCriterion) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Assertions.assertEquals(org.junit.jupiter.api.Assertions.assertEquals) EntryEvent(org.apache.ignite.internal.metastorage.client.EntryEvent) NotNull(org.jetbrains.annotations.NotNull) Mockito.mock(org.mockito.Mockito.mock) Entry(org.apache.ignite.internal.metastorage.client.Entry) EntryEvent(org.apache.ignite.internal.metastorage.client.EntryEvent) ByteArray(org.apache.ignite.lang.ByteArray) WatchEvent(org.apache.ignite.internal.metastorage.client.WatchEvent) WatchAggregator(org.apache.ignite.internal.metastorage.watch.WatchAggregator) Test(org.junit.jupiter.api.Test)

Aggregations

ByteArray (org.apache.ignite.lang.ByteArray)51 Test (org.junit.jupiter.api.Test)35 Cursor (org.apache.ignite.internal.util.Cursor)13 List (java.util.List)12 WatchAggregator (org.apache.ignite.internal.metastorage.watch.WatchAggregator)12 Assertions.assertEquals (org.junit.jupiter.api.Assertions.assertEquals)11 Map (java.util.Map)10 NotNull (org.jetbrains.annotations.NotNull)9 Collection (java.util.Collection)8 Iterator (java.util.Iterator)8 Collectors (java.util.stream.Collectors)8 AfterEach (org.junit.jupiter.api.AfterEach)8 Assertions.assertTrue (org.junit.jupiter.api.Assertions.assertTrue)8 Assertions.fail (org.junit.jupiter.api.Assertions.fail)8 BeforeEach (org.junit.jupiter.api.BeforeEach)8 ByteBuffer (java.nio.ByteBuffer)7 UTF_8 (java.nio.charset.StandardCharsets.UTF_8)7 ArrayList (java.util.ArrayList)7 NoSuchElementException (java.util.NoSuchElementException)7 Function.identity (java.util.function.Function.identity)7