Examples with Race org.neo4j.test.Race used on opensource projects

Example 21 with Race

use of org.neo4j.test.Race in project neo4j by neo4j.

the class ArrayEncoderTest method shouldEncodeProperlyWithMultipleThreadsRacing.

@Test
public void shouldEncodeProperlyWithMultipleThreadsRacing() throws Throwable {
    // given
    String[] INPUT = { "These strings need to be longer than 57 bytes, because that is the line wrapping length of BASE64.", "This next line is also long. The number of strings in this array is the number of threads to use.", "Each thread will get a different string as input to encode, and ensure the result is always the same.", "Should the result of an encoding differ even once, the thread will yield a negative overall result.", "If any of the threads yields a negative result, the test will fail, since that should not happen.", "All threads are allowed to run together for a predetermined amount of time, to try to get contention.", "This predetermined time is the minimum runtime of the test, since the timer starts after all threads.", "The idea to use the input data as documentation for the test was just a cute thing I came up with.", "Since my imagination for coming up with test data is usually poor, I figured I'd do something useful.", "Hopefully this isn't just nonsensical drivel, and maybe, just maybe someone might actually read it." };
    Race race = new Race();
    for (String input : INPUT) {
        final String[] inputArray = new String[] { input };
        race.addContestant(() -> {
            String first = ArrayEncoder.encode(inputArray);
            for (int i = 0; i < 1000; i++) {
                String encoded = ArrayEncoder.encode(inputArray);
                assertEquals("Each attempt at encoding should yield the same result. Turns out that first one was '" + first + "', yet another one was '" + encoded + "'", first, encoded);
            }
        });
    }
    race.go();
}

Example 22 with Race

use of org.neo4j.test.Race in project neo4j by neo4j.

the class KernelTransactionTerminationTest method runTwoThreads.

private void runTwoThreads(Consumer<TestKernelTransaction> thread1Action, Consumer<TestKernelTransaction> thread2Action) throws Throwable {
    TestKernelTransaction tx = TestKernelTransaction.create().initialize();
    AtomicLong t1Count = new AtomicLong();
    AtomicLong t2Count = new AtomicLong();
    long endTime = currentTimeMillis() + TEST_RUN_TIME_MS;
    int limit = 20_000;
    Race race = new Race();
    race.withEndCondition(() -> ((t1Count.get() >= limit) && (t2Count.get() >= limit)) || (currentTimeMillis() >= endTime));
    race.addContestant(() -> {
        thread1Action.accept(tx);
        t1Count.incrementAndGet();
    });
    race.addContestant(() -> {
        thread2Action.accept(tx);
        t2Count.incrementAndGet();
    });
    race.go();
}

Example 23 with Race

use of org.neo4j.test.Race in project neo4j by neo4j.

the class NeoStoresIT method shouldWriteOutThePropertyRecordBeforeReferencingItFromARelationshipRecord.

@Test
public void shouldWriteOutThePropertyRecordBeforeReferencingItFromARelationshipRecord() throws Throwable {
    final long node1Id;
    final long node2Id;
    try (Transaction tx = db.beginTx()) {
        Node node1 = db.createNode();
        node1Id = node1.getId();
        Node node2 = db.createNode();
        node2Id = node2.getId();
        tx.success();
    }
    Race race = new Race();
    final long[] latestRelationshipId = new long[1];
    AtomicLong writes = new AtomicLong();
    AtomicLong reads = new AtomicLong();
    long endTime = currentTimeMillis() + SECONDS.toMillis(2);
    race.withEndCondition(() -> (writes.get() > 100 && reads.get() > 10_000) || currentTimeMillis() > endTime);
    race.addContestant(() -> {
        try (Transaction tx = db.beginTx()) {
            Node node1 = db.getGraphDatabaseAPI().getNodeById(node1Id);
            Node node2 = db.getGraphDatabaseAPI().getNodeById(node2Id);
            Relationship rel = node1.createRelationshipTo(node2, FRIEND);
            latestRelationshipId[0] = rel.getId();
            rel.setProperty("largeProperty", LONG_STRING_VALUE);
            tx.success();
        }
        writes.incrementAndGet();
    });
    race.addContestant(() -> {
        try (Transaction tx = db.getGraphDatabaseAPI().beginTx()) {
            Relationship rel = db.getGraphDatabaseAPI().getRelationshipById(latestRelationshipId[0]);
            for (String propertyKey : rel.getPropertyKeys()) {
                rel.getProperty(propertyKey);
            }
            tx.success();
        } catch (NotFoundException e) {
            if (Exceptions.contains(e, InvalidRecordException.class)) {
                throw e;
            }
        }
        reads.incrementAndGet();
    });
    race.go();
}

Example 24 with Race

use of org.neo4j.test.Race in project neo4j by neo4j.

the class NeoStoresIT method shouldWriteOutTheDynamicChainBeforeUpdatingThePropertyRecord.

@Test
public void shouldWriteOutTheDynamicChainBeforeUpdatingThePropertyRecord() throws Throwable {
    Race race = new Race();
    long[] latestNodeId = new long[1];
    AtomicLong writes = new AtomicLong();
    AtomicLong reads = new AtomicLong();
    long endTime = currentTimeMillis() + SECONDS.toMillis(2);
    race.withEndCondition(() -> (writes.get() > 100 && reads.get() > 10_000) || currentTimeMillis() > endTime);
    race.addContestant(() -> {
        try (Transaction tx = db.beginTx()) {
            Node node = db.createNode();
            latestNodeId[0] = node.getId();
            node.setProperty("largeProperty", LONG_STRING_VALUE);
            tx.success();
        }
        writes.incrementAndGet();
    });
    race.addContestant(() -> {
        try (Transaction tx = db.getGraphDatabaseAPI().beginTx()) {
            Node node = db.getGraphDatabaseAPI().getNodeById(latestNodeId[0]);
            for (String propertyKey : node.getPropertyKeys()) {
                node.getProperty(propertyKey);
            }
            tx.success();
        } catch (NotFoundException e) {
        // This will catch nodes not found (expected) and also PropertyRecords not found (shouldn't happen
        // but handled in shouldWriteOutThePropertyRecordBeforeReferencingItFromANodeRecord)
        }
        reads.incrementAndGet();
    });
    race.go();
}

Example 25 with Race

use of org.neo4j.test.Race in project neo4j by neo4j.

the class MetaDataStoreTest method transactionClosedMustBeAtomic.

@Test
public void transactionClosedMustBeAtomic() throws Throwable {
    try (MetaDataStore store = newMetaDataStore()) {
        PagedFile pf = store.storeFile;
        int initialValue = 2;
        store.transactionClosed(initialValue, initialValue, initialValue);
        AtomicLong writeCount = new AtomicLong();
        AtomicLong fileReadCount = new AtomicLong();
        AtomicLong apiReadCount = new AtomicLong();
        int upperLimit = 10_000;
        int lowerLimit = 100;
        long endTime = currentTimeMillis() + SECONDS.toMillis(10);
        Race race = new Race();
        race.withEndCondition(() -> writeCount.get() >= upperLimit && fileReadCount.get() >= upperLimit && apiReadCount.get() >= upperLimit);
        race.withEndCondition(() -> writeCount.get() >= lowerLimit && fileReadCount.get() >= lowerLimit && apiReadCount.get() >= lowerLimit && currentTimeMillis() >= endTime);
        race.addContestants(3, () -> {
            long count = writeCount.incrementAndGet();
            store.transactionCommitted(count, count, count);
        });
        race.addContestants(3, throwing(() -> {
            try (PageCursor cursor = pf.io(0, PagedFile.PF_SHARED_READ_LOCK)) {
                assertTrue(cursor.next());
                long logVersion, byteOffset;
                do {
                    logVersion = store.getRecordValue(cursor, MetaDataStore.Position.LAST_CLOSED_TRANSACTION_LOG_VERSION);
                    byteOffset = store.getRecordValue(cursor, MetaDataStore.Position.LAST_CLOSED_TRANSACTION_LOG_BYTE_OFFSET);
                } while (cursor.shouldRetry());
                assertLogVersionEqualsByteOffset(logVersion, byteOffset, "file");
                fileReadCount.incrementAndGet();
            }
        }));
        race.addContestants(3, () -> {
            long[] transaction = store.getLastClosedTransaction();
            assertLogVersionEqualsByteOffset(transaction[0], transaction[1], "API");
            apiReadCount.incrementAndGet();
        });
        race.go();
    }
}