From f4253c72f0737752ffee29e1812e29990bc9ef40 Mon Sep 17 00:00:00 2001
From: Rich Evans <rich.evans@arm.com>
Date: Wed, 14 Jan 2015 19:23:00 +0000
Subject: [PATCH] modify tests/scripts/generate_code.pl to allow overriding of 
 from command line

---
 tests/scripts/generate_code.pl | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/scripts/generate_code.pl b/tests/scripts/generate_code.pl
index 318629fe1..6b195e70b 100755
--- a/tests/scripts/generate_code.pl
+++ b/tests/scripts/generate_code.pl
@@ -6,11 +6,11 @@ use strict;
 my $suite_dir = shift or die "Missing suite directory";
 my $suite_name = shift or die "Missing suite name";
 my $data_name = shift or die "Missing data name";
+my $test_main_file = do { my $arg = shift; defined($arg) ? $arg :  $suite_dir."/main_test.function" };
 my $test_file = $data_name.".c";
 my $test_helper_file = $suite_dir."/helpers.function";
 my $test_case_file = $suite_dir."/".$suite_name.".function";
 my $test_case_data = $suite_dir."/".$data_name.".data";
-my $test_main_file = $suite_dir."/main_test.function";
 
 my $line_separator = $/;
 undef $/;