PDEngine/src/test/java/org/rulez/demokracia/pdengine/testhelpers/CastVoteTestHelper.java at 8185adba9a198c45fdec0d2fc16d521c4c02715f · szirbucz/PDEngine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
package org.rulez.demokracia.pdengine.testhelpers;

import java.util.List;

import org.rulez.demokracia.pdengine.choice.Choice;
import org.rulez.demokracia.pdengine.choice.RankedChoice;
import org.rulez.demokracia.pdengine.vote.Vote;
import org.rulez.demokracia.pdengine.votecast.CastVote;

public class CastVoteTestHelper {

  public static final Choice CHOICE_A = new Choice("A", "user");
  public static final Choice CHOICE_B = new Choice("B", "user");
  public static final List<RankedChoice> RANKED_CHOICES =
      List.of(new RankedChoice(CHOICE_A.getId(), 1));

  public static void fillVoteWithDummyCastVotes(final Vote vote) {
    vote.addChoice(CHOICE_A);
    vote.addChoice(CHOICE_B);
    vote.getVotesCast().add(new CastVote("user1", RANKED_CHOICES));
    vote.getVotesCast().add(new CastVote("user2", RANKED_CHOICES));
    vote.getVotesCast().add(new CastVote(null, RANKED_CHOICES));
    vote.getVotesCast().add(new CastVote("user3", RANKED_CHOICES));
  }

}