/** */ @Override public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException { try (Closeable closeable = ObjectifyService.begin()) { chain.doFilter(request, response); } } }
/** * <p>Runs one unit of work, making the root Objectify context available. This does not start a transaction, * but it makes the static ofy() method return an appropriate object.</p> * * <p>Normally you do not need to use this method. When servicing a normal request, the ObjectifyFilter * will run this for you. This method is useful for using Objectify outside of a normal request - * using the remote api, for example.</p> * * <p>Alternatively, you can use the begin() method and close the session manually.</p> * * @return the result of the work. */ public static <R> R run(final Work<R> work) { try (Closeable closeable = begin()) { return work.run(); } }
@BeforeClass public void setupObjectify() { OfyHelper.registerEntityClasses(); closeable = ObjectifyService.begin(); }
protected void doOperationRemotely() throws IOException { String appUrl = ClientProperties.TARGET_URL.replaceAll("^https?://", ""); String appDomain = appUrl.split(":")[0]; int appPort = appUrl.contains(":") ? Integer.parseInt(appUrl.split(":")[1]) : 443; System.out.println("--- Starting remote operation ---"); System.out.println("Going to connect to:" + appDomain + ":" + appPort); RemoteApiOptions options = new RemoteApiOptions().server(appDomain, appPort); if (ClientProperties.isTargetUrlDevServer()) { // Dev Server doesn't require credential. options.useDevelopmentServerCredential(); } else { // Your Google Cloud SDK needs to be authenticated for Application Default Credentials // in order to run any script in production server. // Refer to https://developers.google.com/identity/protocols/application-default-credentials. options.useApplicationDefaultCredential(); } RemoteApiInstaller installer = new RemoteApiInstaller(); installer.install(options); OfyHelper.registerEntityClasses(); Closeable objectifySession = ObjectifyService.begin(); try { doOperation(); } finally { objectifySession.close(); installer.uninstall(); } System.out.println("--- Remote operation completed ---"); }
@Override public <T> T before(Request req, Response resp) { closeables.set(ObjectifyService.begin()); return super.before(req, resp); }
@Before public void setUp() { helper.setUp(); objectifyCloseable = ObjectifyService.begin(); }
@Before public void setUp() { helper.setUp(); objectify = ObjectifyService.begin(); }
@Before public void setUp() { helper.setUp(); objectifyCloseable = ObjectifyService.begin(); }
@Before public void setUp() { helper.setUp(); objectify = ObjectifyService.begin(); }
@Before public void setUp() { helper.setUp(); objectifyCloseable = ObjectifyService.begin(); }
@Before public void setUpDatastore() { helper.setUp(); ofy = ObjectifyService.begin(); // Clear up MySQL catalog caches CountryStructure.clearCache(); AdminEntityTable.clearCache(); }
@Before public void setUp() throws IOException { objectifyService = ObjectifyService.begin(); Provider<AuthenticatedUser> authProvider = Providers.of(new AuthenticatedUser("", USER_ID, "jorden@bdd.com")); databaseProvider = injector.getInstance(DatabaseProvider.class); resourceLocator = new ResourceLocatorSyncImpl( injector.getProvider(FormStorageProvider.class), databaseProvider, authProvider, new BlobAuthorizerStub()); OdkFormFieldBuilderFactory fieldFactory = new OdkFormFieldBuilderFactory(resourceLocator); AuthTokenProvider tokenService = new AuthTokenProvider(); TestBlobstoreService blobstore = new TestBlobstoreService(injector.getInstance(DeploymentConfiguration.class), injector.getInstance(EntityManager.class), injector.getInstance(FormStorageProvider.class), injector.getInstance(DatabaseProvider.class)); TestInstanceIdService idService = new TestInstanceIdService(); SubmissionArchiver backupService = new SubmissionArchiver( new DeploymentConfiguration(new Properties())); formResource = new XFormResources(resourceLocator, authProvider, fieldFactory, tokenService, databaseProvider); formSubmissionResource = new XFormSubmissionResource( getDispatcherSync(), resourceLocator, tokenService, blobstore, idService, backupService); }
@Before public final void setup() { helper.setUp(); user = new User(); userDAO = createMock(UserDAO.class); expect(userDAO.findUserByChangePasswordKey(eq(VALID_KEY))).andReturn(user); expect(userDAO.findUserByChangePasswordKey(EasyMock.not(eq(VALID_KEY)))).andThrow(new NoResultException()); replay(userDAO); AuthenticationDAO authDAO = createMock(AuthenticationDAO.class); authDAO.persist(isA(Authentication.class)); expectLastCall().anyTimes(); MailingListClient mailingListClient = createNiceMock(MailingListClient.class); replay(mailingListClient); AuthTokenProvider authTokenProvider = new AuthTokenProvider( Providers.of(authDAO)); resource = new ConfirmInviteController( Providers.of(userDAO), authTokenProvider, mailingListClient); objectify = ObjectifyService.begin(); }
@Before public void setUp() throws URISyntaxException { resource = new DatabaseResource(formStorageProvider, dispatcher, databaseProvider, Providers.of(em), mailSender, billingOracle, CuidAdapter.databaseId(databaseId).asString()); uri = RestMockUtils.mockUriInfo("http://www.activityinfo.org/"); helper.setUp(); objectifyCloseable = ObjectifyService.begin(); }
@Before public final void uploadBlob() throws IOException { localServiceTestHelper.setUp(); ofy = ObjectifyService.begin(); blobService = injector.getInstance(GcsBlobFieldStorageService.class); blobService.setTestBucketName(); locator = new ResourceLocatorAdaptor( new ActivityInfoClientAsyncStub(injector.getProvider(EntityManager.class), injector.getInstance(DatabaseProvider.class), blobService)); AuthenticationModuleStub.setUserId(1); user = new AuthenticatedUser("x", 1, "user1@user.com"); noAccessUser = new AuthenticatedUser("x", 3, "stefan@user.com"); blobId = BlobId.generate(); blobService.put(user, "attachment;filename=" + FILE_NAME, MimeTypeUtil.mimeTypeFromFileName(FILE_NAME), blobId, resourceId, GcsBlobFieldStorageServiceTest.class.getResourceAsStream("goabout.png")); }
@Before public void setUp() throws URISyntaxException { databaseResource = new DatabaseResource(formStorageProvider, dispatcher, databaseProvider, Providers.of(em), mailSender, billingOracle, CuidAdapter.databaseId(databaseId).asString()); approvalResource = new ApprovalResource(Providers.of(em), dispatcher, formStorageProvider, mailSender, authTokenProvider, databaseProvider, billingOracle); uri = RestMockUtils.mockUriInfo("https://activityinfo.org/"); helper.setUp(); objectifyCloseable = ObjectifyService.begin(); }
@Before public void setUp() throws Exception { // Don't call super.setUp because we initialize slightly differently // here... countdownLatch = new LocalTaskQueueTestConfig.TaskCountDownLatch(1); // See // http://www.ensor.cc/2010/11/unit-testing-named-queues-spring.html // NOTE: THE QUEUE XML PATH RELATIVE TO WEB APP ROOT, More info // below // http://stackoverflow.com/questions/11197058/testing-non-default-app-engine-task-queues final LocalTaskQueueTestConfig localTaskQueueConfig = new LocalTaskQueueTestConfig() .setDisableAutoTaskExecution(false).setQueueXmlPath("src/test/resources/queue.xml") .setTaskExecutionLatch(countdownLatch).setCallbackClass(DeleteShardedCounterDeferredCallback.class); // Use a different queue.xml for testing purposes helper = new LocalServiceTestHelper(new LocalDatastoreServiceTestConfig().setApplyAllHighRepJobPolicy(), new LocalMemcacheServiceTestConfig(), new LocalCapabilitiesServiceTestConfig(), localTaskQueueConfig); helper.setUp(); memcache = MemcacheServiceFactory.getMemcacheService(); capabilitiesService = CapabilitiesServiceFactory.getCapabilitiesService(); // New Objectify 5.1 Way. See https://groups.google.com/forum/#!topic/objectify-appengine/O4FHC_i7EGk this.session = ObjectifyService.begin(); // Enable Joda Translators JodaTimeTranslators.add(ObjectifyService.factory()); ObjectifyService.factory().register(CounterData.class); ObjectifyService.factory().register(CounterShardData.class); ObjectifyService.factory().register(CounterShardOperationData.class); shardedCounterServiceImpl = new ShardedCounterServiceImpl(); this.shardedCounterService = shardedCounterServiceImpl; }