/*
 * Copyright 2000-2014 JetBrains s.r.o.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package jetbrains.buildServer.buildTriggers.vcs.mercurial;

import jetbrains.buildServer.buildTriggers.vcs.mercurial.command.HgVcsRoot;
import jetbrains.buildServer.serverSide.BasePropertiesModel;
import jetbrains.buildServer.serverSide.TeamCityProperties;
import jetbrains.buildServer.util.TestFor;
import jetbrains.buildServer.vcs.VcsRoot;
import jetbrains.buildServer.vcs.impl.VcsRootImpl;
import junit.framework.TestCase;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import java.io.File;

import static jetbrains.buildServer.buildTriggers.vcs.mercurial.VcsRootBuilder.vcsRoot;

/**
 * @author Pavel.Sher
 */
@Test
public class HgVcsRootTest extends TestCase {

  @BeforeMethod
  public void setUp() throws Exception {
    new TeamCityProperties() {{ setModel(new BasePropertiesModel() {});}};
  }

  public void test_url_without_credentials() {
    VcsRootImpl vcsRoot = createVcsRoot("http://host.com/path");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("http://user:pwd@host.com/path", root.getRepositoryUrlWithCredentials());
  }

  public void test_url_with_credentials() {
    VcsRootImpl vcsRoot = createVcsRoot("http://user:pwd@host.com/path");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("http://user:pwd@host.com/path", root.getRepositoryUrlWithCredentials());
  }

  public void test_url_with_username() {
    VcsRootImpl vcsRoot = createVcsRoot("http://user@host.com/path");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("http://user:pwd@host.com/path", root.getRepositoryUrlWithCredentials());
  }

  public void test_url_with_at_after_slash() {
    VcsRootImpl vcsRoot = createVcsRoot("http://host.com/path@");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("http://user:pwd@host.com/path@", root.getRepositoryUrlWithCredentials());
  }

  public void test_url_with_at_in_username() {
    VcsRootImpl vcsRoot = createVcsRoot("http://host.com/path", "my.name@gmail.com", "1234");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("http://my.name%40gmail.com:1234@host.com/path", root.getRepositoryUrlWithCredentials());
  }

  /** TW-13768 */
  public void test_underscore_in_host() {
		VcsRootImpl vcsRoot = createVcsRoot("http://Klekovkin.SDK_GARANT:8000/", "my.name@gmail.com", "1234");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
		assertEquals("http://my.name%40gmail.com:1234@Klekovkin.SDK_GARANT:8000/", root.getRepositoryUrlWithCredentials());
	}

  /** TW-13768 */
  public void test_underscore_in_host_with_credentials_in_url() {
    VcsRootImpl vcsRoot = createVcsRoot("http://me:mypass@Klekovkin.SDK_GARANT:8000/");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
		assertEquals("http://me:mypass@Klekovkin.SDK_GARANT:8000/", root.getRepositoryUrlWithCredentials());
  }

  public void test_windows_path() throws Exception {
    VcsRootImpl vcsRoot = createVcsRoot("c:\\windows\\path");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("c:\\windows\\path", root.getRepositoryUrlWithCredentials());
  }

  public void test_file_scheme_has_no_credentials() {
    VcsRootImpl vcsRoot = createVcsRoot("file:///path/to/repo", "my.name@gmail.com", "1234");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("file:///path/to/repo", root.getRepositoryUrlWithCredentials());
  }

  public void uncompressed_transfer() {
    VcsRootImpl vcsRoot = createVcsRoot("http://host.com/path");
    vcsRoot.addProperty(Constants.UNCOMPRESSED_TRANSFER, "true");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertTrue(root.isUncompressedTransfer());
  }

  //TW-18262
  public void ampersand_in_password() {
    VcsRootImpl vcsRoot = createVcsRoot("http://some.org/path", "user", "m&n");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("http://user:m%26n@some.org/path", root.getRepositoryUrlWithCredentials());
  }

  //TW-18835
  public void test_ssh() {
    VcsRootImpl vcsRoot = createVcsRoot("ssh://ourserver.com/mercurialrepo/", "user", "pwd");
    HgVcsRoot root = new HgVcsRoot(vcsRoot);
    assertEquals("ssh://user:pwd@ourserver.com/mercurialrepo/", root.getRepositoryUrlWithCredentials());
  }

  @TestFor(issues = "TW-36401")
  public void test_disabling_custom_clone_dirs() throws Exception {
    File cloneDir = new File("");
    VcsRoot root = vcsRoot().withCloneRepositoryTo(cloneDir).withUrl("http://some.org/repo").build();
    HgVcsRoot hgRoot1 = new HgVcsRoot(root);
    assertEquals(cloneDir.getAbsolutePath(), hgRoot1.getCustomClonePath());
    try {
      System.setProperty(Constants.CUSTOM_CLONE_PATH_ENABLED, "false");
      HgVcsRoot hgRoot2 = new HgVcsRoot(root);
      assertNull(hgRoot2.getCustomClonePath());
    } finally {
      System.getProperties().remove(Constants.CUSTOM_CLONE_PATH_ENABLED);
    }
  }

  private VcsRootImpl createVcsRoot(String url) {
    return createVcsRoot(url, "user", "pwd");
  }

  private VcsRootImpl createVcsRoot(String url, String userName, String password) {
    VcsRootImpl vcsRoot = new VcsRootImpl(1, Constants.VCS_NAME);
    vcsRoot.addProperty(Constants.HG_COMMAND_PATH_PROP, "hg.exe");
    vcsRoot.addProperty(Constants.REPOSITORY_PROP, url);
    vcsRoot.addProperty(Constants.USERNAME, userName);
    vcsRoot.addProperty(Constants.PASSWORD, password);
    return vcsRoot;
  }
}