package com.armedia.acm.services.functionalaccess.web.api;

/*-
 * #%L
 * ACM Service: Functional Access Control
 * %%
 * Copyright (C) 2014 - 2018 ArkCase LLC
 * %%
 * This file is part of the ArkCase software. 
 * 
 * If the software was purchased under a paid ArkCase license, the terms of 
 * the paid license agreement will prevail.  Otherwise, the software is 
 * provided under the following open source license terms:
 * 
 * ArkCase 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 3 of the License, or
 * (at your option) any later version.
 *  
 * ArkCase 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 ArkCase. If not, see <http://www.gnu.org/licenses/>.
 * #L%
 */

import static org.easymock.EasyMock.expect;
import static org.junit.Assert.assertEquals;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;

import com.armedia.acm.services.functionalaccess.service.FunctionalAccessService;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;

import org.apache.commons.lang.StringUtils;
import org.easymock.EasyMockSupport;
import org.junit.Before;
import org.junit.Test;
import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.security.core.Authentication;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.MvcResult;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.servlet.mvc.method.annotation.ExceptionHandlerExceptionResolver;

import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author riste.tutureski
 */
public class GetApplicationRolesToGroupsAPIControllerTest extends EasyMockSupport
{

    private Logger LOG = LogManager.getLogger(getClass());

    private MockMvc mockMvc;
    private GetApplicationRolesToGroupsAPIController unit;
    private Authentication mockAuthentication;

    private FunctionalAccessService mockFunctionalAccessService;

    @Autowired
    private ExceptionHandlerExceptionResolver exceptionResolver;

    @Before
    public void setUp() throws Exception
    {
        setUnit(new GetApplicationRolesToGroupsAPIController());

        mockFunctionalAccessService = createMock(FunctionalAccessService.class);
        setMockMvc(MockMvcBuilders.standaloneSetup(getUnit()).setHandlerExceptionResolvers(getExceptionResolver()).build());
        setMockAuthentication(createMock(Authentication.class));

        getUnit().setFunctionalAccessService(mockFunctionalAccessService);
    }

    @Test
    public void getApplicationRolesToGroupsTest() throws Exception
    {

        expect(getMockAuthentication().getName()).andReturn("user");

        Map<String, List<String>> rolesToGroups = new HashMap<>();

        rolesToGroups.put("ROLE_INVESTIGATOR", Arrays.asList("acm_investigator_dev", "acm_administrator_dev"));

        expect(mockFunctionalAccessService.getApplicationRolesToGroups()).andReturn(rolesToGroups);

        replayAll();

        MvcResult result = getMockMvc().perform(
                get("/api/v1/functionalaccess/rolestogroups")
                        .accept(MediaType.parseMediaType("application/json;charset=UTF-8"))
                        .contentType(MediaType.APPLICATION_JSON)
                        .principal(getMockAuthentication()))
                .andReturn();

        LOG.info("Results: " + result.getResponse().getContentAsString());

        Map<String, List<String>> resultTestData = new HashMap<>();
        ObjectMapper mapper = new ObjectMapper();

        try
        {
            resultTestData = mapper.readValue(result.getResponse().getContentAsString(), new TypeReference<HashMap<String, List<String>>>()
            {
            });
        }
        catch (Exception e)
        {
            LOG.error("Cannot create map from source: " + result.getResponse().getContentAsString());
        }

        assertEquals(rolesToGroups.size(), resultTestData.size());
        assertEquals(rolesToGroups.get("ROLE_INVESTIGATOR"), resultTestData.get("ROLE_INVESTIGATOR"));
        assertEquals("acm_investigator_dev,acm_administrator_dev", StringUtils.join(resultTestData.get("ROLE_INVESTIGATOR"), ","));
        assertEquals(HttpStatus.OK.value(), result.getResponse().getStatus());

    }

    public MockMvc getMockMvc()
    {
        return mockMvc;
    }

    public void setMockMvc(MockMvc mockMvc)
    {
        this.mockMvc = mockMvc;
    }

    public GetApplicationRolesToGroupsAPIController getUnit()
    {
        return unit;
    }

    public void setUnit(GetApplicationRolesToGroupsAPIController unit)
    {
        this.unit = unit;
    }

    public Authentication getMockAuthentication()
    {
        return mockAuthentication;
    }

    public void setMockAuthentication(Authentication mockAuthentication)
    {
        this.mockAuthentication = mockAuthentication;
    }

    public ExceptionHandlerExceptionResolver getExceptionResolver()
    {
        return exceptionResolver;
    }

    public void setExceptionResolver(
            ExceptionHandlerExceptionResolver exceptionResolver)
    {
        this.exceptionResolver = exceptionResolver;
    }
}
