if (signature.processElement().isSplittable()) { final Coder<?> restrictionCoder = DoFnInvokers.invokerFor(doFn).invokeGetRestrictionCoder(pipeline.getCoderRegistry()); restrictionCoderId = components.registerCoder(restrictionCoder); } else {
when(fn.processElement(mockProcessContext, tracker)).thenReturn(resume()); assertEquals(coder, invoker.invokeGetRestrictionCoder(CoderRegistry.createDefault())); assertEquals(restriction, invoker.invokeGetInitialRestriction("blah")); final List<SomeRestriction> outputs = new ArrayList<>();
RestrictionWithDefaultTracker.class, CoderForDefaultTracker.class)); assertThat( invoker.<RestrictionWithDefaultTracker>invokeGetRestrictionCoder(coderRegistry), instanceOf(CoderForDefaultTracker.class)); invoker.invokeSplitRestriction(
Coder<RestrictionT> restrictionCoder = DoFnInvokers.invokerFor(doFn) .invokeGetRestrictionCoder(input.getPipeline().getCoderRegistry()); Coder<KV<InputT, RestrictionT>> splitCoder = KvCoder.of(input.getCoder(), restrictionCoder);