private void setupMockGadgetAdminStore(boolean isAllowed) { EasyMock.expect(gadgetAdminStore.checkFeatureAdminInfo(EasyMock.isA(Gadget.class))) .andReturn(isAllowed).anyTimes(); EasyMock.expect(gadgetAdminStore.getAdditionalRpcServiceIds(EasyMock.isA(Gadget.class))) .andReturn(Sets.newHashSet(RPC_SERVICE_3)); }
private void setupMockGadgetAdminStore(boolean isAllowed) { EasyMock.expect(gadgetAdminStore.checkFeatureAdminInfo(EasyMock.isA(Gadget.class))) .andReturn(isAllowed).anyTimes(); EasyMock.expect(gadgetAdminStore.getAdditionalRpcServiceIds(EasyMock.isA(Gadget.class))) .andReturn(Sets.newHashSet(RPC_SERVICE_3)); }
private void setupGadgetAdminStore() { EasyMock.expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))) .andReturn(true).anyTimes(); EasyMock.expect(gadgetAdminStore.getAdditionalRpcServiceIds(isA(Gadget.class))) .andReturn((Sets.newHashSet(RPC_SERVICE_3))).anyTimes(); }
private void setupGadgetAdminStore() { EasyMock.expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))) .andReturn(true).anyTimes(); EasyMock.expect(gadgetAdminStore.getAdditionalRpcServiceIds(isA(Gadget.class))) .andReturn((Sets.newHashSet(RPC_SERVICE_3))).anyTimes(); }
if(!gadgetAdminStore.checkFeatureAdminInfo(gadget)) { throw new GadgetException(Code.GADGET_ADMIN_FEATURE_NOT_ALLOWED);
@Test(expected = RewritingException.class) public void exceptionWhenFeatureNotAllowed() throws Exception { Gadget gadget = makeDefaultGadget(); reset(gadgetAdminStore); expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))).andReturn(false); replay(gadgetAdminStore); rewrite(gadget, BODY_CONTENT); }
@Test(expected = RewritingException.class) public void exceptionWhenFeatureNotAllowed() throws Exception { Gadget gadget = makeDefaultGadget(); reset(gadgetAdminStore); expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))).andReturn(false); replay(gadgetAdminStore); rewrite(gadget, BODY_CONTENT); }
private Gadget makeGadgetWithSpec(String gadgetXml) throws GadgetException { GadgetSpec spec = new GadgetSpec(SPEC_URL, gadgetXml); Gadget gadget = new Gadget() .setContext(context) .setPreloads(ImmutableList.<PreloadedData>of()) .setSpec(spec) .setCurrentView(spec.getView(GadgetSpec.DEFAULT_VIEW)) .setGadgetFeatureRegistry(featureRegistry); // Convenience: by default expect no features requested, by gadget or extern. // expectFeatureCalls(...) resets featureRegistry if called again. expectFeatureCalls(gadget, ImmutableList.<FeatureResource>of(), ImmutableSet.<String>of(), ImmutableList.<FeatureResource>of()); //Convenience: by default expect that the gadget is allowed to render reset(gadgetAdminStore); expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))).andReturn(true); expect(gadgetAdminStore.isAllowedFeature(isA(Feature.class), isA(Gadget.class))) .andReturn(true).anyTimes(); replay(gadgetAdminStore); return gadget; }
private Gadget makeGadgetWithSpec(String gadgetXml) throws GadgetException { GadgetSpec spec = new GadgetSpec(SPEC_URL, gadgetXml); Gadget gadget = new Gadget() .setContext(context) .setPreloads(ImmutableList.<PreloadedData>of()) .setSpec(spec) .setCurrentView(spec.getView(GadgetSpec.DEFAULT_VIEW)) .setGadgetFeatureRegistry(featureRegistry); // Convenience: by default expect no features requested, by gadget or extern. // expectFeatureCalls(...) resets featureRegistry if called again. expectFeatureCalls(gadget, ImmutableList.<FeatureResource>of(), ImmutableSet.<String>of(), ImmutableList.<FeatureResource>of()); //Convenience: by default expect that the gadget is allowed to render reset(gadgetAdminStore); expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))).andReturn(true); expect(gadgetAdminStore.isAllowedFeature(isA(Feature.class), isA(Gadget.class))) .andReturn(true).anyTimes(); replay(gadgetAdminStore); return gadget; }
Feature denied = mock(Feature.class); expect(denied.getName()).andReturn("hello"); expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))).andReturn(true); expect(gadgetAdminStore.isAllowedFeature(eq(denied), isA(Gadget.class))).andReturn(false); replay();
Feature denied = mock(Feature.class); expect(denied.getName()).andReturn("hello"); expect(gadgetAdminStore.checkFeatureAdminInfo(isA(Gadget.class))).andReturn(true); expect(gadgetAdminStore.isAllowedFeature(eq(denied), isA(Gadget.class))).andReturn(false); replay();
if(!gadgetAdminStore.checkFeatureAdminInfo(gadget)) { throw new ProcessingException("Gadget is not trusted to render in this container.", HttpResponse.SC_BAD_REQUEST);