/**
 * Copyright (C) 2011  JTalks.org Team
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
package org.jtalks.jcommune.service.transactional;

import org.apache.commons.lang3.RandomStringUtils;
import org.jtalks.common.service.security.SecurityContextFacade;
import org.jtalks.jcommune.model.dao.TopicDraftDao;
import org.jtalks.jcommune.model.entity.Branch;
import org.jtalks.jcommune.model.entity.JCUser;
import org.jtalks.jcommune.model.entity.TopicDraft;
import org.jtalks.jcommune.model.entity.TopicTypeName;
import org.jtalks.jcommune.plugin.api.PluginLoader;
import org.jtalks.jcommune.service.TopicDraftService;
import org.jtalks.jcommune.service.UserService;
import org.mockito.Mock;
import org.springframework.security.access.PermissionEvaluator;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import static org.mockito.MockitoAnnotations.initMocks;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertNull;
import static org.unitils.reflectionassert.ReflectionAssert.assertReflectionEquals;

/**
 * @author Dmitry S. Dolzhenko
 */
public class TransactionalTopicDraftServiceTest {

    @Mock
    private UserService userService;
    @Mock
    private TopicDraftDao topicDraftDao;
    @Mock
    private SecurityContextFacade securityContextFacade;
    @Mock
    private PermissionEvaluator permissionEvaluator;
    @Mock
    private PluginLoader pluginLoader;

    private TopicDraftService topicDraftService;

    private JCUser currentUser;

    @BeforeMethod
    public void setUp() {
        initMocks(this);

        currentUser = new JCUser("current", null, null);
        when(userService.getCurrentUser()).thenReturn(currentUser);

        topicDraftService = new TransactionalTopicDraftService(userService,
                topicDraftDao, securityContextFacade, permissionEvaluator, pluginLoader);
    }

    @Test
    public void getDraftShouldReturnDraftIfUserHasOne() throws Exception {
        TopicDraft expectedDraft = createTopicDraft();
        when(topicDraftDao.getForUser(currentUser)).thenReturn(expectedDraft);

        TopicDraft draft = topicDraftService.getDraft();

        assertReflectionEquals(draft, expectedDraft);
    }


    @Test
    public void getDraftShouldReturnNullIfUserHasNoDraft() throws Exception {
        when(topicDraftDao.getForUser(currentUser)).thenReturn(null);

        assertNull(topicDraftService.getDraft());
    }

    @Test
    public void saveOrUpdateDraftShouldCreateNewDraftIfUserStillHasNoDraft() {
        when(topicDraftDao.getForUser(currentUser)).thenReturn(null);

        TopicDraft topicDraft = topicDraftService.saveOrUpdateDraft(createTopicDraft());

        verify(topicDraftDao).saveOrUpdate(topicDraft);
    }

    @Test
    public void saveOrUpdateDraftShouldUpdateDraftIfUserAlreadyHasOne() {
        TopicDraft topicDraft = createTopicDraft();
        when(topicDraftDao.getForUser(currentUser)).thenReturn(topicDraft);

        topicDraftService.saveOrUpdateDraft(topicDraft);

        verify(topicDraftDao).saveOrUpdate(topicDraft);
    }

    @Test
    public void saveOrUpdateDraftShouldNotRewritePollFieldsWithNullValues() {
        TopicDraft topicDraftWithPoll = createTopicDraft();
        topicDraftWithPoll.setPollTitle(RandomStringUtils.random(5));
        topicDraftWithPoll.setPollItemsValue(RandomStringUtils.random(5));

        when(topicDraftDao.getForUser(currentUser)).thenReturn(topicDraftWithPoll);

        TopicDraft topicDraftWithoutPoll = createTopicDraft();
        topicDraftWithoutPoll.setPollTitle(null);
        topicDraftWithoutPoll.setPollItemsValue(null);

        topicDraftService.saveOrUpdateDraft(topicDraftWithoutPoll);

        assertNotNull(topicDraftWithPoll.getPollTitle());
        assertNotNull(topicDraftWithPoll.getPollItemsValue());
    }

    @Test
    public void deleteDraftShouldDeleteDraftIfUserHasOne() {
        TopicDraft topicDraft = createTopicDraft();
        when(topicDraftDao.getForUser(currentUser)).thenReturn(topicDraft);

        topicDraftService.deleteDraft();

        verify(topicDraftDao).deleteByUser(topicDraft.getTopicStarter());
    }

    private TopicDraft createTopicDraft() {
        Branch branch = createBranch();

        TopicDraft topicDraft = new TopicDraft(currentUser, "title", "content");
        topicDraft.setId(1L);
        topicDraft.setBranchId(branch.getId());
        topicDraft.setTopicType(TopicTypeName.DISCUSSION.getName());

        return topicDraft;
    }

    private Branch createBranch() {
        Branch branch = new Branch("branch name", "branch description");
        branch.setId(1L);
        branch.setUuid("uuid");
        return branch;
    }
}