package com.tdudzik.pokergame;

import static org.mockito.Mockito.*;
import static org.junit.Assert.*;
import org.junit.Test;

import com.tdudzik.deck.Card;
import com.tdudzik.deck.Rank;
import com.tdudzik.deck.Suit;
import java.util.ArrayList;
import java.util.List;

public class HandTest {

    @Test
    public void createHandWithCategory__HIGH_CARD() {
        Hand hand = new Hand(getCardMocks__HIGH_CARD());

        assertEquals(HandCategory.HIGH_CARD, hand.getCategory());
    }

    private List<Card> getCardMocks__HIGH_CARD() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.SEVEN, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.TWO, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.FIVE, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__ONE_PAIR() {
        Hand hand = new Hand(getCardMocks__ONE_PAIR());

        assertEquals(HandCategory.ONE_PAIR, hand.getCategory());
    }

    private List<Card> getCardMocks__ONE_PAIR() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.TWO, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.FIVE, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__TWO_PAIR() {
        Hand hand = new Hand(getCardMocks__TWO_PAIR());

        assertEquals(HandCategory.TWO_PAIR, hand.getCategory());
    }

    private List<Card> getCardMocks__TWO_PAIR() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.FIVE, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__THREE_OF_A_KIND() {
        Hand hand = new Hand(getCardMocks__THREE_OF_A_KIND());

        assertEquals(HandCategory.THREE_OF_A_KIND, hand.getCategory());
    }

    private List<Card> getCardMocks__THREE_OF_A_KIND() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.FIVE, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__STRAIGHT() {
        Hand hand = new Hand(getCardMocks__STRAIGHT());

        assertEquals(HandCategory.STRAIGHT, hand.getCategory());
    }

    private List<Card> getCardMocks__STRAIGHT() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.THREE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.FOUR, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.FIVE, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.SIX, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.SEVEN, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__FLUSH() {
        Hand hand = new Hand(getCardMocks__FLUSH());

        assertEquals(HandCategory.FLUSH, hand.getCategory());
    }

    private List<Card> getCardMocks__FLUSH() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.THREE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.FOUR, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.QUEEN, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.TWO, Suit.CLUBS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__FULL_HOUSE() {
        Hand hand = new Hand(getCardMocks__FULL_HOUSE());

        assertEquals(HandCategory.FULL_HOUSE, hand.getCategory());
    }

    private List<Card> getCardMocks__FULL_HOUSE() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__FOUR_OF_A_KIND() {
        Hand hand = new Hand(getCardMocks__FOUR_OF_A_KIND());

        assertEquals(HandCategory.FOUR_OF_A_KIND, hand.getCategory());
    }

    private List<Card> getCardMocks__FOUR_OF_A_KIND() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.HEARTS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.DIAMONDS));
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.SPADES));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.HEARTS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__STRAIGHT_FLUSH() {
        Hand hand = new Hand(getCardMocks__STRAIGHT_FLUSH());

        assertEquals(HandCategory.STRAIGHT_FLUSH, hand.getCategory());
    }

    private List<Card> getCardMocks__STRAIGHT_FLUSH() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.QUEEN, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.JACK, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.TEN, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.NINE, Suit.CLUBS));

        return listOfCardMocks;
    }

    @Test
    public void createHandWithCategory__ROYAL_FLUSH() {
        Hand hand = new Hand(getCardMocks__ROYAL_FLUSH());

        assertEquals(HandCategory.ROYAL_FLUSH, hand.getCategory());
    }

    private List<Card> getCardMocks__ROYAL_FLUSH() {
        List<Card> listOfCardMocks = new ArrayList();
        listOfCardMocks.add(getCardMock(Rank.ACE, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.KING, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.QUEEN, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.JACK, Suit.CLUBS));
        listOfCardMocks.add(getCardMock(Rank.TEN, Suit.CLUBS));

        return listOfCardMocks;
    }

    private Card getCardMock(Rank rank, Suit suit) {
        Card cardMock = mock(Card.class);
        when(cardMock.getRank()).thenReturn(rank);
        when(cardMock.getSuit()).thenReturn(suit);
        when(cardMock.getRankValue()).thenReturn(rank.getValue());
        when(cardMock.compareTo(any(Card.class))).thenCallRealMethod();

        return cardMock;
    }

}