Examples with AnonymousMethod org.apache.beam.sdk.transforms.reflect.DoFnSignaturesTestUtils.AnonymousMethod used on opensource projects

Example 1 with AnonymousMethod

use of org.apache.beam.sdk.transforms.reflect.DoFnSignaturesTestUtils.AnonymousMethod in project beam by apache.

the class DoFnSignaturesSplittableDoFnTest method testSplittableProcessElementMustNotHaveOtherParams.

@Test
public void testSplittableProcessElementMustNotHaveOtherParams() throws Exception {
    thrown.expect(IllegalArgumentException.class);
    thrown.expectMessage("Illegal parameter");
    thrown.expectMessage("BoundedWindow");
    DoFnSignature.ProcessElementMethod signature = analyzeProcessElementMethod(new AnonymousMethod() {

        private void method(DoFn<Integer, String>.ProcessContext<Integer, String> context, SomeRestrictionTracker tracker, BoundedWindow window) {
        }
    });
}

Example 2 with AnonymousMethod

use of org.apache.beam.sdk.transforms.reflect.DoFnSignaturesTestUtils.AnonymousMethod in project beam by apache.

the class DoFnSignaturesProcessElementTest method testBadGenericWildCards.

@Test
public void testBadGenericWildCards() throws Exception {
    thrown.expect(IllegalArgumentException.class);
    thrown.expectMessage("DoFn<Integer, ? super Integer>.ProcessContext");
    thrown.expectMessage("must have type");
    thrown.expectMessage("DoFn<Integer, String>.ProcessContext");
    analyzeProcessElementMethod(new AnonymousMethod() {

        private void method(DoFn<Integer, ? super Integer>.ProcessContext c) {
        }
    });
}

Example 3 with AnonymousMethod

use of org.apache.beam.sdk.transforms.reflect.DoFnSignaturesTestUtils.AnonymousMethod in project beam by apache.

the class DoFnSignaturesSplittableDoFnTest method testHasRestrictionTracker.

@Test
// used via reflection
@SuppressWarnings("unused")
public void testHasRestrictionTracker() throws Exception {
    DoFnSignature.ProcessElementMethod signature = analyzeProcessElementMethod(new AnonymousMethod() {

        private void method(DoFn<Integer, String>.ProcessContext context, SomeRestrictionTracker tracker) {
        }
    });
    assertTrue(signature.isSplittable());
    assertTrue(signature.extraParameters().stream().anyMatch(Predicates.instanceOf(DoFnSignature.Parameter.RestrictionTrackerParameter.class)::apply));
    assertEquals(SomeRestrictionTracker.class, signature.trackerT().getRawType());
}

Example 4 with AnonymousMethod

use of org.apache.beam.sdk.transforms.reflect.DoFnSignaturesTestUtils.AnonymousMethod in project beam by apache.

the class DoFnSignaturesSplittableDoFnTest method testReturnsProcessContinuation.

@Test
// used via reflection
@SuppressWarnings("unused")
public void testReturnsProcessContinuation() throws Exception {
    DoFnSignature.ProcessElementMethod signature = analyzeProcessElementMethod(new AnonymousMethod() {

        private DoFn.ProcessContinuation method(DoFn<Integer, String>.ProcessContext context) {
            return null;
        }
    });
    assertTrue(signature.hasReturnValue());
}

Example 5 with AnonymousMethod

use of org.apache.beam.sdk.transforms.reflect.DoFnSignaturesTestUtils.AnonymousMethod in project beam by apache.

the class DoFnSignaturesProcessElementTest method testBadExtraProcessContextType.

@Test
public void testBadExtraProcessContextType() throws Exception {
    thrown.expect(IllegalArgumentException.class);
    thrown.expectMessage("Integer is not a valid context parameter.");
    analyzeProcessElementMethod(new AnonymousMethod() {

        private void method(DoFn<Integer, String>.ProcessContext c, Integer n) {
        }
    });
}