path: root/SolidityScanner.cpp

/*
    This file is part of cpp-ethereum.

    cpp-ethereum 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 3 of the License, or
    (at your option) any later version.

    cpp-ethereum 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.

    You should have received a copy of the GNU General Public License
    along with cpp-ethereum.  If not, see <http://www.gnu.org/licenses/>.
*/
/**
 * @author Christian <c@ethdev.com>
 * @date 2014
 * Unit tests for the solidity scanner.
 */

#include <libsolidity/Scanner.h>
#include <boost/test/unit_test.hpp>

namespace dev
{
namespace solidity
{
namespace test
{

BOOST_AUTO_TEST_SUITE(SolidityScanner)

BOOST_AUTO_TEST_CASE(test_empty)
{
    Scanner scanner(CharStream(""));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
}

BOOST_AUTO_TEST_CASE(smoke_test)
{
    Scanner scanner(CharStream("function break;765  \t  \"string1\",'string2'\nidentifier1"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::FUNCTION);
    BOOST_CHECK_EQUAL(scanner.next(), Token::BREAK);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SEMICOLON);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "765");
    BOOST_CHECK_EQUAL(scanner.next(), Token::STRING_LITERAL);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "string1");
    BOOST_CHECK_EQUAL(scanner.next(), Token::COMMA);
    BOOST_CHECK_EQUAL(scanner.next(), Token::STRING_LITERAL);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "string2");
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "identifier1");
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
}

BOOST_AUTO_TEST_CASE(string_escapes)
{
    Scanner scanner(CharStream("  { \"a\\x61\""));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::LBRACE);
    BOOST_CHECK_EQUAL(scanner.next(), Token::STRING_LITERAL);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "aa");
}

BOOST_AUTO_TEST_CASE(string_escapes_with_zero)
{
    Scanner scanner(CharStream("  { \"a\\x61\\x00abc\""));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::LBRACE);
    BOOST_CHECK_EQUAL(scanner.next(), Token::STRING_LITERAL);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), std::string("aa\0abc", 6));
}

BOOST_AUTO_TEST_CASE(string_escape_illegal)
{
    Scanner scanner(CharStream(" bla \"\\x6rf\" (illegalescape)"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ILLEGAL);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "");
    // TODO recovery from illegal tokens should be improved
    BOOST_CHECK_EQUAL(scanner.next(), Token::ILLEGAL);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ILLEGAL);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
}

BOOST_AUTO_TEST_CASE(hex_numbers)
{
    Scanner scanner(CharStream("var x = 0x765432536763762734623472346;"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::VAR);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ASSIGN);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "0x765432536763762734623472346");
    BOOST_CHECK_EQUAL(scanner.next(), Token::SEMICOLON);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
}

BOOST_AUTO_TEST_CASE(negative_numbers)
{
    Scanner scanner(CharStream("var x = -.2 + -0x78 + -7.3 + 8.9;"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::VAR);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ASSIGN);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SUB);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), ".2");
    BOOST_CHECK_EQUAL(scanner.next(), Token::ADD);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SUB);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "0x78");
    BOOST_CHECK_EQUAL(scanner.next(), Token::ADD);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SUB);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "7.3");
    BOOST_CHECK_EQUAL(scanner.next(), Token::ADD);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLiteral(), "8.9");
    BOOST_CHECK_EQUAL(scanner.next(), Token::SEMICOLON);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
}

BOOST_AUTO_TEST_CASE(locations)
{
    Scanner scanner(CharStream("function_identifier has ; -0x743/*comment*/\n ident //comment"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().start, 0);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().end, 19);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().start, 20);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().end, 23);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SEMICOLON);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().start, 24);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().end, 25);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SUB);
    BOOST_CHECK_EQUAL(scanner.next(), Token::NUMBER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().start, 27);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().end, 32);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().start, 45);
    BOOST_CHECK_EQUAL(scanner.getCurrentLocation().end, 50);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
}

BOOST_AUTO_TEST_CASE(ambiguities)
{
    // test scanning of some operators which need look-ahead
    Scanner scanner(CharStream("<=""<""+ +=a++ =>""<<"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::LTE);
    BOOST_CHECK_EQUAL(scanner.next(), Token::LT);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ADD);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ASSIGN_ADD);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::INC);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ARROW);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SHL);
}

BOOST_AUTO_TEST_CASE(documentation_comments_parsed_begin)
{
    Scanner scanner(CharStream("/// Send $(value / 1000) chocolates to the user"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "Send $(value / 1000) chocolates to the user");
}

BOOST_AUTO_TEST_CASE(multiline_documentation_comments_parsed_begin)
{
    Scanner scanner(CharStream("/** Send $(value / 1000) chocolates to the user*/"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "Send $(value / 1000) chocolates to the user");
}

BOOST_AUTO_TEST_CASE(documentation_comments_parsed)
{
    Scanner scanner(CharStream("some other tokens /// Send $(value / 1000) chocolates to the user"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "Send $(value / 1000) chocolates to the user");
}

BOOST_AUTO_TEST_CASE(multiline_documentation_comments_parsed)
{
    Scanner scanner(CharStream("some other tokens /**\n"
                               "* Send $(value / 1000) chocolates to the user\n"
                               "*/"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "Send $(value / 1000) chocolates to the user");
}

BOOST_AUTO_TEST_CASE(multiline_documentation_no_stars)
{
    Scanner scanner(CharStream("some other tokens /**\n"
                               " Send $(value / 1000) chocolates to the user\n"
                               "*/"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "Send $(value / 1000) chocolates to the user");
}

BOOST_AUTO_TEST_CASE(multiline_documentation_whitespace_hell)
{
    Scanner scanner(CharStream("some other tokens /** \t \r \n"
                               "\t \r  * Send $(value / 1000) chocolates to the user\n"
                               "*/"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "Send $(value / 1000) chocolates to the user");
}

BOOST_AUTO_TEST_CASE(comment_before_eos)
{
    Scanner scanner(CharStream("//"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "");
}

BOOST_AUTO_TEST_CASE(documentation_comment_before_eos)
{
    Scanner scanner(CharStream("///"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "");
}

BOOST_AUTO_TEST_CASE(empty_multiline_comment)
{
    Scanner scanner(CharStream("/**/"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "");
}

BOOST_AUTO_TEST_CASE(empty_multiline_documentation_comment_before_eos)
{
    Scanner scanner(CharStream("/***/"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::EOS);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "");
}

BOOST_AUTO_TEST_CASE(comments_mixed_in_sequence)
{
    Scanner scanner(CharStream("hello_world ///documentation comment \n"
                               "//simple comment \n"
                               "<<"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::IDENTIFIER);
    BOOST_CHECK_EQUAL(scanner.next(), Token::SHL);
    BOOST_CHECK_EQUAL(scanner.getCurrentCommentLiteral(), "documentation comment ");
}

BOOST_AUTO_TEST_CASE(ether_subdenominations)
{
    Scanner scanner(CharStream("wei szabo finney ether"));
    BOOST_CHECK_EQUAL(scanner.getCurrentToken(), Token::ETH_SUB_WEI);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ETH_SUB_SZABO);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ETH_SUB_FINNEY);
    BOOST_CHECK_EQUAL(scanner.next(), Token::ETH_SUB_ETHER);
}

BOOST_AUTO_TEST_SUITE_END()

}
}
} // end namespaces