#!/usr/bin/env python
# Copyright 2019 The Fuchsia Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import os
import unittest
import tempfile

import test_env
from lib.host import Host

from host_mock import MockHost


class TestHost(unittest.TestCase):

  def test_join(self):
    fuchsia_dir = os.getenv('FUCHSIA_DIR')
    os.unsetenv('FUCHSIA_DIR')
    del os.environ['FUCHSIA_DIR']
    with self.assertRaises(Host.ConfigError):
      Host.join('foo')
    if not fuchsia_dir:
      return
    os.environ['FUCHSIA_DIR'] = fuchsia_dir
    self.assertTrue(Host.join('foo').endswith('foo'))

  def test_find_build_dir(self):
    host = Host()
    fuchsia_dir = os.getenv('FUCHSIA_DIR')
    os.unsetenv('FUCHSIA_DIR')
    del os.environ['FUCHSIA_DIR']
    with self.assertRaises(Host.ConfigError):
      host.find_build_dir()
    if not fuchsia_dir:
      return
    os.environ['FUCHSIA_DIR'] = fuchsia_dir
    print(host.find_build_dir())
    self.assertTrue(os.path.isdir(host.find_build_dir()))

  def test_set_build_ids(self):
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.set_build_ids('no_such_ids')
    if not os.getenv('FUCHSIA_DIR'):
      return
    build_dir = host.find_build_dir()
    build_ids = Host.join(build_dir, 'ids.txt')
    if not os.path.exists(build_ids):
      return
    host.set_build_ids(build_ids)
    self.assertEqual(host._ids, build_ids)

  def test_set_zxtools(self):
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.set_zxtools('no_such_zxtools')
    if not os.getenv('FUCHSIA_DIR'):
      return
    build_dir = host.find_build_dir()
    zxtools = Host.join(build_dir + '.zircon', 'tools')
    if not os.path.isdir(zxtools):
      return
    host.set_zxtools(zxtools)
    self.assertEqual(host._zxtools, zxtools)

  def test_set_platform(self):
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.set_platform('no_such_platform')
    if not os.getenv('FUCHSIA_DIR'):
      return
    platform = 'mac-x64' if os.uname()[0] == 'Darwin' else 'linux-x64'
    host.set_platform(platform)
    self.assertEqual(host._platform, platform)

  def test_set_symbolizer(self):
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.set_symbolizer('no_such_executable', 'no_such_symbolizer')
    if not os.getenv('FUCHSIA_DIR'):
      return
    platform = 'mac-x64' if os.uname()[0] == 'Darwin' else 'linux-x64'
    executable = Host.join('zircon', 'prebuilt', 'downloads', 'symbolize')
    symbolizer = Host.join('buildtools', platform, 'clang', 'bin',
                           'llvm-symbolizer')
    with self.assertRaises(Host.ConfigError):
      host.set_symbolizer(executable, 'no_such_symbolizer')
    with self.assertRaises(Host.ConfigError):
      host.set_symbolizer('no_such_executable', symbolizer)
    host.set_symbolizer(executable, symbolizer)
    self.assertEqual(host._symbolizer_exec, executable)
    self.assertEqual(host._llvm_symbolizer, symbolizer)

  def test_set_fuzzers_json(self):
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.set_fuzzers_json('no_such_json')
    if not os.getenv('FUCHSIA_DIR'):
      return
    build_dir = host.find_build_dir()
    json_file = Host.join(build_dir, 'fuzzers.json')
    if not os.path.exists(json_file):
      return
    # No guarantee of contents; just ensure it parses without crashing
    host.set_fuzzers_json(json_file)

  def test_set_build_dir(self):
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.set_fuzzers_json('no_such_build_dir')
    if not os.getenv('FUCHSIA_DIR'):
      return
    build_dir = host.find_build_dir()
    ssh_config = Host.join(build_dir, 'ssh-keys', 'ssh_config')
    build_ids = Host.join(build_dir, 'ids.txt')
    zxtools = Host.join(build_dir + '.zircon', 'tools')
    platform = 'mac-x64' if os.uname()[0] == 'Darwin' else 'linux-x64'
    executable = Host.join('zircon', 'prebuilt', 'downloads', 'symbolize')
    symbolizer = Host.join('buildtools', platform, 'clang', 'bin',
                           'llvm-symbolizer')
    if not os.path.exists(ssh_config) or not os.path.exists(
        build_ids) or not os.path.isdir(zxtools):
      return
    host.set_build_dir(build_dir)
    self.assertEqual(host._ids, build_ids)
    self.assertEqual(host._zxtools, zxtools)
    self.assertEqual(host._platform, platform)
    self.assertEqual(host._symbolizer_exec, executable)
    self.assertEqual(host._llvm_symbolizer, symbolizer)

  def test_zircon_tool(self):
    mock = MockHost()
    mock.zircon_tool(['mock_tool', 'mock_arg'])
    self.assertIn('mock/out/default.zircon/tools/mock_tool mock_arg',
                  mock.history)
    # If a build tree is available, try using a zircon tool for "real".
    host = Host()
    try:
      host.set_build_dir(host.find_build_dir())
    except Host.ConfigError:
      return
    with self.assertRaises(Host.ConfigError):
      host.zircon_tool(['no_such_tool'])
    path = os.path.abspath(__file__)
    line = host.zircon_tool(['merkleroot', path])
    self.assertRegexpMatches(line, r'[0-9a-f]* - ' + path)

  def test_killall(self):
    mock = MockHost()
    mock.killall('mock_tool')
    self.assertIn('killall mock_tool', mock.history)

  def test_symbolize(self):
    mock = MockHost()
    mock.symbolize('mock_in', 'mock_out')
    self.assertIn(
        'mock/symbolize-ids-rel -ids mock/ids.txt -llvm-symbolizer mock/llvm_symbolizer',
        mock.history)
    # If a build tree is available, try doing symbolize for "real".
    host = Host()
    try:
      host.set_build_dir(host.find_build_dir())
    except Host.ConfigError:
      return
    tmp_in = tempfile.TemporaryFile()
    tmp_out = tempfile.TemporaryFile()
    host.symbolize(tmp_in, tmp_out)

  def test_notify_user(self):
    host = Host()
    try:
      host.set_build_dir(host.find_build_dir())
    except Host.ConfigError:
      return
    host.notify_user('This is a test', 'This is only a test.')

  def test_snapshot(self):
    fuchsia_dir = os.getenv('FUCHSIA_DIR')
    os.unsetenv('FUCHSIA_DIR')
    del os.environ['FUCHSIA_DIR']
    host = Host()
    with self.assertRaises(Host.ConfigError):
      host.snapshot()
    if not fuchsia_dir:
      return
    os.environ['FUCHSIA_DIR'] = fuchsia_dir
    line = host.snapshot()
    self.assertRegexpMatches(line, r'[0-9a-f]{40}')


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