source: trunk/tests/utilities.php @ 1184

<?php

require_once(dirname($argv[0]) . "/../config.php");
require_once(IA_ROOT_DIR."common/log.php");
require_once(IA_ROOT_DIR."common/common.php");
require_once(IA_ROOT_DIR."common/db/db.php");
require_once(IA_ROOT_DIR."common/db/user.php");

if (!IA_DEVELOPMENT_MODE) {
    // These tests alter the database and can remove user created content
    // by mistake (it has happened in the past).
    log_error("You should never run these tests in a production environment");
}

// Test with curl. $args format:
// * url: url to curl. If http:// is ommited IA_URL_HOST is assumed.
// * post: post arguments (use url_ functions for get args).
// * user: user to curl as. HTTP auth is used, password is always pwd.
// * validate_html: boolean, default true. Automatic html validation.
//
// TODO: random passwords in test_prepare.
function curl_test($args)
{
    $curl_args = array(
            CURLOPT_FOLLOWLOCATION => true,
            CURLOPT_RETURNTRANSFER => true,
//            CURLOPT_VERBOSE => true,
    );

    if (strpos($args["url"], 'http://') === false) {
        $curl_args[CURLOPT_URL] = IA_URL_HOST . $args['url'];
    } else {
        $curl_args[CURLOPT_URL] = $args['url'];
    }

    // Properly encode POSTFIELDS
    if (isset($args['post'])) {
        $post_args = array();
        // Properly encode array args. Fucking awesome
        // FIXME: does not properly handle multiple levels.
        // FIXTHEM: people who use that deserve to die.
        foreach ($args['post'] as $name => $val) {
            if (is_array($val)) {
                // Oh yeah, pure evil baby.
                foreach ($val as $k => $v) {
                    $post_args["{$name}[{$k}]"] = $v;
                }
            } else {
                $post_args[$name] = $val;
            }
        }
        $curl_args[CURLOPT_POSTFIELDS] = $post_args;
    }

    if (isset($args['user'])) {
        if (array_key_exists('pwd', $args)) {
            $curl_args[CURLOPT_USERPWD] = $args['user'].':'.$args['pwd'];
        } else {
            $curl_args[CURLOPT_USERPWD] = $args['user'].':pwd';
        }
    }

    $ch = curl_init();
    curl_setopt_array($ch, $curl_args);

    $content = curl_exec($ch);
    if ($content === false) {
        log_print_r($curl_args);
        log_error("Failed curling.");
    }
    $res = curl_getinfo($ch);
    $res['content'] = $content;
    curl_close($ch);

    if (getattr($args, 'validate_html', false)) {
        validate_html($content);
    }

    return $res;
}

// Validates a string as a html document.
// Uses the external "validate" tool, packages wdg-html-validator in debian.
// dies on failure.
function validate_html($content)
{
    $fname = IA_ROOT_DIR . 'tests/temp.html';
    file_put_contents($fname, $content);

    $result = shell_exec("validate --warn --verbose $fname");

    if (strstr($result, 'Error') || strstr($result, 'Warning')) {
        log_print($result);
        log_error("HTML validation failed");
    }
}

// Create test users.
function test_prepare()
{
    $user = user_init();
    $user['email'] = "no@spam.com";

    $user['username'] = 'test_dudE1';
    $user['password'] = user_hash_password('pwd', $user['username']);
    $user['full_name'] = 'Testing Dude 1';
    log_assert(user_create($user));

    $user['username'] = 'teSt_dude2';
    $user['password'] = user_hash_password('pwd', $user['username']);
    $user['full_name'] = 'Testing Dude 2';
    log_assert(user_create($user));

    $user['username'] = 'teSt_helper1';
    $user['password'] = user_hash_password('pwd', $user['username']);
    $user['full_name'] = 'Testing Helper 1';
    $user['email'] = 'no@spam.com';
    $user['security_level'] = 'helper';
    log_assert(user_create($user));

    $user['username'] = 'test_hElper2';
    $user['password'] = user_hash_password('pwd', $user['username']);
    $user['full_name'] = 'Testing Helper 2';
    $user['security_level'] = 'helper';
    log_assert(user_create($user));

    $user['username'] = 'tEst_adMin';
    $user['password'] = user_hash_password('pwd', $user['username']);
    $user['full_name'] = 'Testing Admin';
    $user['email'] = 'no@spam.com';
    $user['security_level'] = 'admin';
    log_assert(user_create($user));
}

// Cleanup for testing.
// Warning: bugs might fuck the db.
// However, tests are only run on a local copy anyway.
function test_cleanup()
{
    $test_accounts = "('test_dudE1', 'teSt_dude2', 'teSt_helper1', 'test_hElper2', 'tEst_adMin')";
    db_query("DELETE FROM ia_user WHERE `username` IN $test_accounts");
    db_query("DELETE FROM ia_smf_members WHERE `memberName` IN $test_accounts");
    db_query("DELETE FROM ia_task WHERE `id` LIKE 'test\_%'");
    db_query("DELETE FROM ia_round WHERE `id` LIKE 'test\_%'");
    // Remove various stuff from the wiki.
    $prefixes = array('sandbox', 'utilizator', 'runda', 'problema');
    foreach ($prefixes as $prefix) {
        db_query("DELETE FROM `ia_textblock` ".
                 "WHERE `name` LIKE '$prefix/test\_%'");
        db_query("DELETE FROM `ia_textblock_revision` ".
                 "WHERE `name` LIKE '$prefix/test\_%'");
        db_query("DELETE FROM `ia_file` ".
                 "WHERE `page` LIKE '$prefix/test\_%'");
    }
}

db_connect();
mem_cache_purge();
disk_cache_purge();
check_requirements();

// Add log timestamps.
define("IA_LOG_TIMESTAMP_FORMAT", "Y-m-d H:i:s");

?>