// -*- mode: c++; c-basic-offset:4 -*-

// This file is part of libdap, A C++ implementation of the OPeNDAP Data
// Access Protocol.

// Copyright (c) 2002,2003 OPeNDAP, Inc.
// Author: James Gallagher <jgallagher@opendap.org>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//
// You can contact OPeNDAP, Inc. at PO Box 112, Saunderstown, RI. 02874-0112.

#include <cppunit/TextTestRunner.h>
#include <cppunit/extensions/TestFactoryRegistry.h>
#include <cppunit/extensions/HelperMacros.h>

#ifndef TM_IN_SYS_TIME
#include <time.h>
#else
#include <sys/time.h>
#endif

#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>   // for stat
#include <string>
#include <sstream>

//#define DODS_DEBUG

#include "GNURegex.h"

#include "Error.h"
#include "mime_util.h"
#include "debug.h"
#include "test_config.h"

using namespace CppUnit;
using namespace std;
using namespace libdap;

class cgiUtilTest: public TestFixture {
private:

protected:
    bool re_match(Regex &r, const string &s)
    {
	int match = r.match(s.c_str(), s.length());
	DBG(cerr << "RE Match: " << match << endl);
	return match == (int) s.length();
    }

public:
    cgiUtilTest()
    {
    }
    ~cgiUtilTest()
    {
    }

    void setUp()
    {
    }

    void tearDown()
    {
    }

CPPUNIT_TEST_SUITE( cgiUtilTest );
#if 1
	CPPUNIT_TEST(name_path_test);
	CPPUNIT_TEST(set_mime_text_test);
	CPPUNIT_TEST(rfc822_date_test);
	CPPUNIT_TEST(last_modified_time_test);
	CPPUNIT_TEST(read_multipart_headers_test);
#endif
	CPPUNIT_TEST(get_next_mime_header_test);

    CPPUNIT_TEST_SUITE_END();

    void name_path_test()
    {
	CPPUNIT_ASSERT(name_path(string("stuff")) == "stuff");
	CPPUNIT_ASSERT(name_path(string("stuff.Z")) == "stuff.Z");
	CPPUNIT_ASSERT(name_path(string("/usr/local/src/stuff.Z")) == "stuff.Z");
	CPPUNIT_ASSERT(name_path(string("/usr/local/src/stuff.tar.Z")) == "stuff.tar.Z");
	CPPUNIT_ASSERT(name_path(string("/usr/local/src/stuff")) == "stuff");
	CPPUNIT_ASSERT(name_path(string("/usr/local/src/#usr#local#src#stuff")) == "stuff");
	CPPUNIT_ASSERT(name_path(string("/usr/local/src/#usr#local#src#stuff.hdf")) == "stuff.hdf");
	CPPUNIT_ASSERT(name_path(string("")) == "");
    }

    // See note above. jhrg 1/18/06
    void set_mime_text_test()
    {
	Regex
		r1(
			"HTTP/1.0 200 OK\r\n\
XDODS-Server: dods-test/0.00\r\n\
XOPeNDAP-Server: dods-test/0.00\r\n\
XDAP: .*\r\n\
Date: (.*)\r\n\
Last-Modified: \\1\r\n\
Content-Type: text/plain\r\n\
Content-Description: dods_das\r\n\
\r\n.*");
	ostringstream oss;
	set_mime_text(oss, dods_das, "dods-test/0.00");
	DBG(cerr << "DODS DAS" << endl << oss.str());
	CPPUNIT_ASSERT(re_match(r1, oss.str()));

	Regex
		r2(
			"HTTP/1.0 200 OK\r\n\
XDODS-Server: dods-test/0.00\r\n\
XOPeNDAP-Server: dods-test/0.00\r\n\
XDAP: .*\r\n\
Date: (.*)\r\n\
Last-Modified: \\1\r\n\
Content-Type: text/plain\r\n\
Content-Description: dods_dds\r\n\
\r\n.*");

	oss.str("");
	set_mime_text(oss, dods_dds, "dods-test/0.00");
	DBG(cerr << "DODS DDS" << endl << oss);
	CPPUNIT_ASSERT(re_match(r2, oss.str()));

	struct tm tm = { 0, 0, 0, 1, 0, 100, 0, 0, 0, 0, 0 }; // 1 Jan 2000
	time_t t = mktime(&tm);
	Regex
		r3(
			"HTTP/1.0 200 OK\r\n\
XDODS-Server: dods-test/0.00\r\n\
XOPeNDAP-Server: dods-test/0.00\r\n\
XDAP: .*\r\n\
Date: .*\r\n\
Last-Modified: (Sat, 01 Jan 2000 ..:00:00 GMT|Fri, 31 Dec 1999 ..:00:00 GMT)\r\n\
Content-Type: text/plain\r\n\
Content-Description: dods_dds\r\n\
\r\n.*");

	oss.str("");
	set_mime_text(oss, dods_dds, "dods-test/0.00", x_plain, t);
	DBG(cerr << "set_mime_text_test: oss: " << oss.str() << endl);
	CPPUNIT_ASSERT(re_match(r3, oss.str()));
    }

    void rfc822_date_test()
    {
	time_t t = 0;
	CPPUNIT_ASSERT(rfc822_date(t) == "Thu, 01 Jan 1970 00:00:00 GMT");
	struct tm tm = { 0, 0, 0, 1, 0, 100, 0, 0, 0, 0, 0 }; // 1 Jan 2000
	t = mktime(&tm);
	// This test may fail for some locations since mktime interprets t as
	// the local time and returns the corresponding GMT time.
	Regex r1("(Sat, 01 Jan 2000 ..:00:00 GMT|Fri, 31 Dec 1999 ..:00:00 GMT)");
	CPPUNIT_ASSERT(re_match(r1, rfc822_date(t)));
    }

    void last_modified_time_test()
    {
	time_t t = time(0);
	CPPUNIT_ASSERT(last_modified_time("no-such-file") == t);
	struct stat st;
	stat("test_config.h", &st);
	CPPUNIT_ASSERT(last_modified_time("test_config.h") == st.st_mtime);
	stat("/etc/passwd", &st);
	CPPUNIT_ASSERT(last_modified_time("/etc/passwd") == st.st_mtime);
    }

    void get_next_mime_header_test()
    {
	string test_file = (string) TEST_SRC_DIR
		+ "/cgi-util-tests/multipart_mime_header1.txt";
	FILE *in = fopen(test_file.c_str(), "r");
	if (!in)
	    CPPUNIT_FAIL("Could not open the mime header file.");

	try {
	    CPPUNIT_ASSERT(get_next_mime_header(in) == "--my_boundary");
	    CPPUNIT_ASSERT(get_next_mime_header(in) == "Content-Type: Text/xml; charset=iso-8859-1");
	    CPPUNIT_ASSERT(get_next_mime_header(in) == "Content-Description: dods-ddx");
	    CPPUNIT_ASSERT(get_next_mime_header(in) == "Content-Id: <1234@opendap.org>");
	}
	catch (Error &e) {
	    CPPUNIT_FAIL(e.get_error_message());
	}

	fclose(in);
    }

    void read_multipart_headers_test()
    {
	string test_file = (string) TEST_SRC_DIR
		+ "/cgi-util-tests/multipart_mime_header1.txt";
	FILE *in = fopen(test_file.c_str(), "r");
	if (!in)
	    CPPUNIT_FAIL("Could not open the mime header file.");

	try {
	    read_multipart_headers(in, "text/xml", dods_ddx);
	    CPPUNIT_ASSERT(true);
	}
	catch (Error &e) {
	    CPPUNIT_FAIL(e.get_error_message());
	}

	fclose(in);
    }

};

CPPUNIT_TEST_SUITE_REGISTRATION(cgiUtilTest);

int main(int, char**)
{
    CppUnit::TextTestRunner runner;
    runner.addTest(CppUnit::TestFactoryRegistry::getRegistry().makeTest());

    bool wasSuccessful = runner.run("", false);

    return wasSuccessful ? 0 : 1;
}
