1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19 package org.apache.giraph;
20
21 import org.apache.giraph.conf.GiraphConfiguration;
22 import org.apache.giraph.conf.GiraphConstants;
23 import org.apache.giraph.examples.GeneratedVertexReader;
24 import org.apache.giraph.examples.SimplePageRankComputation;
25 import org.apache.giraph.examples.TestComputationStateComputation;
26 import org.apache.giraph.job.GiraphJob;
27 import org.junit.Test;
28
29 import java.io.IOException;
30
31 import static org.junit.Assert.assertTrue;
32
33 public class TestComputationState extends BspCase {
34 public TestComputationState() {
35 super(TestComputationState.class.getName());
36 }
37
38 @Test
39 public void testComputationState() throws IOException,
40 ClassNotFoundException, InterruptedException {
41 if (runningInDistributedMode()) {
42 System.out.println(
43 "testComputeContext: Ignore this test in distributed mode.");
44 return;
45 }
46 GiraphConfiguration conf = new GiraphConfiguration();
47 conf.setComputationClass(TestComputationStateComputation.class);
48 conf.setVertexInputFormatClass(
49 SimplePageRankComputation.SimplePageRankVertexInputFormat.class);
50 conf.setWorkerContextClass(
51 TestComputationStateComputation.TestComputationStateWorkerContext.class);
52 GiraphJob job = prepareJob(getCallingMethodName(), conf);
53
54 job.getConfiguration().setNumComputeThreads(
55 TestComputationStateComputation.NUM_COMPUTE_THREADS);
56
57 GeneratedVertexReader.READER_VERTICES.set(job.getConfiguration(),
58 TestComputationStateComputation.NUM_VERTICES);
59
60 GiraphConstants.USER_PARTITION_COUNT.set(job.getConfiguration(),
61 TestComputationStateComputation.NUM_PARTITIONS);
62 assertTrue(job.run(true));
63 }
64 }