/**
 * =============================================================================
 *
 * ORCID (R) Open Source
 * http://orcid.org
 *
 * Copyright (c) 2012-2014 ORCID, Inc.
 * Licensed under an MIT-Style License (MIT)
 * http://orcid.org/open-source-license
 *
 * This copyright and license information (including a link to the full license)
 * shall be included in its entirety in all copies or substantial portion of
 * the software.
 *
 * =============================================================================
 */
package org.orcid.persistence.dao;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.Date;
import java.util.List;
import java.util.Set;

import javax.annotation.Resource;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.orcid.jaxb.model.notification.amended_v2.AmendedSection;
import org.orcid.jaxb.model.notification_v2.NotificationType;
import org.orcid.persistence.jpa.entities.NotificationAddItemsEntity;
import org.orcid.persistence.jpa.entities.NotificationAmendedEntity;
import org.orcid.persistence.jpa.entities.NotificationCustomEntity;
import org.orcid.persistence.jpa.entities.NotificationEntity;
import org.orcid.persistence.jpa.entities.NotificationItemEntity;
import org.orcid.persistence.jpa.entities.ProfileEntity;
import org.orcid.test.DBUnitTest;
import org.orcid.test.OrcidJUnit4ClassRunner;
import org.orcid.utils.DateUtils;
import org.springframework.test.annotation.Rollback;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.transaction.annotation.Transactional;

@RunWith(OrcidJUnit4ClassRunner.class)
@ContextConfiguration(locations = { "classpath:orcid-persistence-context.xml" })
@Transactional
public class NotificationDaoTest extends DBUnitTest {

    @Resource
    private NotificationDao notificationDao;

    private static final List<String> DATA_FILES = Arrays.asList("/data/SecurityQuestionEntityData.xml", "/data/SourceClientDetailsEntityData.xml",
            "/data/ProfileEntityData.xml", "/data/WorksEntityData.xml", "/data/ClientDetailsEntityData.xml",
            "/data/Oauth2TokenDetailsData.xml", "/data/WebhookEntityData.xml", "/data/NotificationEntityData.xml");

    @BeforeClass
    public static void initDBUnitData() throws Exception {
        initDBUnitData(DATA_FILES);
    }

    @AfterClass
    public static void removeDBUnitData() throws Exception {
        List<String> reversedDataFiles = new ArrayList<String>(DATA_FILES);
        Collections.reverse(reversedDataFiles);
        removeDBUnitData(reversedDataFiles);
    }

    @Test
    public void testFindCustomNotification() {
        NotificationEntity notification = notificationDao.find(1L);
        assertNotNull(notification);
        assertTrue(notification instanceof NotificationCustomEntity);
        assertEquals(NotificationType.CUSTOM, notification.getNotificationType());
    }
    
    @Test
    public void testFindPermissionNotification() {
        NotificationEntity notification = notificationDao.find(5L);
        assertNotNull(notification);
        assertTrue(notification instanceof NotificationAddItemsEntity);
        assertEquals(NotificationType.PERMISSION, notification.getNotificationType());
        NotificationAddItemsEntity addActsNotification = (NotificationAddItemsEntity) notification;
        Set<NotificationItemEntity> acts = addActsNotification.getNotificationItems();
        assertNotNull(acts);
        assertEquals(2, acts.size());
    }

    @Test
    @Rollback(true)
    public void testFindOrcidsWithNotificationsToSend() {
        List<String> orcids = notificationDao.findOrcidsWithNotificationsToSend();
        assertNotNull(orcids);
        assertEquals(1, orcids.size());
        assertEquals("4444-4444-4444-4441", orcids.get(0));
    }

    @Test
    @Rollback(true)
    public void testFindOrcidsWithNotificationsToSendWhenTooSoon() {
        List<String> orcids = notificationDao.findOrcidsWithNotificationsToSend(DateUtils.convertToDate("2014-07-16T12:00:00"));
        assertNotNull(orcids);
        assertEquals(0, orcids.size());
    }
    
    @Test
    public void testFindPermissionByOrcidAndClient() {
    	List<NotificationEntity> entities = notificationDao.findPermissionsByOrcidAndClient("4444-4444-4444-4442", "4444-4444-4444-4445", 0, 10);
    	assertNotNull(entities);
    	assertEquals(2, entities.size());
    	
    	entities = notificationDao.findPermissionsByOrcidAndClient("4444-4444-4444-4441", "4444-4444-4444-4445", 0, 10);
    	assertNotNull(entities);
    	assertEquals(3, entities.size());
    	
     	entities = notificationDao.findPermissionsByOrcidAndClient("4444-4444-4444-4441", "4444-4444-4444-4441", 0, 10);
    	assertNotNull(entities);
    	assertTrue(entities.isEmpty());
    	
    	entities = notificationDao.findPermissionsByOrcidAndClient("4444-4444-4444-4442", "4444-4444-4444-4441", 0, 10);
     	assertNotNull(entities);
    	assertEquals(1, entities.size());
    	
    }
    
    @Test
    public void testFindLatestByOrcid() {
        NotificationEntity entity = notificationDao.findLatestByOrcid("0000-0000-0000-0003");
        assertNull(entity);
        Long lastId = null;
        for(int i = 0; i < 5; i++) {
            Date now = new Date();
            NotificationAmendedEntity newEntity = new NotificationAmendedEntity();
            newEntity.setAmendedSection(AmendedSection.UNKNOWN);
            newEntity.setClientSourceId("APP-6666666666666666");
            newEntity.setDateCreated(now);
            newEntity.setLastModified(now);
            newEntity.setNotificationIntro("Intro");
            newEntity.setNotificationSubject("Subject");
            newEntity.setNotificationType(NotificationType.AMENDED);
            newEntity.setProfile(new ProfileEntity("0000-0000-0000-0003"));
            newEntity.setSendable(true);
            notificationDao.persist(newEntity);
            
            NotificationEntity freshFromDB = notificationDao.findLatestByOrcid("0000-0000-0000-0003");
            assertNotNull(freshFromDB);
            if(lastId == null) {
                lastId = freshFromDB.getId();
            } else {
                assertTrue(lastId < freshFromDB.getId());
                lastId = freshFromDB.getId();
            }
        }                
    }
}