### Question:
HitWeighter implements IHitWeighter { @Override public QueueState setHitWeights(QueueState queueState) { for(ProgressionUnit progressionUnit : queueState.getStructure()){ for(ProgressionUnitBar progressionUnitBar : progressionUnit.getProgressionUnitBars()){ for(Hit hitOption : progressionUnitBar.getHitOptions()){ hitOption.setVerticalWeight(WeightClass.OK); } } } return queueState; }  @Override QueueState setHitWeights(QueueState queueState);  }### Answer:
@Test public void shouldSetWeights() { queueState = hitGenerator.setHitOptions(queueState); queueState = hitWeighter.setHitWeights(queueState); AtomicInteger counter = new AtomicInteger(0); for (ProgressionUnit progressionUnit : queueState.getStructure()) { for (ProgressionUnitBar progressionUnitBar : progressionUnit.getProgressionUnitBars()) { for (Hit hitOption : progressionUnitBar.getHitOptions()) { assertThat(hitOption.getVerticalWeight()).isNotNull(); assertThat(hitOption.getVerticalWeight()).isEqualTo(WeightClass.OK); counter.incrementAndGet(); } } } AssertionsForClassTypes.assertThat(counter.get()).isGreaterThan(0); }