@Override
  public void registerModified(Weapon weapon) {
    LOGGER.info("Registering {} for modify in context.", weapon.getName());
    register(weapon, UnitActions.MODIFY.getActionValue());

  }
@Test
  void shouldSaveModifiedStudentWithoutWritingToDb() {
    armsDealer.registerModified(weapon1);
    armsDealer.registerModified(weapon2);

    assertEquals(2, context.get(UnitActions.MODIFY.getActionValue()).size());
    verifyNoMoreInteractions(weaponDatabase);
  }