package sonia.scm.api.v2.resources;

import com.github.sdorra.shiro.ShiroRule;
import com.github.sdorra.shiro.SubjectAware;
import org.jboss.resteasy.core.Dispatcher;
import org.jboss.resteasy.mock.MockDispatcherFactory;
import org.jboss.resteasy.mock.MockHttpRequest;
import org.jboss.resteasy.mock.MockHttpResponse;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;
import org.junit.runner.RunWith;
import org.mockito.ArgumentCaptor;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;
import sonia.scm.repository.HgConfig;
import sonia.scm.repository.HgRepositoryHandler;
import sonia.scm.web.HgVndMediaType;

import javax.inject.Provider;
import javax.servlet.http.HttpServletResponse;
import java.net.URISyntaxException;

import static org.junit.Assert.*;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@SubjectAware(
  configuration = "classpath:sonia/scm/configuration/shiro.ini",
  password = "secret"
)
@RunWith(MockitoJUnitRunner.class)
public class HgConfigAutoConfigurationResourceTest {

  @Rule
  public ShiroRule shiro = new ShiroRule();

  @Rule
  public ExpectedException thrown = ExpectedException.none();

  private Dispatcher dispatcher = MockDispatcherFactory.createDispatcher();

  @InjectMocks
  private HgConfigDtoToHgConfigMapperImpl dtoToConfigMapper;

  @Mock
  private HgRepositoryHandler repositoryHandler;

  @Mock
  private Provider<HgConfigAutoConfigurationResource> resourceProvider;

  @Before
  public void prepareEnvironment() {
    HgConfigAutoConfigurationResource resource =
      new HgConfigAutoConfigurationResource(dtoToConfigMapper, repositoryHandler);

    when(resourceProvider.get()).thenReturn(resource);
    dispatcher.getRegistry().addSingletonResource(
      new HgConfigResource(null, null, null, null,
                           resourceProvider, null));
  }

  @Test
  @SubjectAware(username = "writeOnly")
  public void shouldSetDefaultConfigAndInstallHg() throws Exception {
    MockHttpResponse response = put(null);

    assertEquals(HttpServletResponse.SC_NO_CONTENT, response.getStatus());

    HgConfig actualConfig = captureConfig();
    assertFalse(actualConfig.isDisabled());
  }

  @Test
  @SubjectAware(username = "readOnly")
  public void shouldNotSetDefaultConfigAndInstallHgWhenNotAuthorized() throws Exception {
    thrown.expectMessage("Subject does not have permission [configuration:write:hg]");

    put(null);
  }

  @Test
  @SubjectAware(username = "writeOnly")
  public void shouldUpdateConfigAndInstallHg() throws Exception {
    MockHttpResponse response = put("{\"disabled\":true}");

    assertEquals(HttpServletResponse.SC_NO_CONTENT, response.getStatus());

    HgConfig actualConfig = captureConfig();
    assertTrue(actualConfig.isDisabled());
  }

  @Test
  @SubjectAware(username = "readOnly")
  public void shouldNotUpdateConfigAndInstallHgWhenNotAuthorized() throws Exception {
    thrown.expectMessage("Subject does not have permission [configuration:write:hg]");

    put("{\"disabled\":true}");
  }

  private MockHttpResponse put(String content) throws URISyntaxException {
    MockHttpRequest request = MockHttpRequest.put("/" + HgConfigResource.HG_CONFIG_PATH_V2 + "/auto-configuration");

    if (content != null) {
      request
        .contentType(HgVndMediaType.CONFIG)
        .content(content.getBytes());
    }

    MockHttpResponse response = new MockHttpResponse();
    dispatcher.invoke(request, response);
    return response;
  }

  private HgConfig captureConfig() {
    ArgumentCaptor<HgConfig> configCaptor = ArgumentCaptor.forClass(HgConfig.class);
    verify(repositoryHandler).doAutoConfiguration(configCaptor.capture());
    return configCaptor.getValue();
  }

}
