server/src/test/java/com/vaadin/data/provider/ReplaceListDataProviderTest.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

package com.vaadin.data.provider;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotEquals;
import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.Arrays;

import org.junit.Test;

/**
 * Test class that verifies that ReplaceListDataProvider functions the way it's
 * meant to.
 *
 */
public class ReplaceListDataProviderTest {

    private static final StrBean TEST_OBJECT = new StrBean("Foo", 10, -1);
    private ReplaceListDataProvider dataProvider = new ReplaceListDataProvider(
            new ArrayList<>(Arrays.asList(TEST_OBJECT)));

    @Test
    public void testGetIdOfItem() {
        Object id = dataProvider.fetch(new Query<>()).findFirst()
                .map(dataProvider::getId).get();
        assertEquals("DataProvider not using correct identifier getter",
                TEST_OBJECT.getId(), id);
    }

    @Test
    public void testGetIdOfReplacementItem() {
        assertFalse("Test object was stale before making any changes.",
                dataProvider.isStale(TEST_OBJECT));

        dataProvider.refreshItem(new StrBean("Replacement TestObject", 10, -2));

        StrBean fromDataProvider = dataProvider.fetch(new Query<>()).findFirst()
                .get();
        Object id = dataProvider.getId(fromDataProvider);

        assertNotEquals("DataProvider did not return the replacement",
                TEST_OBJECT, fromDataProvider);

        assertEquals("DataProvider not using correct identifier getter",
                TEST_OBJECT.getId(), id);

        assertTrue("Old test object should be stale",
                dataProvider.isStale(TEST_OBJECT));
    }
}