void ParserTest::testGetPKB(){ Parser p; std::string src; src = "procedure main{\n" "a = y + 1 + 0 + x + b + a;\n" "b = y;\n" "b = x + y + 1 + u;\n" "while a {\n" "x = y; \n" "hhhhh = 2;\n" "while b { \n" "k = 1; } }}\n"; p.setSource(src); p.startParse(); PKB *pkb = p.getPKB(); // Test Procedure/ProcTable CPPUNIT_ASSERT(pkb->getProcedure(1)->getProcName() == "main"); CPPUNIT_ASSERT_EQUAL(1,pkb->getProcedure(1)->getStartProgLine()); CPPUNIT_ASSERT_EQUAL(9,pkb->getProcedure(1)->getEndProgLine()); //Test Parent PARENT_LIST parentList = pkb->getParent(0, 5); PARENT_LIST::iterator parentItr = parentList.begin(); CPPUNIT_ASSERT(parentItr->first == 4); parentList = pkb->getParent(0, 7); parentItr = parentList.begin(); CPPUNIT_ASSERT(parentItr->first == 4); //Test Follow CPPUNIT_ASSERT_EQUAL(6, pkb->getFollows(0, 7).begin()->first); CPPUNIT_ASSERT_EQUAL(1, pkb->getFollows(0, 2).begin()->first); CPPUNIT_ASSERT_EQUAL(3, pkb->getFollows(2, 0).begin()->second); CPPUNIT_ASSERT(pkb->getFollows(4, 0).empty() == true); //Test VarTable VAR_LIST varList = *pkb->getAllVar(); VAR_LIST::iterator varTableItr = varList.begin(); CPPUNIT_ASSERT(*(varTableItr++) == "a"); CPPUNIT_ASSERT(*(varTableItr++) == "y"); CPPUNIT_ASSERT(*(varTableItr++) == "hhhhh"); CPPUNIT_ASSERT(*(varTableItr++) == "x"); //Test Modifies MODIFIES_LIST modifiesList = pkb->getModifies(PROCEDURE, 1, 0); MODIFIES_LIST::iterator modifiesListItr = modifiesList.begin(); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "a"); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "x"); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "b"); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "hhhhh"); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "k"); //Test Modifies While LOOP modifiesList = pkb->getModifies(WHILE, 4, 0); modifiesListItr = modifiesList.begin(); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "x"); CPPUNIT_ASSERT( pkb->getVarName((modifiesListItr++->second)) == "hhhhh"); //Test Uses USES_LIST useList = pkb->getUses(PROCEDURE, 1, 0); CPPUNIT_ASSERT(useList.size() == 5); //Test Uses WHILE LOOP useList = pkb->getUses(WHILE, 4, 0); CPPUNIT_ASSERT(useList.size() == 3); // 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()->getFirstDescendant()) == VARIABLE); CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == PLUS); ast = ast->getFirstDescendant()->getFirstDescendant()->getRightSibling(); CPPUNIT_ASSERT(pkb->getType(ast) == ASSIGNMENT); CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()) == ASSIGNMENT); CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()->getRightSibling()) == WHILE); ast = ast->getRightSibling()->getRightSibling(); CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getRightSibling()) == STMT_LIST); CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == ASSIGNMENT); ast = ast->getFirstDescendant()->getRightSibling()->getFirstDescendant(); CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()) == ASSIGNMENT); CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()->getRightSibling()) == WHILE); }
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); }