From 3515ee98ff75a0dd60d1e481e37cb47b7ed51022 Mon Sep 17 00:00:00 2001
From: Alex Beregszaszi <alex@rtfs.hu>
Date: Wed, 14 Jun 2017 20:51:00 +0100
Subject: Add tests for when/unless/if in LLL

---
 test/liblll/EndToEndTest.cpp | 62 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 62 insertions(+)

(limited to 'test/liblll')

diff --git a/test/liblll/EndToEndTest.cpp b/test/liblll/EndToEndTest.cpp
index 3928ff45..20fba72b 100644
--- a/test/liblll/EndToEndTest.cpp
+++ b/test/liblll/EndToEndTest.cpp
@@ -57,6 +57,68 @@ BOOST_AUTO_TEST_CASE(panic)
 	BOOST_REQUIRE(m_output.empty());
 }
 
+BOOST_AUTO_TEST_CASE(when)
+{
+	char const* sourceCode = R"(
+		(returnlll
+			(seq
+				(when (= (calldatasize) 0) (return 1))
+				(when (!= (calldatasize) 0) (return 2))))
+	)";
+	compileAndRun(sourceCode);
+	BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
+	BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
+}
+
+BOOST_AUTO_TEST_CASE(unless)
+{
+	char const* sourceCode = R"(
+		(returnlll
+			(seq
+				(unless (!= (calldatasize) 0) (return 1))
+				(unless (= (calldatasize) 0) (return 2))))
+	)";
+	compileAndRun(sourceCode);
+	BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
+	BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
+}
+
+BOOST_AUTO_TEST_CASE(conditional_literal)
+{
+	char const* sourceCode = R"(
+		(returnlll
+			(seq
+				(return (if (= (calldatasize) 0) 1 2))))
+	)";
+	compileAndRun(sourceCode);
+	BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
+	BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
+}
+
+BOOST_AUTO_TEST_CASE(conditional)
+{
+	char const* sourceCode = R"(
+		(returnlll
+			(seq
+				(if (= (calldatasize) 0) (return 1) (return 2))))
+	)";
+	compileAndRun(sourceCode);
+	BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
+	BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
+}
+
+BOOST_AUTO_TEST_CASE(conditional_seq)
+{
+	char const* sourceCode = R"(
+		(returnlll
+			(seq
+				(return (if (= (calldatasize) 0) { 0 2 1 } { 0 1 2 }))))
+	)";
+	compileAndRun(sourceCode);
+	BOOST_CHECK(callContractFunction("test()") == encodeArgs(u256(2)));
+	BOOST_CHECK(callFallback() == toBigEndian(u256(1)));
+}
+
 BOOST_AUTO_TEST_CASE(exp_operator_const)
 {
 	char const* sourceCode = R"(
-- 
cgit