/**
 * @file generator.h
 * 
 * @brief Tests to test the Generator class generator_t
 * 
 */

/*
 * Copyright (C) 2005 Jan Hutter, Martin Willi
 * Hochschule fuer Technik Rapperswil
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the
 * Free Software Foundation; either version 2 of the License, or (at your
 * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
 *
 * This program 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 General Public License
 * for more details.
 */
 
#include <freeswan.h>
#include <pluto/constants.h>
#include <pluto/defs.h>

#include "../allocator.h"
#include "generator_test.h"
#include "../tester.h"
#include "../encodings.h"
#include "../generator.h"
#include "../encodings/ike_header.h"

extern payload_info_t *payload_infos[];

void test_generator_with_unsupported_payload(tester_t *tester)
{
	generator_t *generator;
	void * data_struct;
	chunk_t generated_data;
	
	generator = generator_create(payload_infos);
	tester->assert_true(tester,(generator != NULL), "generator create check");
	tester->assert_true(tester,(generator->generate_payload(generator,(payload_type_t) -1,data_struct,&generated_data) == NOT_SUPPORTED),"generate_payload call check");
	
	tester->assert_true(tester,(generator->destroy(generator) == SUCCESS), "generator destroy call check");
}

void test_generator_with_header_payload(tester_t *tester)
{
	generator_t *generator;
	ike_header_t header_data;
	chunk_t generated_data;
	
	header_data.initiator_spi = 1;
	header_data.responder_spi = 2;
	header_data.next_payload = 3;
	header_data.maj_version = 4;
	header_data.min_version = 5;
	header_data.exchange_type = 6;
	header_data.flags.initiator = TRUE;
	header_data.flags.version = TRUE;
	header_data.flags.response = TRUE;
	header_data.message_id = 7;
	header_data.length = 8;
	
	generator = generator_create(payload_infos);
	tester->assert_true(tester,(generator != NULL), "generator create check");
	tester->assert_true(tester,(generator->generate_payload(generator,HEADER,&header_data,&generated_data) == SUCCESS),"generate_payload call check");

	DBG_dump("test:",generated_data.ptr,generated_data.len);
	
	allocator_free_chunk(generated_data);
	
	tester->assert_true(tester,(generator->destroy(generator) == SUCCESS), "generator destroy call check");
}