package at.asitplus.eidas.specific.modules.auth.eidas.v2.test.tasks;

import static org.mockito.ArgumentMatchers.any;
import static org.powermock.api.mockito.PowerMockito.when;

import java.io.IOException;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.HashMap;
import java.util.Map;

import javax.xml.namespace.QName;

import org.apache.commons.lang3.RandomStringUtils;
import org.jetbrains.annotations.NotNull;
import org.junit.Assert;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpServletResponse;
import org.springframework.test.annotation.DirtiesContext;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import com.skjolberg.mockito.soap.SoapServiceRule;

import at.asitplus.eidas.specific.connector.MsEidasNodeConstants;
import at.asitplus.eidas.specific.modules.auth.eidas.v2.Constants;
import at.asitplus.eidas.specific.modules.auth.eidas.v2.service.EidasAttributeRegistry;
import at.asitplus.eidas.specific.modules.auth.eidas.v2.tasks.CreateIdentityLinkTask;
import at.asitplus.eidas.specific.modules.auth.eidas.v2.test.dummy.DummySpecificCommunicationService;
import at.gv.egiz.eaaf.core.api.data.EaafConfigConstants;
import at.gv.egiz.eaaf.core.api.data.EaafConstants;
import at.gv.egiz.eaaf.core.api.idp.IConfiguration;
import at.gv.egiz.eaaf.core.api.idp.process.ExecutionContext;
import at.gv.egiz.eaaf.core.exceptions.EaafStorageException;
import at.gv.egiz.eaaf.core.impl.idp.auth.data.AuthProcessDataWrapper;
import at.gv.egiz.eaaf.core.impl.idp.module.test.DummySpConfiguration;
import at.gv.egiz.eaaf.core.impl.idp.module.test.TestRequestImpl;
import at.gv.egiz.eaaf.core.impl.idp.process.ExecutionContextImpl;
import eu.eidas.auth.commons.attribute.AttributeDefinition;
import eu.eidas.auth.commons.attribute.ImmutableAttributeMap;
import eu.eidas.auth.commons.attribute.PersonType;
import eu.eidas.auth.commons.protocol.impl.AuthenticationResponse;
import lombok.val;
import szrservices.PersonInfoType;
import szrservices.SZR;
import szrservices.SignContentEntry;
import szrservices.SignContentResponseType;

@RunWith(SpringJUnit4ClassRunner.class)
//@RunWith(PowerMockRunner.class)
//@PowerMockRunnerDelegate(SpringJUnit4ClassRunner.class)
@PrepareForTest(CreateIdentityLinkTask.class)
@DirtiesContext(classMode = DirtiesContext.ClassMode.BEFORE_CLASS)
@ContextConfiguration("/SpringTest-context_tasks_test.xml")
public class CreateIdentityLinkTaskEidNewTest {


  @Autowired(required = true)
  private CreateIdentityLinkTask task;
//  @Autowired(required = true)
//  private FinalizeAuthenticationTask authTask;
  @Autowired(required = true)
  private DummySpecificCommunicationService commService;
  @Autowired(required = true)
  private IConfiguration basicConfig;
  @Autowired
  protected EidasAttributeRegistry attrRegistry;

  final ExecutionContext executionContext = new ExecutionContextImpl();
  private MockHttpServletRequest httpReq;
  private MockHttpServletResponse httpResp;
  private TestRequestImpl pendingReq;
  private DummySpConfiguration oaParam;
  private SZR szrMock;
  
  @Rule
  public final SoapServiceRule soap = SoapServiceRule.newInstance();

  /**
   * jUnit class initializer.
   *
   * @throws IOException In case of an error
   */
  @BeforeClass
  public static void classInitializer() throws IOException {
    final String current = new java.io.File(".").toURI().toString();
    System.setProperty("eidas.ms.configuration", current
        + "src/test/resources/config/junit_config_3.properties");

  }

  /**
   * jUnit test set-up.
   */
  @Before
  public void setUp() throws EaafStorageException, URISyntaxException {

    httpReq = new MockHttpServletRequest("POST", "https://localhost/authhandler");
    httpResp = new MockHttpServletResponse();
    RequestContextHolder.resetRequestAttributes();
    RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(httpReq, httpResp));

    final Map<String, String> spConfig = new HashMap<>();
    spConfig.put(EaafConfigConstants.SERVICE_UNIQUEIDENTIFIER, "testSp");
    spConfig.put("target", "urn:publicid:gv.at:cdid+XX");
    spConfig.put("eidMode", "new");
    oaParam = new DummySpConfiguration(spConfig, basicConfig);
    pendingReq = new TestRequestImpl();

    AuthenticationResponse response = buildDummyAuthResponse();


    pendingReq.getSessionData(AuthProcessDataWrapper.class).setGenericDataToSession(Constants.DATA_FULL_EIDAS_RESPONSE, response);
    pendingReq.setSpConfig(oaParam);
    pendingReq.setPendingReqId(at.gv.egiz.eaaf.core.impl.utils.Random.nextProcessReferenceValue());
    pendingReq.setAuthUrl("http://test.com/");
    pendingReq.setTransactionId("avaasbav");

    executionContext.put(MsEidasNodeConstants.REQ_PARAM_SELECTED_COUNTRY, "XX");
    executionContext.put(EaafConstants.PROCESS_ENGINE_REQUIRES_NO_POSTAUTH_REDIRECT, true);

    szrMock = soap.mock(SZR.class, "http://localhost:1234/demoszr");
  }

  @NotNull
  private AuthenticationResponse buildDummyAuthResponse() throws URISyntaxException {
    AttributeDefinition attributeDef = AttributeDefinition.builder().friendlyName(Constants.eIDAS_ATTR_PERSONALIDENTIFIER)
        .nameUri(new URI("ad", "sd", "ff")).personType(PersonType.LEGAL_PERSON)
        .xmlType(new QName("http://saf", "as", "af"))
        .attributeValueMarshaller("eu.eidas.auth.commons.attribute.impl.LiteralStringAttributeValueMarshaller").build();
    AttributeDefinition attributeDef2 = AttributeDefinition.builder().friendlyName(Constants.eIDAS_ATTR_CURRENTFAMILYNAME)
        .nameUri(new URI("ad", "sd", "fff")).personType(PersonType.LEGAL_PERSON)
        .xmlType(new QName("http://saf", "as", "aff"))
        .attributeValueMarshaller("eu.eidas.auth.commons.attribute.impl.LiteralStringAttributeValueMarshaller").build();
    AttributeDefinition attributeDef3 = AttributeDefinition.builder().friendlyName(Constants.eIDAS_ATTR_CURRENTGIVENNAME)
        .nameUri(new URI("ad", "sd", "ffff")).personType(PersonType.LEGAL_PERSON)
        .xmlType(new QName("http://saf", "as", "afff"))
        .attributeValueMarshaller("eu.eidas.auth.commons.attribute.impl.LiteralStringAttributeValueMarshaller").build();
    AttributeDefinition attributeDef4 = AttributeDefinition.builder().friendlyName(Constants.eIDAS_ATTR_DATEOFBIRTH)
        .nameUri(new URI("ad", "sd", "fffff")).personType(PersonType.LEGAL_PERSON)
        .xmlType(new QName("http://saf", "as", "affff"))
        .attributeValueMarshaller("eu.eidas.auth.commons.attribute.impl.DateTimeAttributeValueMarshaller").build();

    ImmutableAttributeMap attributeMap = ImmutableAttributeMap.builder().put(attributeDef, "de/st/" + RandomStringUtils.randomNumeric(64))
        .put(attributeDef2, RandomStringUtils.randomAlphabetic(10))
        .put(attributeDef3, RandomStringUtils.randomAlphabetic(10))
        .put(attributeDef4, "2001-01-01")
        .build();


    val b = new AuthenticationResponse.Builder();
    return b.id("aasdf").issuer("asd").subject("asf").statusCode("200").inResponseTo("asdf")
        .subjectNameIdFormat("afaf").attributes(attributeMap).build();
  }

  @Test
  public void fullTest() {
    // keystore password f/+saJBc3a}*/T^s
    try {

      String test = szrMock.getStammzahlEncrypted(new PersonInfoType(), false);

//      SZR szrMock2 = Mockito.mock(SZR.class, Mockito.CALLS_REAL_METHODS);
      when(szrMock, "getStammzahlEncrypted", any(), any()).thenReturn(RandomStringUtils.randomNumeric(10));
      val signContentResp = new SignContentResponseType();
      SignContentEntry signContentEntry = new SignContentEntry();
      signContentEntry.setValue(RandomStringUtils.randomAlphanumeric(10));
      signContentResp.getOut().add(signContentEntry);
      when(szrMock, "signContent", any(), any(), any()).thenReturn(signContentResp);
//      when(szrMock2, "getStammzahlEncrypted", any(), any()).thenReturn(RandomStringUtils.randomNumeric(10));
      // Wie kann ich das vom Bean rein gehängte SZR mit dem Mock überschreiben?

      task.execute(pendingReq, executionContext);

//      authTask.execute(pendingReq, executionContext);
    } catch (Exception e) {
      e.printStackTrace();
      Assert.fail();
    }

//    val spyTask = spy(task);
//    Map<String, Object> requiredAttributes = new HashMap<>();
//    requiredAttributes.put(Constants.eIDAS_ATTR_PERSONALIDENTIFIER, "asdf");
//    requiredAttributes.put(Constants.eIDAS_ATTR_CURRENTFAMILYNAME, "asdf");
//    requiredAttributes.put(Constants.eIDAS_ATTR_CURRENTGIVENNAME, "asdf");
//    requiredAttributes.put(Constants.eIDAS_ATTR_DATEOFBIRTH, new DateTime());
//    requiredAttributes.put(Constants.eIDAS_ATTR_PLACEOFBIRTH, "adf");
//    requiredAttributes.put(Constants.eIDAS_ATTR_BIRTHNAME, "adf");
//    requiredAttributes.put(Constants.eIDAS_ATTR_CURRENTADDRESS, new PostalAddressType());
//    try {
//    doReturn(requiredAttributes).when(spyTask, "convertEidasAttrToSimpleMap", (any(ImmutableMap.class)));
//      spyTask.execute(pendingReq, executionContext);
//    } catch (final Exception e) {
//      e.printStackTrace();
//      Assert.fail();
//    }
  }

}