Package org.apache.aurora.scheduler.storage.entities

Examples of org.apache.aurora.scheduler.storage.entities.IJobUpdate

891
892
893
894
895
896
897
898
899
900
901
  @Test
  public void testAddInstances() throws Exception {
    control.replay();

    IJobUpdate update = makeJobUpdate();
    insertPendingTasks(NEW_CONFIG, ImmutableSet.of(0, 1));

    changeState(JOB, 0, ASSIGNED, STARTING, RUNNING);
    changeState(JOB, 1, ASSIGNED, STARTING, RUNNING);
    clock.advance(WATCH_TIMEOUT);
View Full Code Here
293
294
295
296
297
298
299
300
301
302
303
304
  public void testSaveJobUpdateWithoutEventFailsSelect() {
    final String updateId = "u3";
    storage.write(new MutateWork.NoResult.Quiet() {
      @Override
      public void execute(MutableStoreProvider storeProvider) {
        IJobUpdate update = makeJobUpdate(JOB, updateId);
        storeProvider.getLockStore().saveLock(makeLock(update.getSummary().getJobKey(), "lock1"));
        storeProvider.getJobUpdateStore().saveJobUpdate(update, Optional.of("lock1"));
      }
    });
    assertEquals(Optional.<IJobUpdateDetails>absent(), getUpdateDetails(updateId));
  }
View Full Code Here
922
923
924
925
926
927
928
929
930
931
932
    expectTaskKilled();

    control.replay();

    // Set instance count such that instance 1 is removed.
    IJobUpdate update = setInstanceCount(makeJobUpdate(makeInstanceConfig(0, 1, NEW_CONFIG)), 1);
    insertInitialTasks(update);

    changeState(JOB, 0, ASSIGNED, STARTING, RUNNING);
    changeState(JOB, 1, ASSIGNED, STARTING, RUNNING);
    clock.advance(WATCH_TIMEOUT);
View Full Code Here
365
366
367
368
369
370
371
372
373
374
375
  }

  @Test
  public void testTruncateJobUpdates() {
    String updateId = "u5";
    IJobUpdate update = makeJobUpdate(JOB, updateId);
    IJobUpdateEvent updateEvent = IJobUpdateEvent.build(new JobUpdateEvent(ROLLING_FORWARD, 123L));
    IJobInstanceUpdateEvent instanceEvent = IJobInstanceUpdateEvent.build(
        new JobInstanceUpdateEvent(0, 125L, INSTANCE_ROLLBACK_FAILED));

    saveUpdate(update, Optional.of("lock"));
View Full Code Here
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
    String updateId5 = "u15";
    String updateId6 = "u16";
    String updateId7 = "u17";

    IJobKey job2 = JobKeys.from("testRole2", "testEnv2", "job2");
    IJobUpdate update1 = makeJobUpdate(JOB, updateId1);
    IJobUpdate update2 = makeJobUpdate(JOB, updateId2);
    IJobUpdate update3 = makeJobUpdate(JOB, updateId3);
    IJobUpdate update4 = makeJobUpdate(JOB, updateId4);
    IJobUpdate update5 = makeJobUpdate(job2, updateId5);
    IJobUpdate update6 = makeJobUpdate(job2, updateId6);
    IJobUpdate update7 = makeJobUpdate(job2, updateId7);

    IJobUpdateEvent updateEvent1 = IJobUpdateEvent.build(new JobUpdateEvent(ROLLING_BACK, 123L));
    IJobUpdateEvent updateEvent2 = IJobUpdateEvent.build(new JobUpdateEvent(ABORTED, 124L));
    IJobUpdateEvent updateEvent3 = IJobUpdateEvent.build(new JobUpdateEvent(ROLLED_BACK, 125L));
    IJobUpdateEvent updateEvent4 = IJobUpdateEvent.build(new JobUpdateEvent(FAILED, 126L));
View Full Code Here
495
496
497
498
499
500
501
502
503
504
505
    assertEquals(Optional.of(update1), getUpdate(updateId1)); // active update
  }

  @Test(expected = StorageException.class)
  public void testSaveUpdateWithoutLock() {
    final IJobUpdate update = makeJobUpdate(JOB, "updateId");
    storage.write(new MutateWork.NoResult.Quiet() {
      @Override
      public void execute(MutableStoreProvider storeProvider) {
        storeProvider.getJobUpdateStore().saveJobUpdate(update, Optional.of("lock"));
      }
View Full Code Here
506
507
508
509
510
511
512
513
514
    });
  }

  @Test(expected = StorageException.class)
  public void testSaveTwoUpdatesForOneJob() {
    final IJobUpdate update = makeJobUpdate(JOB, "updateId");
    saveUpdate(update, Optional.of("lock1"));
    saveUpdate(update, Optional.of("lock2"));
  }
View Full Code Here
513
514
515
516
517
518
519
520
521
522
    saveUpdate(update, Optional.of("lock2"));
  }

  @Test(expected = StorageException.class)
  public void testSaveTwoUpdatesSameJobKey() {
    final IJobUpdate update1 = makeJobUpdate(JOB, "update1");
    final IJobUpdate update2 = makeJobUpdate(JOB, "update2");
    saveUpdate(update1, Optional.of("lock1"));
    saveUpdate(update2, Optional.of("lock1"));
  }
View Full Code Here
521
522
523
524
525
526
527
528
529
530
531
    saveUpdate(update2, Optional.of("lock1"));
  }

  @Test
  public void testLockCleared() {
    final IJobUpdate update = makeJobUpdate(JOB, "update1");
    saveUpdate(update, Optional.of("lock1"));

    removeLock(update, "lock1");

    assertEquals(
View Full Code Here
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
  @Test
  public void testGetLockToken() {
    storage.write(new MutateWork.NoResult.Quiet() {
      @Override
      public void execute(MutableStoreProvider storeProvider) {
        final IJobUpdate update1 = makeJobUpdate(JobKeys.from("role", "env", "name1"), "update1");
        final IJobUpdate update2 = makeJobUpdate(JobKeys.from("role", "env", "name2"), "update2");
        saveUpdate(update1, Optional.of("lock1"));
        assertEquals(
            Optional.of("lock1"),
            storeProvider.getJobUpdateStore().getLockToken("update1"));
        assertEquals(NO_TOKEN, storeProvider.getJobUpdateStore().getLockToken("update2"));

        saveUpdate(update2, Optional.of("lock2"));
        assertEquals(
            Optional.of("lock1"),
            storeProvider.getJobUpdateStore().getLockToken("update1"));
        assertEquals(
            Optional.of("lock2"),
            storeProvider.getJobUpdateStore().getLockToken("update2"));

        storeProvider.getLockStore().removeLock(
            makeLock(update1.getSummary().getJobKey(), "lock1").getKey());
        assertEquals(NO_TOKEN, storeProvider.getJobUpdateStore().getLockToken("update1"));
        assertEquals(
            Optional.of("lock2"),
            storeProvider.getJobUpdateStore().getLockToken("update2"));

        storeProvider.getLockStore().removeLock(
            makeLock(update2.getSummary().getJobKey(), "lock2").getKey());
        assertEquals(NO_TOKEN, storeProvider.getJobUpdateStore().getLockToken("update1"));
        assertEquals(NO_TOKEN, storeProvider.getJobUpdateStore().getLockToken("update2"));
      }
    });
  }
View Full Code Here
TOP

Related Classes of org.apache.aurora.scheduler.storage.entities.IJobUpdate

  • org.apache.aurora.scheduler.quota.QuotaManagerImplTest
  • org.apache.aurora.scheduler.storage.db.DBJobUpdateStoreTest
  • org.apache.aurora.scheduler.storage.log.LogStorageTest
  • org.apache.aurora.scheduler.thrift.SchedulerThriftInterface
  • org.apache.aurora.scheduler.thrift.SchedulerThriftInterfaceTest
  • org.apache.aurora.scheduler.updater.JobUpdateControllerImpl
  • org.apache.aurora.scheduler.updater.JobUpdaterIT
    • Copyright © 2018 www.massapicom. All rights reserved.
    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.