void ParserTest::testCall(){
Parser p;
std::string src;
src = "procedure main{\n"
/*1*/ "a = c;"
/*2*/ "call test2;"
/*3*/ "a = d;}"
/**/ "procedure test2{\n"
/*4*/ "b = 1;}";
p.setSource(src);
CPPUNIT_ASSERT_EQUAL(1, p.startParse());
PKB *pkb = p.getPKB();
AST *ast = pkb->getRootAST();
CPPUNIT_ASSERT(ast->getRootData() == 1);
CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()) == PROCEDURE);
CPPUNIT_ASSERT(pkb->getType(ast->getRightSibling()->getFirstDescendant()->getFirstDescendant()) == ASSIGNMENT);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()) == STMT_LIST);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()) == ASSIGNMENT);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == CALL);
}
RELATION_LIST EvaluateSibling::getSiblingUnknown(TYPE type1,TYPE type2){
RELATION_LIST returnList;
if(type1==PROCEDURE&&type2==PROCEDURE){ //1. sibling ( p1,p2)
int procTableSize = pkb->getProceTableSize();
for(int i =1;i<=procTableSize;i++){
for(int j =1;j<=procTableSize;j++){
if(i!=j) returnList.push_back(pair<int,int>(i,j));
}
}
}
if(Helper::isStatement(type1)&&Helper::isStatement(type2)){//2. sibling(stmt/a/w/if/call,stmt/a/w/if/call)
//get sequential sibling from Follows*
returnList = EvaluateFollows::getFollowsStarResult(type1,type2);
RELATION_LIST tempList;
if(type1!=type2){
tempList = EvaluateFollows::getFollowsStarResult(type2,type1);
}else{
tempList = returnList;
}
//insert reversed sibling list
for(RELATION_LIST::iterator itr=tempList.begin();itr!=tempList.end();itr++){
returnList.push_back(pair<int,int>(itr->second,itr->first));
}
}
if(type1==VARIABLE&&type2==STMT_LIST){//3. a. sibling( x, stmtList)
//get all the whiles
DATA_LIST * whileList = pkb->getAllWhiles();
for(DATA_LIST::iterator itr=whileList->begin();itr!=whileList->end();itr++){
returnList.push_back(pair<int,int>(pkb->getWhileCtrVar(*itr),*itr+1)); // No. of stmtList = stmtNo of while+1
}
delete whileList;
//get all the ifs
DATA_LIST * ifList = pkb->getAllIfs();
for(DATA_LIST::iterator itr=ifList->begin();itr!=ifList->end();itr++){
AST * ifNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==IF){
ifNode = *astItr;
break;
}
}
AST * ifVarNode = ifNode->getFirstDescendant();
AST * elseNode = ifVarNode->getRightSibling();
AST * thenNode = elseNode->getRightSibling();
returnList.push_back(pair<int,int>(ifVarNode->getRootData(),elseNode->getRootStatementNum()));
returnList.push_back(pair<int,int>(ifVarNode->getRootData(),thenNode->getRootStatementNum()));
}
delete ifList;
}
if(type1==STMT_LIST&&type2==VARIABLE){ //3.b sibling(stmtList,x)
//get all the whiles
DATA_LIST * whileList = pkb->getAllWhiles();
for(DATA_LIST::iterator itr=whileList->begin();itr!=whileList->end();itr++){
returnList.push_back(pair<int,int>(*itr+1,pkb->getWhileCtrVar(*itr)));
}
delete whileList;
//get all the ifs
DATA_LIST * ifList = pkb->getAllIfs();
for(DATA_LIST::iterator itr=ifList->begin();itr!=ifList->end();itr++){
AST * ifNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==IF){
ifNode = *astItr;
break;
}
}
AST * ifVarNode = ifNode->getFirstDescendant();
AST * thenNode = ifVarNode->getRightSibling();
AST * elseNode = thenNode->getRightSibling();
returnList.push_back(pair<int,int>(elseNode->getRootStatementNum(),ifVarNode->getRootData()));
returnList.push_back(pair<int,int>(thenNode->getRootStatementNum(),ifVarNode->getRootData()));
}
delete ifList;
}
if(type1==STMT_LIST&&type2==STMT_LIST){ //sibling (stmtList,stmtList),only possible in If stmts
//get all the ifs
DATA_LIST * ifList = pkb->getAllIfs();
for(DATA_LIST::iterator itr=ifList->begin();itr!=ifList->end();itr++){
AST * ifNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==IF){
ifNode = *astItr;
break;
}
}
AST * ifVarNode = ifNode->getFirstDescendant();
AST * thenNode = ifVarNode->getRightSibling();
AST * elseNode = thenNode->getRightSibling();
//(else,then)
returnList.push_back(pair<int,int>(elseNode->getRootStatementNum(),thenNode->getRootStatementNum()));
//(then,else)
returnList.push_back(pair<int,int>(thenNode->getRootStatementNum(),elseNode->getRootStatementNum()));
}
delete ifList;
}
if(Helper::isExprElement(type1)&&Helper::isExprElement(type2)){//4. Sibling (var/plus/minus/multiply,var/plus/minus/multiply).
DATA_LIST * assignList = pkb->getAllAssigns();
for(DATA_LIST::iterator itr= assignList->begin();itr!=assignList->end();itr++){
//get assignNode
AST_LIST * assignNodes = pkb->getASTBy(*itr);
AST * assignNode;
for(AST_LIST::iterator astItr=assignNodes->begin();astItr!=assignNodes->end();astItr++){
if((**astItr).getRootType()==ASSIGNMENT){
assignNode = *astItr;
break;
}
}
//check if modified var and expression tree root are sibling
AST * leftVarNode = assignNode->getFirstDescendant();
AST * expressionRoot = leftVarNode->getRightSibling();
//pattern is symmetric
bool isPattern1 = (leftVarNode->getRootType()==type1)&&(expressionRoot->getRootType()==type2);
bool isPattern2 =(expressionRoot->getRootType()==type1)&&(leftVarNode->getRootType()==type2);
//if in the case of x=x,y=y, pattern1 and pattern2 are same
bool isLeftRightEqual = (leftVarNode->getRootType()==expressionRoot->getRootType())&&(leftVarNode->getRootData()==expressionRoot->getRootData());
if(isPattern1){ //type1 matched with left, type2 matched with right
returnList.push_back(pair<int,int>(leftVarNode->getRootData(),expressionRoot->getRootData()));
}else if(isPattern2){ //to avoid duplicate result,type1 matched with right, type2 matched with left
returnList.push_back(pair<int,int>(expressionRoot->getRootData(),leftVarNode->getRootData()));
// x = expression, x is matched with var or +,-,*
}
//traverse the tree and check matched pattern
stack<AST *> exprStack;
if(expressionRoot->getRootType()!=VARIABLE&&expressionRoot->getRootType()!=CONSTANT){
exprStack.push(expressionRoot);
AST * left;
AST * right;
while(!exprStack.empty()){
AST * curNode = exprStack.top();
exprStack.pop();
left = curNode->getFirstDescendant();
right = left->getRightSibling();
//pattern is symmetric
isPattern1 = (left->getRootType()==type1)&&(right->getRootType()==type2);
isPattern2 =(right->getRootType()==type1)&&(left->getRootType()==type2);
isLeftRightEqual = (left->getRootType()==right->getRootType())&&(left->getRootData()==right->getRootData());
if(isPattern1){
returnList.push_back(pair<int,int>(left->getRootData(),right->getRootData()));
} else if(isPattern2){
returnList.push_back(pair<int,int>(right->getRootData(),left->getRootData()));
// x = expression, x is matched with var or +,-,*
}
//push the next operator into stack
if(left->getRootType()!=VARIABLE&&left->getRootType()!=CONSTANT) exprStack.push(left);
if(right->getRootType()!=VARIABLE&&right->getRootType()!=CONSTANT) exprStack.push(right);
}
}
}
delete assignList;
}
return returnList;
}
RELATION_LIST EvaluateSibling::getOneSibling(TYPE known,TYPE unknown,INDEX knownIndex,int knownPos){
RELATION_LIST returnList;
if(known==PROCEDURE&&unknown==PROCEDURE){ //1. sibling ( "procName",p) or sibling ( p,"procName")
int procTableSize = pkb->getProceTableSize();
for(int i =1;i<=procTableSize;i++){
if(knownIndex!=i){
if(knownPos==1){ //sibling (known,unknown)
returnList.push_back(pair<int,int>(knownIndex,i));
}else{ //sibling(unknown,known)
returnList.push_back(pair<int,int>(i,knownIndex));
}
}
}
}
if(Helper::isStatement(known)&&Helper::isStatement(unknown)){//2. sibling(stmt/a/w/if/call,stmt/a/w/if/call), either field is unknown
if(knownPos==1){
//get sequential sibling from Follows*
returnList = EvaluateFollows::getFollowsStarResult(unknown,knownIndex,0);
//get sibling in front of the knownIndex
RELATION_LIST tempList = EvaluateFollows::getFollowsStarResult(unknown,0,knownIndex);
for(RELATION_LIST::iterator itr=tempList.begin();itr!=tempList.end();itr++){
returnList.push_back(pair<int,int>(knownIndex,itr->first));
}
}else{
//get sibling before current index from Follows*
returnList = EvaluateFollows::getFollowsStarResult(unknown,0,knownIndex);
//get sequential siblings
RELATION_LIST tempList = EvaluateFollows::getFollowsStarResult(unknown,knownIndex,0);
for(RELATION_LIST::iterator itr=tempList.begin();itr!=tempList.end();itr++){
returnList.push_back(pair<int,int>(itr->second,knownIndex));
}
}
}
if(known==VARIABLE&&unknown==STMT_LIST){//3. a. sibling( "x", stmtList) or sibling(stmtList,"x")
//sibling("x",stmtList)
//get all the whiles
DATA_LIST * whileList = pkb->getAllWhiles();
for(DATA_LIST::iterator itr=whileList->begin();itr!=whileList->end();itr++){
if(pkb->getWhileCtrVar(*itr)==knownIndex){
if(knownPos==1)
returnList.push_back(pair<int,int>(knownIndex,*itr+1));
else
returnList.push_back(pair<int,int>(*itr+1,knownIndex));
}
}
delete whileList;
//get all the ifs
DATA_LIST * ifList = pkb->getAllIfs();
for(DATA_LIST::iterator itr=ifList->begin();itr!=ifList->end();itr++){
AST * ifNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==IF){
ifNode = *astItr;
break;
}
}
AST * ifVarNode = ifNode->getFirstDescendant();
AST * thenNode = ifVarNode->getRightSibling();
AST * elseNode = thenNode->getRightSibling();
if(ifVarNode->getRootData()==knownIndex){
if(knownIndex==1){
returnList.push_back(pair<int,int>(knownIndex,thenNode->getRootStatementNum()));
returnList.push_back(pair<int,int>(knownIndex,elseNode->getRootStatementNum()));
}else{
returnList.push_back(pair<int,int>(thenNode->getRootStatementNum(),knownIndex));
returnList.push_back(pair<int,int>(elseNode->getRootStatementNum(),knownIndex));
}
}
}
delete ifList;
}
if(known==STMT_LIST&&unknown==VARIABLE){//3.b sibling(x,2) or sibling(2,x) where 2 is stmtList no.
AST * stmtListNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(knownIndex);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==STMT_LIST){
stmtListNode = *astItr;
break;
}
}
if(stmtListNode!=NULL){
AST * varNode = stmtListNode->getFirstDescendant();
if(varNode->getRootType()==VARIABLE){
if(knownPos==1)
returnList.push_back(pair<int,int>(knownIndex,varNode->getRootData()));
else
returnList.push_back(pair<int,int>(varNode->getRootData(),knownIndex));
}
}
}
if(known==STMT_LIST&&unknown==STMT_LIST){ //sibling (stmtList,2) or sibling(2,stmtList)
//get all the ifs
DATA_LIST * ifList = pkb->getAllIfs();
for(DATA_LIST::iterator itr=ifList->begin();itr!=ifList->end();itr++){
AST * ifNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==IF){
ifNode = *astItr;
break;
}
}
AST * ifVarNode = ifNode->getFirstDescendant();
AST * thenNode = ifVarNode->getRightSibling();
AST * elseNode = thenNode->getRightSibling();
if(knownIndex==elseNode->getRootStatementNum()){
if(knownPos==1)
returnList.push_back(pair<int,int>(knownIndex,thenNode->getRootStatementNum()));
else
returnList.push_back(pair<int,int>(thenNode->getRootStatementNum(),knownIndex));
}else if(knownIndex= thenNode->getRootStatementNum()){
if(knownPos==1)
returnList.push_back(pair<int,int>(knownIndex,elseNode->getRootStatementNum()));
else
returnList.push_back(pair<int,int>(elseNode->getRootStatementNum(),knownIndex));
}
}
delete ifList;
}else if(Helper::isExprElement(known)&&Helper::isExprElement(unknown)){//4. Sibling (var/plus/minus/multiply,var/plus/minus/multiply).
DATA_LIST * assignList = pkb->getAllAssigns();
for(DATA_LIST::iterator itr= assignList->begin();itr!=assignList->end();itr++){
//get assignNode
AST * assignNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==ASSIGNMENT){
assignNode = *astItr;
break;
}
}
//check if modified var and expression tree root are sibling
AST * leftVarNode = assignNode->getFirstDescendant();
AST * expressionRoot = leftVarNode->getRightSibling();
int exprData = expressionRoot->getRootData();
if(expressionRoot->getRootType()==CONSTANT) exprData = pkb->getConstantValue(exprData);
/*pattern is symmetric
if else pattern check is used to avoid repeated result
eg. x->x in expression tree, sibling("x",var)/sibling(var,"x")
if pattern1 is matched, no need to check pattern2
eg. x->+ in expressoin tree, sibling("+",var)/sibling(var,"+")
if pattern1 is matched, pattern2 can't be matched
*/
bool isPattern1 = (leftVarNode->getRootType()==known&&leftVarNode->getRootData()==knownIndex)&&(expressionRoot->getRootType()==unknown);
bool isPattern2 =(expressionRoot->getRootType()==known&&exprData==knownIndex)&&(leftVarNode->getRootType()==unknown);
if(isPattern1){
if(knownPos==1){
returnList.push_back(pair<int,int>(knownIndex,expressionRoot->getRootData()));
}else{
returnList.push_back(pair<int,int>(expressionRoot->getRootData(),knownIndex));
}
} else if(isPattern2){
if(knownPos==1){
returnList.push_back(pair<int,int>(knownIndex,leftVarNode->getRootData()));
}else{
returnList.push_back(pair<int,int>(leftVarNode->getRootData(),knownIndex));
}
} // x = expression, x is matched with var or +,-,*
//traverse the tree and check matched pattern
stack<AST *> exprStack;
if(expressionRoot->getRootType()!=VARIABLE&&expressionRoot->getRootType()!=CONSTANT){
exprStack.push(expressionRoot);
AST * left;
AST * right;
while(!exprStack.empty()){
AST * curNode = exprStack.top();
exprStack.pop();
left = curNode->getFirstDescendant();
right = left->getRightSibling();
int leftData = left->getRootData();
if(left->getRootType()==CONSTANT) leftData = pkb->getConstantValue(leftData);
int rightData = right->getRootData();
if(right->getRootType()==CONSTANT) rightData = pkb->getConstantValue(rightData);
//pattern is symmetric
isPattern1 = (left->getRootType()==known&&leftData==knownIndex)&&(right->getRootType()==unknown);
isPattern2 =(right->getRootType()==known&&rightData==knownIndex)&&(left->getRootType()==unknown);
if(isPattern1){
if(knownPos==1){
returnList.push_back(pair<int,int>(known,right->getRootData()));
}else{
returnList.push_back(pair<int,int>(right->getRootData(),known));
}
} else if(isPattern2){
if(knownPos==1){
returnList.push_back(pair<int,int>(knownIndex,left->getRootData()));
}else{
returnList.push_back(pair<int,int>(left->getRootData(),knownIndex));
}
} // x = expression, x is matched with var or +,-,*
//push the next operator into stack
if(left->getRootType()!=VARIABLE&&left->getRootType()!=CONSTANT) exprStack.push(left);
if(right->getRootType()!=VARIABLE&&right->getRootType()!=CONSTANT) exprStack.push(right);
}
}
}
delete assignList;
}
return returnList;
}
bool EvaluateSibling::getIsSibling(TYPE type1,TYPE type2,INDEX index1,INDEX index2){
bool returnValue = false;
if(type1 ==PROCEDURE&&type2 ==PROCEDURE){ //1. Sibling("procName1","procName2"
if(index1!=NULL&&index2!=NULL) returnValue = true;
}
if(Helper::isStatement(type1)&&Helper::isStatement(type2)){//2. Sibling(1,5),Sibling(4,2) for stmt/w/a/if/call
if((type1==STATEMENT||Helper::isStatementTypeOf(type1,index1))&&(type2==STATEMENT||Helper::isStatementTypeOf(type2,index2))){
returnValue = (EvaluateFollows::getIsFollowsStarResult(index1,index2)||EvaluateFollows::getIsFollowsStarResult(index2,index1));
}
}
if(type1==VARIABLE&&(type2==STMT_LIST||type2==STATEMENT)){//3. a.Sibling("x",4) for while or if stmt,4 is stmtList
AST * stmtListNode =NULL;
AST_LIST * nodeList = pkb->getASTBy(index2);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==STMT_LIST){
stmtListNode = *astItr;
break;
}
}
if(stmtListNode!=NULL){
AST * parentNode = stmtListNode->getAncestor(); //get the parentNode to check if it is while or if
AST * varNode = parentNode->getFirstDescendant(); //get the varNode of while or if
if(varNode->getRootType() ==VARIABLE&&varNode->getRootData()==index1)
returnValue= true;
}
}
if(type2==VARIABLE&&(type1==STMT_LIST||type1==STATEMENT)){ //3. b.Sibling(4,"x") for while of if stmt
AST * stmtListNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(index1);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==STMT_LIST){
stmtListNode = *astItr;
break;
}
}
if(stmtListNode!=NULL){
AST * parentNode = stmtListNode->getAncestor(); //get the parentNode to check if it is while or if
AST * varNode = parentNode->getFirstDescendant(); //get the varNode of while or if
if(varNode->getRootType() ==VARIABLE&&varNode->getRootData()==index2)
returnValue = true;
}
}
if((type1==STMT_LIST||type1==STATEMENT)&&(type2==STMT_LIST||type2==STATEMENT)){//3. c. Sibling( 22,23) for stmtList of if
AST * stmtListNode1 =NULL;
AST_LIST * nodeList = pkb->getASTBy(index1);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==STMT_LIST){
stmtListNode1 = *astItr;
break;
}
}
AST * stmtListNode2=NULL;
nodeList = pkb->getASTBy(index2);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==STMT_LIST){
stmtListNode2 = *astItr;
break;
}
}
if(stmtListNode1!=NULL&&stmtListNode2!=NULL){
if(stmtListNode1->getAncestor()->getRootData()==stmtListNode2->getAncestor()->getRootData())
returnValue= true;
}
}
if(Helper::isExprElement(type1)&&Helper::isExprElement(type2)){//4. Sibling ("x"/"*"/"+"/"-","x"/"*"/"+"/"-"). operator data: -1
DATA_LIST * assignList = pkb->getAllAssigns();
for(DATA_LIST::iterator itr= assignList->begin();itr!=assignList->end();itr++){
//get assignNode
AST * assignNode = NULL;
AST_LIST * nodeList = pkb->getASTBy(*itr);
for(AST_LIST::iterator astItr=nodeList->begin();astItr!=nodeList->end();astItr++){
if((**astItr).getRootType()==ASSIGNMENT){
assignNode = *astItr;
break;
}
}
//check if modified var and expression tree root are sibling
AST * leftVarNode = assignNode->getFirstDescendant();
AST * expressionRoot = leftVarNode->getRightSibling();
int exprData = expressionRoot->getRootData();
if(expressionRoot->getRootType()==CONSTANT) exprData = pkb->getConstantValue(exprData);
//pattern is symmetric, find if x is matched with some elements in x=y+z+...
bool isPattern1 = (leftVarNode->getRootType()==type1&&leftVarNode->getRootData()==index1)&&(expressionRoot->getRootType()==type2&&exprData==index2);
bool isPattern2 =(expressionRoot->getRootType()==type1&&exprData==index1)&&(leftVarNode->getRootType()==type2&&leftVarNode->getRootData()==index2);
if(isPattern1||isPattern2)
return true; // x = expression, x is matched with var or +,-,*
//traverse the tree and check matched pattern, search in left hand side x= y+z...
stack<AST *> exprStack;
if(expressionRoot->getRootType()!=VARIABLE&&expressionRoot->getRootType()!=CONSTANT){
exprStack.push(expressionRoot);
AST * left;
AST * right;
while(!exprStack.empty()){
AST * curNode = exprStack.top();
exprStack.pop();
left = curNode->getFirstDescendant();
right = left->getRightSibling();
int leftData = left->getRootData();
if(left->getRootType()==CONSTANT) leftData = pkb->getConstantValue(leftData);
int rightData = right->getRootData();
if(right->getRootType()==CONSTANT) rightData = pkb->getConstantValue(rightData);
//pattern is symmetric
isPattern1 = (left->getRootType()==type1&&leftData==index1)&&(right->getRootType()==type2&&rightData==index2);
isPattern2 =(right->getRootType()==type1&&rightData==index1)&&(left->getRootType()==type2&&leftData==index2);
if(isPattern1||isPattern2)
return true;
//push the next operator into stack
if(left->getRootType()!=VARIABLE&&left->getRootType()!=CONSTANT) exprStack.push(left);
if(right->getRootType()!=VARIABLE&&right->getRootType()!=CONSTANT) exprStack.push(right);
}
}
}
delete assignList;
}
return returnValue;
}
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::testOperators(){
Parser p;
std::string src;
src = "procedure main{\n"
"a = y;}\n";
p.setSource(src);
p.startParse();
PKB *pkb = p.getPKB();
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()) == VARIABLE);
src = "procedure main{\n"
"a = 1;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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()) == CONSTANT);
src = "procedure main{\n"
"a = y*1;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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()) == MULTIPLY);
src = "procedure main{\n"
"a = y+1;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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);
src = "procedure main{\n"
"a = y-1;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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()) == MINUS);
src = "procedure main{\n"
"a = (y+1);}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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);
src = "procedure main{\n"
"a = (y-1);}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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()) == MINUS);
src = "procedure main{\n"
"a = (y*1);}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
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()) == MULTIPLY);
src = "procedure main{\n"
"a = b + c * d;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
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);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getVarName(pkb->getData(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()))=="b");
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getVarName(pkb->getData(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()))=="c");
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getVarName(pkb->getData(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()))=="d");
src = "procedure main{\n"
"a = b - c * d;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
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()) == MINUS);
src = "procedure main{\n"
"a = b * c + d;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
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);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == VARIABLE);
src = "procedure main{\n"
"a = b * c - d;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
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()) == MINUS);
src = "procedure main{\n"
"a = (b + c) * d;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
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()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == PLUS);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == VARIABLE);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == VARIABLE);
src = "procedure main{\n"
"a = (c + d) * (d * e) ;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == PLUS);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == MULTIPLY);
src = "procedure main{\n"
"a = b + c + d;}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == PLUS);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == PLUS);
src = "procedure main{\n"
"a = (b * c - d) * a + (t + e * f * e * g);}\n";
p.setSource(src);
p.startParse();
pkb = p.getPKB();
ast = pkb->getRootAST();
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()) == PLUS);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()) == MINUS);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == PLUS);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()) == MULTIPLY);
CPPUNIT_ASSERT(pkb->getType(ast->getFirstDescendant()->getFirstDescendant()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getRightSibling()->getFirstDescendant()->getFirstDescendant()) == MULTIPLY);
}
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);
}
