package com.ubikvoip.billing.server.api.workers;

import java.math.BigDecimal;
import java.util.ArrayList;

import junit.framework.TestCase;

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

import com.ubikvoip.billing.server.api.message.Message;
import com.ubikvoip.billing.server.api.message.reply.FailureReply;
import com.ubikvoip.billing.server.api.message.reply.SuccessReply;
import com.ubikvoip.billing.server.api.message.request.AddTransferRequest;
import com.ubikvoip.billing.server.client.Client;
import com.ubikvoip.billing.server.client.ClientProviderFactory;
import com.ubikvoip.billing.server.com.APIClientTestImpl;
import com.ubikvoip.billing.server.config.ConfigFactory;
import com.ubikvoip.billing.server.daemon.ApiDaemon;
import com.ubikvoip.billing.server.db.CustomPGSQLTest;
import com.ubikvoip.billing.server.db.DatabaseHelper;
import com.ubikvoip.billing.server.invoice.service.BillingService;
import com.ubikvoip.billing.server.invoice.service.TransferService.Transfer;
import com.ubikvoip.billing.server.mail.MailServerTest;
import com.ubikvoip.billing.server.operation.Operation;
import com.ubikvoip.billing.server.session.BillingSessionProviderFactory;
import com.ubikvoip.billing.server.utils.Hash;

public class TestAddTransferRequestWorker extends TestCase
{

    private static final String TEST_HOST = "localhost";
    private static final int TEST_PORT = 1337;
    private static final String TEST_CLIENT_ID = "JUNIT";
    private static final String TEST_PASSWORD = "JUNIT_PASSWORD";

    private APIClientTestImpl apiClient = null;

    @Before
    protected void setUp()
    {
        try
        {
            DatabaseHelper.bootstrap();
            MailServerTest.reset();
            ApiDaemon.doStart();

            if (this.apiClient != null)
            {
                this.apiClient.close();
                this.apiClient = null;
            }

            this.apiClient = new APIClientTestImpl(TEST_HOST, TEST_PORT, TEST_CLIENT_ID, Hash.getMD5(TEST_PASSWORD));

            CustomPGSQLTest t = CustomPGSQLTest.getInstance();
            t.customQuery("DELETE FROM billing_session");
            t.customQuery("DELETE FROM users");
            t.customQuery("DELETE FROM client");
            t.customQuery("DELETE FROM service_client_rate");

            Client client = new Client();
            client.setId(TEST_CLIENT_ID);
            ClientProviderFactory.getInstance().addClient(client);

            BillingSessionProviderFactory.getInstance().addUser(
                    TEST_CLIENT_ID, Hash.getMD5(TEST_PASSWORD), TEST_CLIENT_ID);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }

    }

    @After
    protected void tearDown()
    {
        try
        {
            MailServerTest.reset();

            if (this.apiClient != null)
            {
                this.apiClient.close();
                this.apiClient = null;
            }

            CustomPGSQLTest t = CustomPGSQLTest.getInstance();
            t.customQuery("DELETE FROM billing_session");
            t.customQuery("DELETE FROM users");
            t.customQuery("DELETE FROM client");
            t.customQuery("DELETE FROM service_client_rate");

            ConfigFactory.getInstance().clear();

        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }

    @Test
    public void testWithNoData()
    {
        try
        {
            String session_id = this.apiClient.doInitConnection();

            AddTransferRequest request = new AddTransferRequest();
            request.setTransferList(new ArrayList<Transfer>());

            request.setSessionID(session_id);

            ArrayList<Class<? extends Message>> expected_replies = new ArrayList<>();

            expected_replies.add(SuccessReply.class);

            Message reply = this.apiClient.sendToServerAndWaitForReply(request, expected_replies);

            // Nothing to add
            assertTrue(reply instanceof SuccessReply);

            this.apiClient.doClose(session_id);

            assertEquals(0, MailServerTest.getReceivedMessages().length);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }

    @Test
    public void testWithNoRate()
    {
        try
        {
            String session_id = this.apiClient.doInitConnection();

            ArrayList<Transfer> transferList = new ArrayList<>();
            transferList.add(new Transfer("300", "4182661111"));
            AddTransferRequest request = new AddTransferRequest();
            request.setTransferList(transferList);

            request.setSessionID(session_id);

            ArrayList<Class<? extends Message>> expected_replies = new ArrayList<>();

            expected_replies.add(FailureReply.class);

            Message reply = this.apiClient.sendToServerAndWaitForReply(request, expected_replies);

            // Nothing to add
            assertTrue(reply instanceof FailureReply);

            this.apiClient.doClose(session_id);

            assertEquals(0, MailServerTest.getReceivedMessages().length);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }

    @Test
    public void testSingleTransfer()
    {
        try
        {
            ClientProviderFactory.getInstance().addRate(
                    TEST_CLIENT_ID, BillingService.TRANSFER, Operation.ADD, "", new BigDecimal(0));

            String session_id = this.apiClient.doInitConnection();

            ArrayList<Transfer> transferList = new ArrayList<>();
            transferList.add(new Transfer("300", "4182661111"));
            AddTransferRequest request = new AddTransferRequest();
            request.setTransferList(transferList);

            request.setSessionID(session_id);

            ArrayList<Class<? extends Message>> expected_replies = new ArrayList<>();

            expected_replies.add(SuccessReply.class);

            Message reply = this.apiClient.sendToServerAndWaitForReply(request, expected_replies);

            // Nothing to add
            assertTrue(reply instanceof SuccessReply);

            this.apiClient.doClose(session_id);

            assertEquals(0, MailServerTest.getReceivedMessages().length);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }

    @Test
    public void testAddMulti()
    {
        try
        {
            ClientProviderFactory.getInstance().addRate(
                    TEST_CLIENT_ID, BillingService.TRANSFER, Operation.ADD, "", new BigDecimal(0));

            String session_id = this.apiClient.doInitConnection();

            ArrayList<Transfer> transferList = new ArrayList<>();
            transferList.add(new Transfer("300", "4182661111"));
            transferList.add(new Transfer("400", "4182661111"));
            transferList.add(new Transfer("300", "4443332211"));
            transferList.add(new Transfer("500", "4182661111"));

            AddTransferRequest request = new AddTransferRequest();
            request.setTransferList(transferList);

            request.setSessionID(session_id);

            ArrayList<Class<? extends Message>> expected_replies = new ArrayList<>();

            expected_replies.add(SuccessReply.class);

            Message reply = this.apiClient.sendToServerAndWaitForReply(request, expected_replies);

            // Nothing to add
            assertTrue(reply instanceof SuccessReply);

            this.apiClient.doClose(session_id);

            assertEquals(0, MailServerTest.getReceivedMessages().length);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }

    @Test
    public void testAddDuplicate()
    {
        try
        {
            ClientProviderFactory.getInstance().addRate(
                    TEST_CLIENT_ID, BillingService.TRANSFER, Operation.ADD, "", new BigDecimal(0));

            String session_id = this.apiClient.doInitConnection();

            ArrayList<Transfer> transferList = new ArrayList<>();
            transferList.add(new Transfer("300", "4182661111"));
            transferList.add(new Transfer("300", "4182661111"));

            AddTransferRequest request = new AddTransferRequest();
            request.setTransferList(transferList);

            request.setSessionID(session_id);

            ArrayList<Class<? extends Message>> expected_replies = new ArrayList<>();

            expected_replies.add(FailureReply.class);

            Message reply = this.apiClient.sendToServerAndWaitForReply(request, expected_replies);

            // Nothing to add
            assertTrue(reply instanceof FailureReply);

            this.apiClient.doClose(session_id);

            assertEquals(0, MailServerTest.getReceivedMessages().length);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }

    @Test
    public void testWithInvalidSession()
    {
        try
        {
            ClientProviderFactory.getInstance().addRate(
                    TEST_CLIENT_ID, BillingService.TRANSFER, Operation.ADD, "", new BigDecimal(0));

            ArrayList<Transfer> transferList = new ArrayList<>();
            transferList.add(new Transfer("300", "4182661111"));
            transferList.add(new Transfer("400", "4182661111"));

            AddTransferRequest request = new AddTransferRequest();
            request.setTransferList(transferList);

            ArrayList<Class<? extends Message>> expected_replies = new ArrayList<>();

            expected_replies.add(FailureReply.class);

            Message reply = this.apiClient.sendToServerAndWaitForReply(request, expected_replies);

            // Nothing to add
            assertTrue(reply instanceof FailureReply);

            assertEquals(0, MailServerTest.getReceivedMessages().length);
        }
        catch (Exception e)
        {
            fail(e.getMessage());
        }
    }
}