@Test @SuppressWarnings("unchecked") public void testJavaPairRDDFunctions() throws Exception { JavaPairRDD<String, Integer> mockPairRDD = mock(JavaPairRDD.class); RDD<Tuple2<String, Integer>> mockTuple2RDD = mock(RDD.class); when(mockPairRDD.rdd()).thenReturn(mockTuple2RDD); GemFireJavaPairRDDFunctions wrapper = javaFunctions(mockPairRDD); assertTrue(mockTuple2RDD == wrapper.rddf.rdd()); Tuple3<SparkContext, GemFireConnectionConf, GemFireConnection> tuple3 = createCommonMocks(); when(mockTuple2RDD.sparkContext()).thenReturn(tuple3._1()); String regionPath = "testregion"; wrapper.saveToGemfire(regionPath, tuple3._2()); verify(mockTuple2RDD, times(1)).sparkContext(); verify(tuple3._1(), times(1)) .runJob(eq(mockTuple2RDD), any(Function2.class), any(ClassTag.class)); }