Class1.java
@Autowired
Private ClassX cx;
Public list method1(JavaPairRDD data){
List list1 = new ArrayList();
List list2 = new ArrayList();
JavaPairRDD computed = data.map(
new Function<Tuple2<object, list>>() {
Public List call(object obj) throws exception {
cx.method2(list2);
Return list1;
}});
Return computed.flatMap(new FlatMapfunc() { do something }}).collect();
}
}
Class1Test.java
@Mock
Private ClassX cx;
@InjectMocks
Private Class1 c1;
@Before
Public void init() {
Super.init();
Sc = getSparkContext();
doNothing().when(cx).method2(Collections.<object>emptyList());
}
@Test
Public void testMethod1() {
JavaRDD alldata = Sc.paraalelize("");
JavaPairRDD data = createrdd(alldata);
c1.method1(rdd);
}
Sudershan Malpani
Sent from my iPhone
> On Sep 9, 2014, at 5:16 PM, Reynold Xin <[email protected]> wrote:
>
> Can you be a little bit more specific, maybe give a code snippet?
>
>
>> On Tue, Sep 9, 2014 at 5:14 PM, Sudershan Malpani
>> <[email protected]> wrote:
>> Hi all,
>>
>> I am calling an object which in turn is calling a method inside a map RDD in
>> spark. While writing the tests how can I mock that object's call? Currently
>> I did doNothing().when(class).method() is called but it is giving task not
>> serializable exception. I tried making the class both spy and mock.
>>
>> Sudershan Malpani
>> Sent from my iPhone
>> ---------------------------------------------------------------------
>> To unsubscribe, e-mail: [email protected]
>> For additional commands, e-mail: [email protected]
>>
>
