package jrummikub.control;

import static org.junit.Assert.*;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import jrummikub.model.MockHand;
import jrummikub.model.MockTable;
import jrummikub.model.Position;
import jrummikub.model.Stone;
import jrummikub.model.StoneColor;
import jrummikub.model.StoneSet;
import jrummikub.util.Event;
import jrummikub.util.IEvent;
import jrummikub.util.IListener;
import jrummikub.util.Pair;
import jrummikub.view.MockView;

import org.junit.Before;
import org.junit.Test;

import static jrummikub.model.StoneColor.*;

public class TurnControlTest {

	class MockTimer implements ITurnTimer {
		public boolean timerRunning = false;
		public Event timeRunOutEvent = new Event();

		@Override
		public void startTimer() {
			timerRunning = true;
		}

		@Override
		public void stopTimer() {
			timerRunning = false;
		}

		@Override
		public IEvent getTimeRunOutEvent() {
			return timeRunOutEvent;
		}

	}

	TurnControl testControl;
	MockView mockView;
	MockTimer mockTimer;
	MockTable mockTable;
	MockHand mockHand;
	boolean eventFired;

	@Before
	public void setUp() {
		mockView = new MockView();
		mockTimer = new MockTimer();
		mockTable = new MockTable();
		mockHand = new MockHand();
		testControl = new TurnControl(mockHand, mockTable, mockView, mockTimer);
		testControl.startTurn();
	}

	@Test
	public void startTimer() {
		assertTrue(mockTimer.timerRunning);
	}

	@SuppressWarnings("unchecked")
	@Test
	public void showInitialHand() {
		mockView = new MockView();
		mockTimer = new MockTimer();
		mockTable = new MockTable();
		mockHand = new MockHand();

		mockView.displayStartTurnPanel = true;

		List<Pair<Stone, Position>> stones = Arrays
				.asList(new Pair<Stone, Position>(new Stone(RED), new Position(
						0, 0)), new Pair<Stone, Position>(new Stone(BLACK),
						new Position(1, 0)));

		mockHand.iterable = stones;

		testControl = new TurnControl(mockHand, mockTable, mockView, mockTimer);

		testControl.startTurn();

		int i = 0;
		for (Pair<Stone, Position> pair : mockView.playerPanel.handPanel.stones) {
			assertSame(stones.get(i), pair);
			i++;
		}
		assertEquals(stones.size(), i);

		assertFalse(mockView.displayStartTurnPanel);
	}

	@Test
	public void viewEndOfTurn() {
		eventFired = false;
		mockTimer.timerRunning = true;

		testControl.getEndOfTurnEvent().add(new IListener() {

			@Override
			public void handle() {
				eventFired = true;
			}
		});

		mockView.playerPanel.endTurnEvent.emit();

		assertTrue(eventFired);
		assertFalse(mockTimer.timerRunning);
		assertTrue(mockView.playerPanel.endTurnEvent.listeners.isEmpty());
	}

	@Test
	public void timerEndOfTurn() {
		eventFired = false;
		mockTimer.timerRunning = true;

		testControl.getEndOfTurnEvent().add(new IListener() {

			@Override
			public void handle() {
				eventFired = true;
			}
		});

		mockTimer.timeRunOutEvent.emit();

		assertTrue(eventFired);
		assertFalse(mockTimer.timerRunning);
	}

	@Test
	public void deselctOnEndOfTurn() {

		Stone firstStone = new Stone(StoneColor.RED);

		// Select first stone
		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, false);
		mockTimer.timeRunOutEvent.emit();

		assertCollection(new ArrayList<Stone>());
	}

	@Test
	public void selectStoneInHand() {

		Stone firstStone = new Stone(StoneColor.RED);

		// Select first stone
		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, false);

		assertCollection(Arrays.asList(firstStone));

		// Select second stone
		Stone secondStone = new Stone(StoneColor.BLACK);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(secondStone, false);

		assertCollection(Arrays.asList(secondStone));

	}

	@Test
	public void collectStoneInHand() {

		Stone firstStone = new Stone(StoneColor.RED);

		// Select first stone
		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, true);

		assertCollection(Arrays.asList(firstStone));

		// Select second stone
		Stone secondStone = new Stone(StoneColor.BLACK);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(secondStone, true);

		assertCollection(Arrays.asList(firstStone, secondStone));

		// De-select first stone
		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, true);

		assertCollection(Arrays.asList(secondStone));
	}

	@Test
	public void deselectStoneInCollection() {
		Stone firstStone = new Stone(StoneColor.RED);
		Stone secondStone = new Stone(StoneColor.BLACK);

		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, true);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(secondStone, true);

		mockView.tablePanel.stoneCollectionPanel.stoneClickEvent.emit(
				firstStone, false);

		assertCollection(Arrays.asList(secondStone));
	}

	@Test
	public void reorderCollection() {

		Stone firstStone = new Stone(StoneColor.RED);
		Stone secondStone = new Stone(StoneColor.BLACK);

		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, true);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(secondStone, true);

		mockView.tablePanel.stoneCollectionPanel.stoneClickEvent.emit(
				firstStone, true);

		assertCollection(Arrays.asList(secondStone, firstStone));
	}

	@Test
	public void deselectWholeCollection() {

		Stone firstStone = new Stone(StoneColor.RED);
		Stone secondStone = new Stone(StoneColor.BLACK);

		mockView.playerPanel.handPanel.stoneClickEvent.emit(firstStone, true);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(secondStone, true);

		mockView.tablePanel.stoneCollectionPanel.setClickEvent.emit(firstStone,
				true);

		assertCollection(new ArrayList<Stone>());
	}

	@Test
	public void selectStoneOnTable() {

		Stone firstStone = new Stone(StoneColor.RED);

		// Select first stone
		mockView.tablePanel.stoneClickEvent.emit(firstStone, false);

		assertCollection(Arrays.asList(firstStone));

		// Select second stone
		Stone secondStone = new Stone(StoneColor.BLACK);
		mockView.tablePanel.stoneClickEvent.emit(secondStone, false);

		assertCollection(Arrays.asList(secondStone));

	}

	@Test
	public void collectStoneOnTable() {

		Stone firstStone = new Stone(StoneColor.RED);

		// Select first stone
		mockView.tablePanel.stoneClickEvent.emit(firstStone, true);

		assertCollection(Arrays.asList(firstStone));

		// Select second stone
		Stone secondStone = new Stone(StoneColor.BLACK);
		mockView.tablePanel.stoneClickEvent.emit(secondStone, true);

		assertCollection(Arrays.asList(firstStone, secondStone));

		// De-select first stone
		mockView.tablePanel.stoneClickEvent.emit(firstStone, true);

		assertCollection(Arrays.asList(secondStone));
	}

	@Test
	public void selectSetOnTable() {

		Stone stone1 = new Stone(StoneColor.RED);
		Stone stone2 = new Stone(StoneColor.BLACK);
		StoneSet set1 = new StoneSet(Arrays.asList(stone1, stone2));
		Stone stone3 = new Stone(1, StoneColor.RED);
		Stone stone4 = new Stone(1, StoneColor.BLACK);
		StoneSet set2 = new StoneSet(Arrays.asList(stone3, stone4));

		mockTable.findStoneSet.put(stone1, set1);
		mockTable.findStoneSet.put(stone4, set2);

		mockView.tablePanel.setClickEvent.emit(stone1, false);
		assertCollection(Arrays.asList(stone1, stone2));
		mockView.tablePanel.setClickEvent.emit(stone4, false);
		assertCollection(Arrays.asList(stone3, stone4));
	}

	@Test
	public void collectSetOnTable() {

		Stone stone1 = new Stone(StoneColor.RED);
		Stone stone2 = new Stone(StoneColor.BLACK);
		StoneSet set1 = new StoneSet(Arrays.asList(stone1, stone2));
		Stone stone3 = new Stone(1, StoneColor.RED);
		Stone stone4 = new Stone(1, StoneColor.BLACK);
		StoneSet set2 = new StoneSet(Arrays.asList(stone3, stone4));

		mockTable.findStoneSet.put(stone1, set1);
		mockTable.findStoneSet.put(stone4, set2);

		mockView.tablePanel.setClickEvent.emit(stone1, true);
		assertCollection(Arrays.asList(stone1, stone2));
		mockView.tablePanel.setClickEvent.emit(stone4, true);
		assertCollection(Arrays.asList(stone1, stone2, stone3, stone4));
	}

	@Test
	public void rangeSelectOnTable() {

		Stone stone1 = new Stone(1, StoneColor.RED);
		Stone stone2 = new Stone(2, StoneColor.RED);
		Stone stone3 = new Stone(3, StoneColor.RED);
		Stone stone4 = new Stone(4, StoneColor.RED);
		StoneSet set1 = new StoneSet(Arrays.asList(stone1, stone2, stone3,
				stone4));

		mockTable.findStoneSet.put(stone1, set1);
		mockTable.findStoneSet.put(stone3, set1);

		mockView.tablePanel.stoneClickEvent.emit(stone1, false);
		mockView.tablePanel.rangeClickEvent.emit(stone3, true);

		assertCollection(Arrays.asList(stone1, stone2, stone3));

	}

	@Test
	public void rangeCollectOnTable() {
		Stone extraStone = new Stone(StoneColor.RED);

		Stone stone1 = new Stone(1, StoneColor.RED);
		Stone stone2 = new Stone(2, StoneColor.RED);
		Stone stone3 = new Stone(3, StoneColor.RED);
		Stone stone4 = new Stone(4, StoneColor.RED);
		StoneSet set1 = new StoneSet(Arrays.asList(stone1, stone2, stone3,
				stone4));

		mockTable.findStoneSet.put(stone1, set1);
		mockTable.findStoneSet.put(stone3, set1);

		mockView.tablePanel.stoneClickEvent.emit(extraStone, false);

		mockView.tablePanel.stoneClickEvent.emit(stone1, true);
		mockView.tablePanel.rangeClickEvent.emit(stone3, false);

		assertCollection(Arrays.asList(extraStone, stone1, stone2, stone3));
	}

	@Test
	public void rangeFailSelect() {
		Stone stone1 = new Stone(1, StoneColor.RED);
		Stone stone2 = new Stone(2, StoneColor.RED);
		StoneSet set1 = new StoneSet(Arrays.asList(stone1));
		StoneSet set2 = new StoneSet(Arrays.asList(stone2));

		mockTable.findStoneSet.put(stone1, set1);
		mockTable.findStoneSet.put(stone2, set2);

		// Select first stone
		mockView.tablePanel.stoneClickEvent.emit(stone1, false);

		assertCollection(Arrays.asList(stone1));

		// Select second stone
		mockView.tablePanel.rangeClickEvent.emit(stone2, false);

		assertCollection(Arrays.asList(stone2));

	}

	@Test
	public void rangeFailCollect() {
		Stone stone1 = new Stone(1, StoneColor.RED);
		Stone stone2 = new Stone(2, StoneColor.RED);
		StoneSet set1 = new StoneSet(Arrays.asList(stone1));
		StoneSet set2 = new StoneSet(Arrays.asList(stone2));

		mockTable.findStoneSet.put(stone1, set1);
		mockTable.findStoneSet.put(stone2, set2);

		// Select first stone
		mockView.tablePanel.stoneClickEvent.emit(stone1, true);

		assertCollection(Arrays.asList(stone1));

		// Select second stone
		mockView.tablePanel.rangeClickEvent.emit(stone2, true);

		assertCollection(Arrays.asList(stone1, stone2));
	}

	private void assertCollection(List<Stone> expected) {
		ArrayList<Stone> selectedStones = new ArrayList<Stone>(
				mockView.selectedStones);
		ArrayList<Stone> expectedStones = new ArrayList<Stone>(expected);
		assertEquals(expectedStones, selectedStones);
	}

	@Test
	public void testAddLeft() {
		Stone blueOne = new Stone(1, BLUE);
		Stone redOne = new Stone(1, RED);
		Stone blackOne = new Stone(1, BLACK);
		Stone blueTwo = new Stone(2, BLUE);
		Stone blueThree = new Stone(3, BLUE);
		Stone blueFour = new Stone(4, BLUE);
		Stone redTwo = new Stone(2, RED);
		Stone redThree = new Stone(3, RED);
		Stone redFour = new Stone(4, RED);
		Stone blackTwo = new Stone(2, BLACK);
		Stone blackThree = new Stone(3, BLACK);
		Stone blackFour = new Stone(4, BLACK);
		Stone blackFive = new Stone(5, BLACK);
		StoneSet oldSet1 = new StoneSet(Arrays.asList(blueOne, redOne,
				blackOne, redTwo, redThree, redFour, blackTwo, blackThree));
		StoneSet oldSet2 = new StoneSet(Arrays.asList(blueTwo, blackFour,
				blackFive));
		mockTable.drop(oldSet1, new Position(0, 0));
		mockTable.drop(oldSet2, new Position(0, 0));
		mockHand.drop(blueThree, new Position(0, 0));
		mockHand.drop(blueFour, new Position(0, 0));
		mockView.playerPanel.handPanel.stoneClickEvent.emit(blueThree, false);
		mockView.tablePanel.stoneClickEvent.emit(redOne, true);
		mockView.tablePanel.stoneClickEvent.emit(redThree, true);
		mockView.tablePanel.leftConnectorClickEvent.emit(oldSet1);
		// handcheck
		assertEquals(1, mockHand.getSize());
		assertSame(mockHand.stones.get(0), blueFour);
		// tablecheck
		assertEquals(2, mockTable.getSize());
		StoneSet newSet1, newSet2;
		if (mockTable.sets.get(0).getFirst().size() == 3) {
			newSet2 = mockTable.sets.get(0).getFirst();
			newSet1 = mockTable.sets.get(1).getFirst();
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			newSet2 = mockTable.sets.get(1).getFirst();
		}
		assertSame(oldSet2, newSet2);
		// setcheck
		assertEquals(9, newSet1.size());
		assertSame(newSet1.get(0), blueThree);
		assertSame(newSet1.get(1), redOne);
		assertSame(newSet1.get(2), redThree);
		assertSame(newSet1.get(3), blueOne);
		assertSame(newSet1.get(4), blackOne);
		assertSame(newSet1.get(5), redTwo);
		assertSame(newSet1.get(6), redFour);
		assertSame(newSet1.get(7), blackTwo);
		assertSame(newSet1.get(8), blackThree);

		mockView.tablePanel.stoneClickEvent.emit(redOne, true);
		mockView.tablePanel.stoneClickEvent.emit(redThree, true);
		mockView.tablePanel.leftConnectorClickEvent.emit(oldSet2);
		// handcheck
		assertEquals(1, mockHand.getSize());
		assertSame(mockHand.stones.get(0), blueFour);
		// tablecheck
		assertEquals(2, mockTable.getSize());
		if (mockTable.sets.get(0).getFirst().size() == 5) {
			newSet2 = mockTable.sets.get(0).getFirst();
			newSet1 = mockTable.sets.get(1).getFirst();
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			newSet2 = mockTable.sets.get(1).getFirst();
		}
		// setcheck1
		assertEquals(7, newSet1.size());
		assertSame(newSet1.get(0), blueThree);
		assertSame(newSet1.get(1), blueOne);
		assertSame(newSet1.get(2), blackOne);
		assertSame(newSet1.get(3), redTwo);
		assertSame(newSet1.get(4), redFour);
		assertSame(newSet1.get(5), blackTwo);
		assertSame(newSet1.get(6), blackThree);
		// setcheck2
		assertEquals(5, newSet2.size());
		assertSame(newSet2.get(0), redOne);
		assertSame(newSet2.get(1), redThree);
		assertSame(newSet2.get(2), blueTwo);
		assertSame(newSet2.get(3), blackFour);
		assertSame(newSet2.get(4), blackFive);
		// versuche, links was wegzunehmen und wieder anzuhängen
		mockView.tablePanel.stoneClickEvent.emit(redOne, false);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(blueFour, true);
		// handcheck
		assertEquals(0, mockHand.getSize());
		// tablecheck
		assertEquals(2, mockTable.getSize());
		if (mockTable.sets.get(0).getFirst().size() == 6) {
			newSet2 = mockTable.sets.get(0).getFirst();
			newSet1 = mockTable.sets.get(1).getFirst();
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			newSet2 = mockTable.sets.get(1).getFirst();
		}
		// setcheck1
		assertEquals(7, newSet1.size());
		// setcheck2
		assertEquals(6, newSet2.size());
		assertSame(newSet2.get(0), redOne);
		assertSame(newSet2.get(1), blueFour);
		assertSame(newSet2.get(2), redThree);
		assertSame(newSet2.get(3), blueTwo);
		assertSame(newSet2.get(4), blackFour);
		assertSame(newSet2.get(5), blackFive);
	}

	@Test
	public void testAddRight() {
		Stone blueOne = new Stone(1, BLUE);
		Stone redOne = new Stone(1, RED);
		Stone blackOne = new Stone(1, BLACK);
		Stone blueTwo = new Stone(2, BLUE);
		Stone blueThree = new Stone(3, BLUE);
		Stone blueFour = new Stone(4, BLUE);
		Stone redTwo = new Stone(2, RED);
		Stone redThree = new Stone(3, RED);
		Stone redFour = new Stone(4, RED);
		Stone blackTwo = new Stone(2, BLACK);
		Stone blackThree = new Stone(3, BLACK);
		Stone blackFour = new Stone(4, BLACK);
		Stone blackFive = new Stone(5, BLACK);
		StoneSet oldSet1 = new StoneSet(Arrays.asList(blueOne, redOne,
				blackOne, redTwo, redThree, redFour, blackTwo, blackThree));
		StoneSet oldSet2 = new StoneSet(Arrays.asList(blueTwo, blackFour,
				blackFive));
		mockTable.drop(oldSet1, new Position(0, 0));
		mockTable.drop(oldSet2, new Position(0, 0));
		mockHand.drop(blueThree, new Position(0, 0));
		mockHand.drop(blueFour, new Position(0, 0));
		mockView.playerPanel.handPanel.stoneClickEvent.emit(blueThree, false);
		mockView.tablePanel.stoneClickEvent.emit(redOne, true);
		mockView.tablePanel.stoneClickEvent.emit(redThree, true);
		mockView.tablePanel.rightConnectorClickEvent.emit(oldSet1);
		// handcheck
		assertEquals(1, mockHand.getSize());
		assertSame(mockHand.stones.get(0), blueFour);
		// tablecheck
		assertEquals(2, mockTable.getSize());
		StoneSet newSet1, newSet2;
		if (mockTable.sets.get(0).getFirst().size() == 3) {
			newSet2 = mockTable.sets.get(0).getFirst();
			newSet1 = mockTable.sets.get(1).getFirst();
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			newSet2 = mockTable.sets.get(1).getFirst();
		}
		assertSame(oldSet2, newSet2);
		// setcheck
		assertEquals(9, newSet1.size());
		assertSame(newSet1.get(0), blueOne);
		assertSame(newSet1.get(1), blackOne);
		assertSame(newSet1.get(2), redTwo);
		assertSame(newSet1.get(3), redFour);
		assertSame(newSet1.get(4), blackTwo);
		assertSame(newSet1.get(5), blackThree);
		assertSame(newSet1.get(6), blueThree);
		assertSame(newSet1.get(7), redOne);
		assertSame(newSet1.get(8), redThree);

		mockView.tablePanel.stoneClickEvent.emit(redOne, true);
		mockView.tablePanel.stoneClickEvent.emit(redThree, true);
		mockView.tablePanel.rightConnectorClickEvent.emit(oldSet2);
		// handcheck
		assertEquals(1, mockHand.getSize());
		assertSame(mockHand.stones.get(0), blueFour);
		// tablecheck
		assertEquals(2, mockTable.getSize());
		if (mockTable.sets.get(0).getFirst().size() == 5) {
			newSet2 = mockTable.sets.get(0).getFirst();
			newSet1 = mockTable.sets.get(1).getFirst();
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			newSet2 = mockTable.sets.get(1).getFirst();
		}
		// setcheck1
		assertEquals(7, newSet1.size());
		assertSame(newSet1.get(0), blueOne);
		assertSame(newSet1.get(1), blackOne);
		assertSame(newSet1.get(2), redTwo);
		assertSame(newSet1.get(3), redFour);
		assertSame(newSet1.get(4), blackTwo);
		assertSame(newSet1.get(5), blackThree);
		assertSame(newSet1.get(6), blueThree);
		// setcheck2
		assertEquals(5, newSet2.size());
		assertSame(newSet2.get(0), blueTwo);
		assertSame(newSet2.get(1), blackFour);
		assertSame(newSet2.get(2), blackFive);
		assertSame(newSet2.get(3), redOne);
		assertSame(newSet2.get(4), redThree);
		// versuche, rechts was wegzunehmen und wieder anzuhängen
		mockView.tablePanel.stoneClickEvent.emit(redThree, false);
		mockView.playerPanel.handPanel.stoneClickEvent.emit(blueFour, true);
		// handcheck
		assertEquals(0, mockHand.getSize());
		// tablecheck
		assertEquals(2, mockTable.getSize());
		if (mockTable.sets.get(0).getFirst().size() == 6) {
			newSet2 = mockTable.sets.get(0).getFirst();
			newSet1 = mockTable.sets.get(1).getFirst();
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			newSet2 = mockTable.sets.get(1).getFirst();
		}
		// setcheck1
		assertEquals(7, newSet1.size());
		// setcheck2
		assertEquals(6, newSet2.size());
		assertSame(newSet2.get(0), blueTwo);
		assertSame(newSet2.get(1), blackFour);
		assertSame(newSet2.get(2), blackFive);
		assertSame(newSet2.get(3), redOne);
		assertSame(newSet2.get(4), blueFour);
		assertSame(newSet2.get(5), redThree);
	}

	@Test
	public void testAddNewSet() {
		Stone blueOne = new Stone(1, BLUE);
		Stone redOne = new Stone(1, RED);
		Stone blackOne = new Stone(1, BLACK);
		Stone blueTwo = new Stone(2, BLUE);
		Stone blueThree = new Stone(3, BLUE);
		Stone blueFour = new Stone(4, BLUE);
		Stone redTwo = new Stone(2, RED);
		Stone redThree = new Stone(3, RED);
		Stone redFour = new Stone(4, RED);
		Stone blackTwo = new Stone(2, BLACK);
		Stone blackThree = new Stone(3, BLACK);
		Stone blackFour = new Stone(4, BLACK);
		Stone blackFive = new Stone(5, BLACK);
		StoneSet oldSet1 = new StoneSet(Arrays.asList(blueOne, redOne,
				blackOne, redTwo, redThree, redFour, blackTwo, blackThree));
		StoneSet oldSet2 = new StoneSet(Arrays.asList(blueTwo, blackFour,
				blackFive));
		mockTable.drop(oldSet1, new Position(0, 0));
		mockTable.drop(oldSet2, new Position(0, 0));
		mockHand.drop(blueThree, new Position(0, 0));
		mockHand.drop(blueFour, new Position(0, 0));
		mockView.playerPanel.handPanel.stoneClickEvent.emit(blueThree, false);
		mockView.tablePanel.stoneClickEvent.emit(redOne, true);
		mockView.tablePanel.stoneClickEvent.emit(redThree, true);
		mockView.tablePanel.stoneClickEvent.emit(blueTwo, true);
		mockView.tablePanel.clickEvent.emit(new Position(0, 0));
		// handcheck
		assertEquals(1, mockHand.getSize());
		assertSame(blueFour, mockHand.stones.get(0));
		// tablecheck
		StoneSet newSet1, newSet2, newSet3;
		assertEquals(2, mockTable.getSize());
		if (mockTable.sets.get(0).getFirst().size() == 2) {
			newSet2 = mockTable.sets.get(0).getFirst();
			if (mockTable.sets.get(1).getFirst().size() == 4) {
				newSet3 = mockTable.sets.get(1).getFirst();
				newSet1 = mockTable.sets.get(2).getFirst();
			} else {
				newSet3 = mockTable.sets.get(2).getFirst();
				newSet1 = mockTable.sets.get(1).getFirst();
			}
		} else if (mockTable.sets.get(0).getFirst().size() == 4) {
			newSet3 = mockTable.sets.get(0).getFirst();
			if (mockTable.sets.get(1).getFirst().size() == 2) {
				newSet2 = mockTable.sets.get(1).getFirst();
				newSet1 = mockTable.sets.get(2).getFirst();
			} else {
				newSet2 = mockTable.sets.get(2).getFirst();
				newSet1 = mockTable.sets.get(1).getFirst();
			}
		} else {
			newSet1 = mockTable.sets.get(0).getFirst();
			if (mockTable.sets.get(1).getFirst().size() == 2) {
				newSet2 = mockTable.sets.get(1).getFirst();
				newSet3 = mockTable.sets.get(2).getFirst();
			} else {
				newSet2 = mockTable.sets.get(2).getFirst();
				newSet3 = mockTable.sets.get(1).getFirst();
			}
		}

		// setcheck1
		assertEquals(6, newSet1.size());
		assertSame(newSet1.get(0), blueOne);
		assertSame(newSet1.get(1), blackOne);
		assertSame(newSet1.get(2), redTwo);
		assertSame(newSet1.get(3), redFour);
		assertSame(newSet1.get(4), blackTwo);
		assertSame(newSet1.get(5), blackThree);
		// setcheck2
		assertEquals(2, newSet2.size());
		assertSame(newSet2.get(0), blackFour);
		assertSame(newSet2.get(1), blackFive);
		// setcheck1
		assertEquals(4, newSet3.size());
		assertSame(newSet3.get(0), blueThree);
		assertSame(newSet3.get(1), redOne);
		assertSame(newSet3.get(2), redThree);
		assertSame(newSet3.get(3), blueTwo);
	}
}