<?php

/*
 * Copyright 2011 Facebook, Inc.
 *
 * 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.
 */

class PhutilUnitTestEngine extends ArcanistBaseUnitTestEngine {

  public function run() {

    $bootloader = PhutilBootloader::getInstance();

    $tests = array();
    foreach ($this->getPaths() as $path) {
      $library_root = phutil_get_library_root_for_path($path);
      if (!$library_root) {
        continue;
      }
      $library_name = phutil_get_library_name_for_root($library_root);

      $path = Filesystem::resolvePath($path);
      if ($path == $library_root) {
        continue;
      }

      if (!is_dir($path)) {
        $path = dirname($path);
      }

      $library_path = Filesystem::readablePath($path, $library_root);
      if (basename($library_path) == '__tests__') {
        // Okay, this is a __tests__ module.
      } else {
        $exists = $bootloader->moduleExists(
          $library_name,
          $library_path.'/__tests__');
        if ($exists) {
          // This is a module which has a __tests__ module in it.
          $path .= '/__tests__';
        } else {
          // Look for a parent named __tests__.
          $rpos = strrpos($library_path, '/__tests__');
          if ($rpos === false) {
            // No tests to run since there is no child or parent module named
            // __tests__.
            continue;
          }
          // Select the parent named __tests__.
          $path = substr($path, 0, $rpos + strlen('/__tests__'));
        }
      }


      $module_name = Filesystem::readablePath($path, $library_root);
      $module_key = $library_name.':'.$module_name;
      $tests[$module_key] = array(
        'library' => $library_name,
        'root'    => $library_root,
        'module'  => $module_name,
      );
    }

    if (!$tests) {
      throw new ArcanistNoEffectException("No tests to run.");
    }

    $run_tests = array();
    foreach ($tests as $test) {
      $symbols = id(new PhutilSymbolLoader())
        ->setType('class')
        ->setLibrary($test['library'])
        ->setModule($test['module'])
        ->setAncestorClass('ArcanistPhutilTestCase')
        ->selectAndLoadSymbols();
      foreach ($symbols as $symbol) {
        $run_tests[$symbol['name']] = true;
      }
    }
    $run_tests = array_keys($run_tests);

    if (!$run_tests) {
      throw new ArcanistNoEffectException(
        "No tests to run. You may need to rebuild the phutil library map.");
    }

    $results = array();
    foreach ($run_tests as $test_class) {
      PhutilSymbolLoader::loadClass($test_class);
      $test_case = newv($test_class, array());
      $results[] = $test_case->run();
    }
    if ($results) {
      $results = call_user_func_array('array_merge', $results);
    }


    return $results;
  }

}