flights.php

<?php

include_once dirname(__FILE__) . '/config.php';

// Test cases for php/flights.php

/**
 * Check demo user map
 */
class BlockAnonExportCase extends WebTestCase {
    public function test() {
        global $webroot;

        $params = array("export" => "true");
        $this->get($webroot . "php/flights.php", $params);
        $this->assertText("You must be logged in to export.");
    }
}

class ExportAirlineToCSVCase extends WebTestCase {
    public function test() {
        global $webroot, $route;

        // First figure out the correct results
        $dbh = db_connect();
        $sth = $dbh->prepare("SELECT alid FROM airlines WHERE iata = ?");
        if (!$sth->execute([$route["core_al_iata"]])) {
            die($sth->errorInfo());
        }
        $row = $sth->fetch();
        if ($row) {
            $alid = $row["alid"];
        }

        $sth = $dbh->prepare("SELECT COUNT(*) AS count FROM routes WHERE alid = ?");
        if (!$sth->execute([$alid])) {
            die($sth->errorInfo());
        }
        $row = $sth->fetch();
        if ($row) {
            $rowCount = intval($row["count"]);
        }

        // Then test
        assert_login($this);
        $params = array("export" => "export", "id" => "L" . $alid);
        $csv = $this->get($webroot . "php/flights.php", $params);
        $rows = explode("\n", $csv);
        $this->assertEqual(count($rows), $rowCount + 1);
    }
}