From 49958e4720528e4ba325dfc0892888f319649aa9 Mon Sep 17 00:00:00 2001
From: chriseth <chris@ethereum.org>
Date: Thu, 11 Oct 2018 17:45:12 +0200
Subject: Move main function tests.

---
 test/libjulia/YulOptimizerTest.cpp | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'test/libjulia/YulOptimizerTest.cpp')

diff --git a/test/libjulia/YulOptimizerTest.cpp b/test/libjulia/YulOptimizerTest.cpp
index 9f028bcd..0e55c87d 100644
--- a/test/libjulia/YulOptimizerTest.cpp
+++ b/test/libjulia/YulOptimizerTest.cpp
@@ -29,6 +29,7 @@
 #include <libjulia/optimiser/FunctionHoister.h>
 #include <libjulia/optimiser/ExpressionInliner.h>
 #include <libjulia/optimiser/FullInliner.h>
+#include <libjulia/optimiser/MainFunction.h>
 
 #include <libsolidity/parsing/Scanner.h>
 #include <libsolidity/inlineasm/AsmPrinter.h>
@@ -121,6 +122,12 @@ bool YulOptimizerTest::run(ostream& _stream, string const& _linePrefix, bool con
 		(FunctionGrouper{})(*m_ast);
 		FullInliner(*m_ast).run();
 	}
+	else if (m_optimizerStep == "mainFunction")
+	{
+		disambiguate();
+		(FunctionGrouper{})(*m_ast);
+		(MainFunction{})(*m_ast);
+	}
 	else
 	{
 		FormattedScope(_stream, _formatted, {formatting::BOLD, formatting::RED}) << _linePrefix << "Invalid optimizer step: " << m_optimizerStep << endl;
-- 
cgit v1.2.3