void ParserTest::testMultipleProcedures()
{
Parser p;
std::string src;
src = "procedure main{\n"
" a = a + 2;}\n"
"procedure text1{\n"
" b = a * (2 + c);\n"
" c = c - 2;}"
"procedure text2{\n"
" while a{\n"
" d = d - 2;}\n"
" e = e + 1;\n"
" if a then {\n"
" h = h * (a * b);\n"
" b = b + 1;}\n"
" else{\n"
" c = b;\n"
" while z{\n"
" abc = xyz;\n"
" abc1 = xyz2 * xyz + h - y;}\n"
" a=abc1*2;}}";
p.setSource(src);
CPPUNIT_ASSERT_EQUAL(1, p.startParse());
PKB *pkb = p.getPKB();
// AST Test
AST *ast = pkb->getRootAST();
CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()) == PROCEDURE);
CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()->getRightSibling()) == PROCEDURE);
CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()->getRightSibling()->getFirstDescendant()->getFirstDescendant()) == WHILE);
// Parent Test to ensure insertion is correct.
CPPUNIT_ASSERT(!pkb->isParent(1, 9));
CPPUNIT_ASSERT(pkb->isParent(4, 5));
// Follow Test
CPPUNIT_ASSERT(!pkb->isFollows(1, 2));
CPPUNIT_ASSERT(!pkb->isFollows(3, 4));
CPPUNIT_ASSERT(pkb->isFollows(4, 6));
}
void ParserTest::testIf()
{
Parser p;
std::string src;
// Follows(2, 3), Follows(4, 5)
//cout << "Start=====\n";
src = "procedure test1{\n"
/*1*/ "if b then {"
/*2*/ "a = c;"
/*3*/ "a = d;}"
/**/ "else{"
/*4*/ "c = d;"
/*5*/ "e = d;}}";
p.setSource(src);
CPPUNIT_ASSERT_EQUAL(1, p.startParse());
PKB *pkb = p.getPKB();
CPPUNIT_ASSERT(pkb->isFollows(1, 2) == false);
CPPUNIT_ASSERT(pkb->isFollows(1, 3) == false);
CPPUNIT_ASSERT(pkb->isFollows(2, 3) == true);
CPPUNIT_ASSERT(pkb->isFollows(4, 5) == true);
CPPUNIT_ASSERT(pkb->isFollows(2, 4) == false);
CPPUNIT_ASSERT(pkb->isFollows(2, 5) == false);
CPPUNIT_ASSERT(pkb->isFollows(2, 6) == false);
CPPUNIT_ASSERT(pkb->isFollows(3, 4) == false);
CPPUNIT_ASSERT(pkb->isFollows(4, 6) == false);
// Test Uses and Modifies for if
MODIFIES_LIST modifiesList = pkb->getModifies(IF, 1, 0);
MODIFIES_LIST::iterator modifiesListItr = modifiesList.begin();
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "a");
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "c");
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "e");
USES_LIST useList = pkb->getUses(IF, 1, 0);
USES_LIST::iterator usesListItr = useList.begin();
CPPUNIT_ASSERT( pkb->getVarName((usesListItr++->second)) == "b");
CPPUNIT_ASSERT( pkb->getVarName((usesListItr++->second)) == "c");
CPPUNIT_ASSERT( pkb->getVarName((usesListItr++->second)) == "d");
// Follows(2, 3), Follows(3, 4), Follows(7, 8), Follows(1, 9)
// Parent(1, 2) (1, 3), (1, 4), (4, 5), (4, 6), (1, 7), (1, 8)
src = "procedure test2{\n"
/*1*/ "if b then {"
/*2*/ "a = c;"
/*3*/ "a = d;"
/*4*/ "if b then{"
/*5*/ "a = 1;}"
/**/ "else{"
/*6*/ "b = 1;}}"
/**/ "else{"
/*7*/ "c = d;"
/*8*/ "e = d;}"
/*9*/ "z = y;}";
p.setSource(src);
CPPUNIT_ASSERT_EQUAL(1, p.startParse());
pkb = p.getPKB();
CPPUNIT_ASSERT(pkb->isFollows(2, 3) == true);
CPPUNIT_ASSERT(pkb->isFollows(3, 4) == true);
CPPUNIT_ASSERT(pkb->isFollows(7, 8) == true);
CPPUNIT_ASSERT(pkb->isFollows(1, 9) == true);
CPPUNIT_ASSERT(pkb->isFollows(1, 8) == false);
CPPUNIT_ASSERT(pkb->isFollows(1, 2) == false);
CPPUNIT_ASSERT(pkb->isFollows(1, 4) == false);
CPPUNIT_ASSERT(pkb->isParent(1, 2) == true);
CPPUNIT_ASSERT(pkb->isParent(1, 3) == true);
CPPUNIT_ASSERT(pkb->isParent(1, 4) == true);
CPPUNIT_ASSERT(pkb->isParent(4, 5) == true);
CPPUNIT_ASSERT(pkb->isParent(4, 6) == true);
CPPUNIT_ASSERT(pkb->isParent(1, 7) == true);
CPPUNIT_ASSERT(pkb->isParent(1, 8) == true);
CPPUNIT_ASSERT(pkb->isParent(1, 9) == false);
// Test Uses and Modifies for if
modifiesList = pkb->getModifies(IF, 1, 0);
modifiesListItr = modifiesList.begin();
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "b");
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "a");
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "c");
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "e");
modifiesList = pkb->getModifies(IF, 4, 0);
modifiesListItr = modifiesList.begin();
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "b");
CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "a");
useList = pkb->getUses(IF, 1, 0);
usesListItr = useList.begin();
CPPUNIT_ASSERT( pkb->getVarName((usesListItr++->second)) == "b");
CPPUNIT_ASSERT( pkb->getVarName((usesListItr++->second)) == "c");
CPPUNIT_ASSERT( pkb->getVarName((usesListItr++->second)) == "d");
// varindex 7 is variable y
CPPUNIT_ASSERT(pkb->isUses(IF, 1, 7) == false);
// Follows(1, 2), Follows(3, 4), Follows(5, 6), Follows(2, 7)
src = "procedure test3{\n"
"a = y;"
"if b then {"
"a = c;"
"a = d;}"
"else{"
"c = d;"
"e = d;}"
"a = 1;}";
p.setSource(src);
CPPUNIT_ASSERT_EQUAL(1, p.startParse());
pkb = p.getPKB();
// Test Follows, Parent
CPPUNIT_ASSERT(pkb->isFollows(1, 2) == true);
CPPUNIT_ASSERT(pkb->isFollows(3, 4) == true);
CPPUNIT_ASSERT(pkb->isFollows(5, 6) == true);
CPPUNIT_ASSERT(pkb->isFollows(2, 7) == true);
CPPUNIT_ASSERT(pkb->isFollows(1, 3) == false);
CPPUNIT_ASSERT(pkb->isFollows(1, 4) == false);
CPPUNIT_ASSERT(pkb->isFollows(1, 5) == false);
CPPUNIT_ASSERT(pkb->isFollows(1, 6) == false);
CPPUNIT_ASSERT(pkb->isParent(2, 3) == true);
CPPUNIT_ASSERT(pkb->isParent(2, 4) == true);
CPPUNIT_ASSERT(pkb->isParent(2, 5) == true);
CPPUNIT_ASSERT(pkb->isParent(2, 6) == true);
CPPUNIT_ASSERT(pkb->isParent(2, 7) == false);
// Test AST
AST *ast = pkb->getRootAST();
CPPUNIT_ASSERT(ast->getRootData() == 1);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()) == STMT_LIST);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()) == ASSIGNMENT);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == IF);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == STMT_LIST);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == ASSIGNMENT);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getRightSibling()) == STMT_LIST);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getRightSibling()->getFirstDescendant()) == ASSIGNMENT);
}
