#!/usr/bin/python

# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Unittests for chrome_set_ver."""

import mox
import os
import shutil
import sys
import tempfile
import unittest

sys.path.append('../..')
import chromite.lib.cros_build_lib as cros_lib
import chromite.bin.chrome_set_ver as chrome_set_ver


# pylint: disable=W0212,R0904
class TestEndedException(Exception):
  """Exception thrown by test to mark completion."""
  pass


class DEPSFileTest(mox.MoxTestBase):

  def setUp(self):
    mox.MoxTestBase.setUp(self)

    self.tempdir = tempfile.mkdtemp()
    self.test_base = os.path.join(os.path.realpath(os.path.dirname(__file__)),
                             'testdata/chrome_set_ver_unittest')
    self.repo_root = os.path.join(self.tempdir, 'repo_root')
    self.repo_root_zip = os.path.join(self.test_base,
                                          'repo_root.tar.gz')

    assert(not os.path.exists(self.repo_root))
    print('Unzipping test repo_root')
    cros_lib.RunCommand(['tar', '-xvf', self.repo_root_zip],
                        cwd=self.tempdir, redirect_stdout=True)
    assert(os.path.exists(self.repo_root))

    self.old_dir = os.getcwd()
    os.chdir(self.repo_root)
    repo_root = cros_lib.FindRepoDir()
    assert(os.path.realpath(os.path.dirname(repo_root)) == self.repo_root)


  def _VerifyReposAndSymlink(self):
    # Verify symlink was generated
    test_file = 'chromium/src/third_party/cros/HELLO'
    self.assertTrue(os.path.exists(os.path.join(self.repo_root, test_file)))
    test_file = 'chromium/src/third_party/cros_system_api/WORLD'
    self.assertTrue(os.path.exists(os.path.join(self.repo_root, test_file)))

    # Verify repo's were reset properly
    repos = {'repo1': '8d35063e1836c79c9ef97bf81eb43f450dc111ac',
             'repo2': 'b2f03c74b48866eb3da5c4cab554c792a70aeda8',
             'repo3': '8d35063e1836c79c9ef97bf81eb43f450dc111ac',
             'repo4': 'b2f03c74b48866eb3da5c4cab554c792a70aeda8',
             'third_party/cros': '8d35063e1836c79c9ef97bf81eb43f450dc111ac',
             'third_party/cros_system_api':
             'b2f03c74b48866eb3da5c4cab554c792a70aeda8'}

    for repo, revision in repos.iteritems():
      repo_path = os.path.join(self.repo_root, 'chromium', 'src', repo)
      self.assertTrue(cros_lib.GetGitRepoRevision(repo_path) == revision)


  def testParseAndRunDEPSFileNoHooks(self):
    """Test resetting of repos, usage of Var's."""
    chrome_set_ver.main(['-d',
                         os.path.join(self.test_base, 'test_1/DEPS.git')])

    self._VerifyReposAndSymlink()

  def testParseAndRunDEPSFileWithHooks(self):
    """Test correct operation with --runhooks option."""
    chrome_set_ver.main(['--runhooks', '-d',
                         os.path.join(self.test_base, 'test_1/DEPS.git')])

    self._VerifyReposAndSymlink()

    # Verify hooks were run
    hook1_output = os.path.join(self.repo_root, 'chromium', 'out1')
    hook2_output = os.path.join(self.repo_root, 'chromium', 'out2')
    self.assertTrue(os.path.exists(hook1_output))
    self.assertTrue(os.path.exists(hook2_output))

  def testParseInternalDEPSFile(self):
    """Test that the internal DEPS file is found and parsed properly."""
    shutil.copyfile(os.path.join(self.test_base, 'test_internal/DEPS.git'),
                    os.path.join(self.repo_root,
                                 'chromium/src-internal/.DEPS.git'))

    chrome_set_ver.main(['-d',
                         os.path.join(self.test_base, 'test_1/DEPS.git')])

    repos = {'repo_internal': '8d35063e1836c79c9ef97bf81eb43f450dc111ac'}

    for repo, revision in repos.iteritems():
      repo_path = os.path.join(self.repo_root, 'chromium', 'src', repo)
      self.assertTrue(cros_lib.GetGitRepoRevision(repo_path) == revision)

  def testErrorOnOverlap(self):
    """Test that an overlapping entry in unix deps causes error."""
    self.assertRaises(AssertionError, chrome_set_ver.main,
                      ['-d', os.path.join(self.test_base, 'test_2/DEPS.git')])

  def testErrorOnFromSyntax(self):
    """Test that using the 'From' keyword causes error."""
    self.assertRaises(NotImplementedError, chrome_set_ver.main,
                      ['-d', os.path.join(self.test_base, 'test_3/DEPS.git')])

  def testErrorOnNonexistentHash(self):
    """Test dying cleanly when trying to pin to nonexistent hash."""
    self.mox.StubOutWithMock(cros_lib, 'Die')
    cros_lib.Die(mox.IgnoreArg()).AndRaise(TestEndedException)
    self.mox.ReplayAll()
    self.assertRaises(TestEndedException, chrome_set_ver.main,
                      ['-d', os.path.join(self.test_base, 'test_4/DEPS.git')])

  def testProjectInManifestNotWorkingDir(self):
    """Test erroring out and telling user to repo sync."""
    self.mox.StubOutWithMock(cros_lib, 'Die')
    cros_lib.Die(mox.StrContains('not in working tree')).AndRaise(
        TestEndedException)
    self.mox.ReplayAll()
    self.assertRaises(TestEndedException, chrome_set_ver.main,
                      ['-d', os.path.join(self.test_base, 'test_5/DEPS.git')])

  def testAutoSyncProject(self):
    """Test pulling down of projects not in manifest."""
    chrome_set_ver.main(['-d',
                         os.path.join(self.test_base, 'test_6/DEPS.git')])
    repo_path = os.path.join(self.repo_root, 'chromium/src/does_not_exist')
    self.assertTrue(cros_lib.GetGitRepoRevision(repo_path) ==
                    '8d35063e1836c79c9ef97bf81eb43f450dc111ac')

  def testProjectThatWeManageNowInManifest(self):
    """Test erroring out and telling user to remove project."""
    self.mox.StubOutWithMock(cros_lib, 'Die')
    cros_lib.Die(mox.StrContains('needs to be replaced')).AndRaise(
        TestEndedException)
    self.mox.ReplayAll()
    self.assertRaises(TestEndedException, chrome_set_ver.main,
                      ['-d', os.path.join(self.test_base, 'test_7/DEPS.git')])

  def testParseUrl(self):
    """Test extracting of project name from url."""
    url = 'http://git.chromium.org/external/libjingle.git'
    self.assertEquals('external/libjingle',
                      chrome_set_ver._ExtractProjectFromUrl(url))

    url = 'ssh://gerrit-int.chromium.org:29419/chrome/data/page_cycler.git'
    self.assertEquals('chrome/data/page_cycler',
                      chrome_set_ver._ExtractProjectFromUrl(url))

    url = 'git+ssh://gerrit-int.chromium.org:29419/chrome/data/page_cycler.git'
    self.assertEquals('chrome/data/page_cycler',
                      chrome_set_ver._ExtractProjectFromUrl(url))

    url = '../chrome/data/page_cycler.git'
    self.assertEquals('../chrome/data/page_cycler',
                      chrome_set_ver._ExtractProjectFromUrl(url))


  def tearDown(self):
    os.chdir(self.old_dir)
    cros_lib.RunCommand(['rm', '-rf', self.tempdir])
    assert(not os.path.exists(self.repo_root))
    mox.MoxTestBase.tearDown(self)

if __name__ == '__main__':
  unittest.main()