#ifndef SSL_TEST
#define SSL_TEST

#include "munit/munit.h"
#include "../src/ssl.h"
#include "../src/util.h"


MunitResult testNewRsaKeySave(const MunitParameter params[],
		void* user_data_or_fixture){
	char keyFile[] = "/tmp/yaip-test-keyfile";
	create_and_save_key( keyFile );
	//TODO: make better tests.
	// For now checking there is at least 3 lines seems ok
	munit_assert_int( 3, <, countLines(keyFile) );
	remove(keyFile);

	return MUNIT_OK;
}

MunitResult testNewCACertificate(const MunitParameter params[],
		void* user_data_or_fixture){
	EVP_PKEY *key = generate_ca_key();
	const char *nameEntry =munit_parameters_get(params, "Name Entry" );

	X509 *x509 = generate_ca_cert(key);
	munit_assert_not_null( x509 );

	X509_NAME *name = X509_get_subject_name(x509);
	munit_assert_not_null( name );

	char buff[200] = {'\0'};

	if ( strcmp( nameEntry, "C" ) == 0 ){
		X509_NAME_get_text_by_NID( name, NID_countryName, buff, 200 );
		munit_assert_string_equal( buff, "UK");
	} else if ( strcmp( nameEntry, "O" ) == 0 ){
		X509_NAME_get_text_by_NID( name, NID_organizationName, buff, 200 );
		munit_assert_string_equal( buff, "Yet Another Intercepting Proxy");
	} else if ( strcmp( nameEntry, "CN" ) == 0 ){
		X509_NAME_get_text_by_NID( name, NID_commonName, buff, 200 );
		munit_assert_string_equal( buff, "yaip.yaip");
	}

	return MUNIT_OK;
}

MunitResult testNewCertificateSave(const MunitParameter params[],
		void* user_data_or_fixture){
	char certFile[] = "/tmp/yaip-test-certfile";
	EVP_PKEY *key = generate_ca_key();
	create_and_save_cert( certFile, key );
	//TODO: make better tests.
	// For now checking there is at least 3 lines seems ok
	munit_assert_int( 3, <, countLines(certFile) );
	remove( certFile );

	return MUNIT_OK;
}

MunitResult testCerlistCount(const MunitParameter params[],
		void* user_data_or_fixture){
	CertList *head = NULL;
	CertList *last = head;

	unsigned int count = atoi(munit_parameters_get(params, "Count" ));
	for ( unsigned int i = 0; i < count; i++ ){
		CertList *newItem = malloc(sizeof( CertList ));
		newItem->next = NULL;
		if ( head == NULL ){
			head = newItem;
			last = newItem;
		} else {
			last->next = newItem;
			last = last->next;
		}
	}
	munit_assert_int( countCertListItems(head), ==, count );

	return MUNIT_OK;
}

MunitResult testCerlistLast(const MunitParameter params[],
		void* user_data_or_fixture){

	const char *countstr = munit_parameters_get(params, "Count" );
	//const char *countstr = "2";
	const unsigned int count = atoi(countstr);

	CertList *head = NULL;
	CertList *last = head;

	for ( unsigned int i = 0; i < count; i++ ){
		CertList *newItem = malloc(sizeof( CertList ));
		char str[5] = {'\0'};
		sprintf(str, "%d", i+1);
		newItem->next = NULL;
		newItem->host = strdup(str);
		if ( head == NULL ){
			head = newItem;
			last = newItem;
		} else {
			last->next = newItem;
			last = last->next;
		}
	}

	CertList *lastCert = getLastCertListItem(head);
	if ( count == 0 ){
		munit_assert_null( lastCert );
	} else {
		munit_assert_int( countCertListItems(head), ==, count );
		munit_assert_string_equal( countstr, lastCert->host );
	}
	return MUNIT_OK;
}

MunitResult testNewHostCertificate(const MunitParameter params[],
		void* user_data_or_fixture){
	
	EVP_PKEY *key = generate_ca_key();
	X509 *x509 = generate_ca_cert(key);
	X509 *siteCert = generate_site_cert( key, x509, "example.com" );
	const char *nameType = munit_parameters_get(params, "Name" );
	X509_NAME *name;

	char buff[200] = {'\0'};

	if ( strcmp( nameType, "subject" ) ){
		name = X509_get_subject_name(siteCert);
		X509_NAME_get_text_by_NID( name, NID_commonName, buff, 200 );
		munit_assert_string_equal( buff, "example.com" );
	} else if ( strcmp( nameType, "issuer" ) ){
		name = X509_get_issuer_name(siteCert);
		X509_NAME_get_text_by_NID( name, NID_commonName, buff, 200 );
		munit_assert_string_equal( buff, "yaip.yaip" );
	} else {
		return MUNIT_ERROR;
	}
	//FILE * pkey_file = fopen("examplecert.pem", "wb");
    //// Write the cert to disk.
	//PEM_write_X509( pkey_file, siteCert );
    //fclose(pkey_file);

	return MUNIT_OK;
}

MunitResult testFindCertListItem(const MunitParameter params[],
		void* user_data_or_fixture){
	
	CertList *head = newCertListItem( "example.com", NULL, NULL );
	CertList *last = head;
	CertList *curr;
	for ( unsigned int i = 0; i < 10; i++ ){
		char host[15] = {'\0'};
		sprintf(host, "example%d.com", i+1);
		last->next = newCertListItem( strdup(host), NULL, NULL );
		last = last->next;
	}

	munit_assert_int( countCertListItems( head ), ==, 11 );

	curr = findCertListItem( head, "example.com" );
	munit_assert_not_null( curr );
	munit_assert_string_equal( curr->host, "example.com" );

	curr = findCertListItem( head, "example1.com" );
	munit_assert_not_null( curr );
	munit_assert_string_equal( curr->host, "example1.com" );

	curr = findCertListItem( head, "example5.com" );
	munit_assert_not_null( curr );
	munit_assert_string_equal( curr->host, "example5.com" );

	curr = findCertListItem( head, "doesnt-exist.com" );
	munit_assert_null( curr );
	return MUNIT_OK;
}

static char* count_parameters[] = {
  "0", "1", "2", "10", "50", NULL
};

static char* x509_name_parameters[] = {
  "C", "O", "CN", NULL
};

static char* x509_issuer_subject_parameters[] = {
  "issuer", "subject", NULL
};

static MunitParameterEnum count_params[] = {
  { "Count", count_parameters },
  { NULL, NULL },
};

static MunitParameterEnum x509_name_params[] = {
  { "Name Entry", x509_name_parameters },
  { NULL, NULL },
};

static MunitParameterEnum x509_issuer_subject_params[] = {
  { "Name", x509_issuer_subject_parameters },
  { NULL, NULL },
};

static MunitTest ssl_tests[] = {
	//   name        test                setup tear_down   options                 parameters
	{ "/ca/rsa_key/save", testNewRsaKeySave,      NULL, NULL,       MUNIT_TEST_OPTION_NONE, NULL },
	{ "/ca/cert/new",    testNewCACertificate, NULL, NULL,       MUNIT_TEST_OPTION_NONE, x509_name_params },
	{ "/ca/cert/save",    testNewCertificateSave, NULL, NULL,       MUNIT_TEST_OPTION_NONE, NULL },
	{ "/CertList/count",    testCerlistCount, NULL, NULL,       MUNIT_TEST_OPTION_NONE, count_params },
	{ "/CertList/last",    testCerlistLast, NULL, NULL,       MUNIT_TEST_OPTION_NONE, count_params },
	{ "/CertList/find",    testFindCertListItem, NULL, NULL,       MUNIT_TEST_OPTION_NONE, NULL },
	{ "/hostcert/new",    testNewHostCertificate, NULL, NULL,       MUNIT_TEST_OPTION_NONE, x509_issuer_subject_params },
/* Mark the end of the array with an entry where the test
* function is NULL */
	{ NULL, NULL, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL }
};

MunitSuite ssl_test_suite = {
	"/ssl", /* name */
	ssl_tests, /* tests */
	NULL, /* suites */
	1, /* iterations */
	MUNIT_SUITE_OPTION_NONE /* options */
};

#ifndef MAINTEST
#define MAINTEST

int main (int argc, char* argv[]) {
	return munit_suite_main(&ssl_test_suite, NULL, argc, argv);
}

#endif /* ifndef MAINTEST */

#endif