Examples of SaveTasks

org.apache.aurora.gen.storage.SaveTasks

Examples of org.apache.aurora.gen.storage.SaveTasks

    ScheduledTask snapshotTask = makeTask("snapshotTask", ScheduleStatus.ASSIGNED);
    ScheduledTask transactionTask = makeTask("transactionTask", ScheduleStatus.RUNNING);
    Iterable<Entry> recoveredEntries = toEntries(
        LogEntry.snapshot(new Snapshot().setTasks(ImmutableSet.of(snapshotTask))),
        LogEntry.transaction(new Transaction(
            ImmutableList.of(Op.saveTasks(new SaveTasks(ImmutableSet.of(transactionTask)))),
            storageConstants.CURRENT_SCHEMA_VERSION)));


    expect(log.open()).andReturn(logStream);
    expect(logStream.readAll()).andReturn(recoveredEntries.iterator()).anyTimes();
    // An empty saveTasks is an artifact of the fact that mutateTasks always writes a log operation
    // even if nothing is changed.
    streamMatcher.expectTransaction(Op.saveTasks(new SaveTasks(ImmutableSet.<ScheduledTask>of())))
        .andReturn(nextPosition());
    streamMatcher.expectTransaction(Op.saveFrameworkId(new SaveFrameworkId(FRAMEWORK_ID)))
        .andReturn(nextPosition());


    final CountDownLatch driverStarted = new CountDownLatch(1);

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

    ImmutableSet<IScheduledTask> mutated = ImmutableSet.of(IScheduledTask.build(
            new ScheduledTask().setAssignedTask(new AssignedTask().setTaskId("a"))));


    expect(taskStore.mutateTasks(query, mutator)).andReturn(mutated);
    expect(log.isLoggable(Level.FINE)).andReturn(false);
    expectOp(Op.saveTasks(new SaveTasks(IScheduledTask.toBuildersSet(mutated))));


    // With increased logging.
    expect(taskStore.mutateTasks(query, mutator)).andReturn(mutated);
    expect(log.isLoggable(Level.FINE)).andReturn(true);
    expectOp(Op.saveTasks(new SaveTasks(IScheduledTask.toBuildersSet(mutated))));
    log.fine(EasyMock.anyString());


    control.replay();


    assertEquals(mutated, storage.mutateTasks(query, mutator));

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

    return null;
  }


  @Test
  public void testCoalesce() throws CodingException {
    SaveTasks saveTasks1 = createSaveTasks("1", "2");
    createSaveTasks("2");
    SaveTasks saveTasks2 = createSaveTasks("1", "3");
    SaveTasks saveTasks3 = createSaveTasks("4", "5");


    // saveTasks1 is unrepresented because both of its operations were trumped.
    // saveTasks3 is unrepresented because its operations were deleted.
    SaveTasks coalescedSaves = createSaveTasks("3", "2", "1");


    RemoveTasks removeTasks1 = createRemoveTasks("1", "2");
    RemoveTasks removeTasks2 = createRemoveTasks("3");
    RemoveTasks removeTasks3 = createRemoveTasks("4", "5");

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

                .setAssignedTask(new AssignedTask().setTaskId("task_id")
                    .setTask(new TaskConfig().setJobName("job_name")))));
  }


  private SaveTasks createSaveTasks(String... taskIds) {
    return new SaveTasks(ImmutableSet.copyOf(Iterables.transform(ImmutableList.copyOf(taskIds),
        new Function<String, ScheduledTask>() {
          @Override
          public ScheduledTask apply(String taskId) {
            return new ScheduledTask().setAssignedTask(new AssignedTask().setTaskId(taskId));
          }

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

      @Override
      protected void setupExpectations() throws Exception {
        storageUtil.expectWriteOperation();
        storageUtil.taskStore.saveTasks(tasks);
        streamMatcher.expectTransaction(
            Op.saveTasks(new SaveTasks(IScheduledTask.toBuildersSet(tasks))))
            .andReturn(position);
      }


      @Override
      protected void performMutations(MutableStoreProvider storeProvider) {

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

      @Override
      protected void setupExpectations() throws Exception {
        storageUtil.expectWriteOperation();
        expect(storageUtil.taskStore.mutateTasks(query, mutation)).andReturn(mutated);
        streamMatcher.expectTransaction(
            Op.saveTasks(new SaveTasks(IScheduledTask.toBuildersSet(mutated))))
            .andReturn(null);
      }


      @Override
      protected void performMutations(MutableStoreProvider storeProvider) {

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

        expect(storageUtil.taskStore.mutateTasks(query, mutation)).andReturn(mutated);


        storageUtil.taskStore.deleteTasks(tasksToRemove);


        streamMatcher.expectTransaction(
            Op.saveTasks(new SaveTasks(IScheduledTask.toBuildersSet(mutated))),
            Op.removeTasks(new RemoveTasks(tasksToRemove)))
            .andReturn(position);
      }


      @Override

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

        // Nested transaction with result.
        expect(storageUtil.taskStore.mutateTasks(query, mutation)).andReturn(mutated);


        // Resulting stream operation.
        streamMatcher.expectTransaction(Op.saveTasks(
            new SaveTasks(IScheduledTask.toBuildersSet(mutated))))
            .andReturn(null);
      }


      @Override
      protected void performMutations(MutableStoreProvider storeProvider) {

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks

        // Nested transaction with result.
        expect(storageUtil.taskStore.mutateTasks(query, mutation)).andReturn(mutated);


        // Resulting stream operation.
        streamMatcher.expectTransaction(
            Op.saveTasks(new SaveTasks(
                ImmutableSet.<ScheduledTask>builder()
                    .addAll(IScheduledTask.toBuildersList(saved))
                    .addAll(IScheduledTask.toBuildersList(mutated))
                    .build())))
            .andReturn(position);

View Full Code Here

Examples of org.apache.aurora.gen.storage.SaveTasks


    RemoveJob removeJob = new RemoveJob(JOB_KEY.newBuilder());
    builder.add(createTransaction(Op.removeJob(removeJob)));
    storageUtil.jobStore.removeJob(JOB_KEY);


    SaveTasks saveTasks = new SaveTasks(ImmutableSet.of(new ScheduledTask()));
    builder.add(createTransaction(Op.saveTasks(saveTasks)));
    storageUtil.taskStore.saveTasks(IScheduledTask.setFromBuilders(saveTasks.getTasks()));


    RewriteTask rewriteTask = new RewriteTask("id1", new TaskConfig());
    builder.add(createTransaction(Op.rewriteTask(rewriteTask)));
    expect(storageUtil.taskStore.unsafeModifyInPlace(
        rewriteTask.getTaskId(),

View Full Code Here

0 1 2

TOP

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.