func TestIfElseControlFlowGraph(t *testing.T) {
filePath := "./testcode/_ifelse.go"
sourceFile, err := ioutil.ReadFile(filePath)
if err != nil {
t.Fatal(err)
}
basicBlocks, err := bblock.GetBasicBlocksFromSourceCode(filePath, sourceFile)
if err != nil {
t.Fatal(err)
}
expectedGraph := cfgraph.GetControlFlowGraph(basicBlocks)
correctGraph := graph.NewGraph()
START := bblock.NewBasicBlock(-1, bblock.START, 0)
EXIT := bblock.NewBasicBlock(-1, bblock.EXIT, 0)
BB0 := bblock.NewBasicBlock(0, bblock.FUNCTION_ENTRY, 8)
BB1 := bblock.NewBasicBlock(1, bblock.IF_CONDITION, 13)
BB2 := bblock.NewBasicBlock(2, bblock.ELSE_CONDITION, 16)
BB3 := bblock.NewBasicBlock(3, bblock.ELSE_BODY, 20)
BB4 := bblock.NewBasicBlock(4, bblock.RETURN_STMT, 23)
BB0.AddSuccessorBlock(BB1)
BB1.AddSuccessorBlock(BB2, BB3)
BB2.AddSuccessorBlock(BB4)
BB3.AddSuccessorBlock(BB4)
correctBasicBlocks := []*bblock.BasicBlock{BB0, BB1, BB2, BB3, BB4}
//Test basic-blocks.
if err := VerifyBasicBlocks(basicBlocks, correctBasicBlocks); err != nil {
t.Fatal(err)
}
correctGraph.InsertEdge(&graph.Node{Value: START}, &graph.Node{Value: BB0})
correctGraph.InsertEdge(&graph.Node{Value: BB0}, &graph.Node{Value: BB1})
correctGraph.InsertEdge(&graph.Node{Value: BB1}, &graph.Node{Value: BB2})
correctGraph.InsertEdge(&graph.Node{Value: BB1}, &graph.Node{Value: BB3})
correctGraph.InsertEdge(&graph.Node{Value: BB2}, &graph.Node{Value: BB4})
correctGraph.InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: BB4})
correctGraph.InsertEdge(&graph.Node{Value: BB4}, &graph.Node{Value: EXIT})
correctGraph.InsertEdge(&graph.Node{Value: EXIT}, &graph.Node{Value: START})
if err := VerifyControlFlowGraphs(expectedGraph[0], correctGraph); err != nil {
t.Fatal(err)
}
}
func GetCyclomaticComplexityFunctionLevel(goFilePath string, goSrcFile []byte) (functions []*FunctionComplexity, err error) {
blocks, err := bblock.GetBasicBlocksFromSourceCode(goFilePath, goSrcFile)
if err != nil {
return nil, err
}
for _, cfg := range cfgraph.GetControlFlowGraph(blocks) {
complexity := GetCyclomaticComplexity(cfg)
funcBlock := cfg.Root.Value.(*bblock.BasicBlock)
functions = append(functions, &FunctionComplexity{
Name: funcBlock.FunctionName,
SrcLine: funcBlock.FunctionDeclLine,
Complexity: complexity,
ControlFlowGraph: cfg,
BasicBlocks: blocks,
})
}
return functions, nil
}
func TestSimpleControlFlowGraph(t *testing.T) {
filePath := "./testcode/_simple.go"
sourceFile, err := ioutil.ReadFile(filePath)
if err != nil {
t.Fatal(err)
}
basicBlocks, err := bblock.GetBasicBlocksFromSourceCode(filePath, sourceFile)
if err != nil {
t.Fatal(err)
}
expectedGraph := cfgraph.GetControlFlowGraph(basicBlocks)
correctGraph := graph.NewGraph()
START := bblock.NewBasicBlock(-1, bblock.START, 0)
EXIT := bblock.NewBasicBlock(-1, bblock.EXIT, 0)
BB0 := bblock.NewBasicBlock(0, bblock.FUNCTION_ENTRY, 8)
BB1 := bblock.NewBasicBlock(1, bblock.RETURN_STMT, 11)
BB0.AddSuccessorBlock(BB1)
correctBasicBlocks := []*bblock.BasicBlock{BB0, BB1}
//Test basic-blocks.
if err := VerifyBasicBlocks(basicBlocks, correctBasicBlocks); err != nil {
t.Fatal(err)
}
correctGraph.InsertEdge(&graph.Node{Value: START}, &graph.Node{Value: BB0})
correctGraph.InsertEdge(&graph.Node{Value: BB0}, &graph.Node{Value: BB1})
correctGraph.InsertEdge(&graph.Node{Value: BB1}, &graph.Node{Value: EXIT})
correctGraph.InsertEdge(&graph.Node{Value: EXIT}, &graph.Node{Value: START})
if err := VerifyControlFlowGraphs(expectedGraph[0], correctGraph); err != nil {
t.Fatal(err)
}
}
func TestGreatestCommonDivisorControlFlowGraph(t *testing.T) {
filePath := "./testcode/_gcd.go"
sourceFile, err := ioutil.ReadFile(filePath)
if err != nil {
t.Fatal(err)
}
basicBlocks, err := bblock.GetBasicBlocksFromSourceCode(filePath, sourceFile)
if err != nil {
t.Fatal(err)
}
expectedGraphs := cfgraph.GetControlFlowGraph(basicBlocks)
correctGraph := []*graph.Graph{
graph.NewGraph(), // func 'gcd'
graph.NewGraph(), // func 'main'
}
// Function 'gcd'
START0 := bblock.NewBasicBlock(-1, bblock.START, 0)
EXIT0 := bblock.NewBasicBlock(-1, bblock.EXIT, 0)
BB0 := bblock.NewBasicBlock(0, bblock.FUNCTION_ENTRY, 8)
BB1 := bblock.NewBasicBlock(1, bblock.FOR_STATEMENT, 10)
BB2 := bblock.NewBasicBlock(2, bblock.FOR_BODY, 13)
BB3 := bblock.NewBasicBlock(3, bblock.RETURN_STMT, 14)
// Function 'main'
START1 := bblock.NewBasicBlock(-1, bblock.START, 0)
EXIT1 := bblock.NewBasicBlock(-1, bblock.EXIT, 0)
BB4 := bblock.NewBasicBlock(4, bblock.FUNCTION_ENTRY, 17)
BB5 := bblock.NewBasicBlock(5, bblock.RETURN_STMT, 21)
BB0.AddSuccessorBlock(BB1)
BB1.AddSuccessorBlock(BB2, BB3)
BB2.AddSuccessorBlock(BB1)
BB4.AddSuccessorBlock(BB5)
correctBasicBlocks := []*bblock.BasicBlock{BB0, BB1, BB2, BB3, BB4, BB5}
// Function 'gcd'
correctGraph[0].InsertEdge(&graph.Node{Value: START0}, &graph.Node{Value: BB0})
correctGraph[0].InsertEdge(&graph.Node{Value: BB0}, &graph.Node{Value: BB1})
correctGraph[0].InsertEdge(&graph.Node{Value: BB1}, &graph.Node{Value: BB2})
correctGraph[0].InsertEdge(&graph.Node{Value: BB2}, &graph.Node{Value: BB1})
correctGraph[0].InsertEdge(&graph.Node{Value: BB1}, &graph.Node{Value: BB3})
correctGraph[0].InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: EXIT0})
correctGraph[0].InsertEdge(&graph.Node{Value: EXIT0}, &graph.Node{Value: START0})
// Function 'main'
correctGraph[1].InsertEdge(&graph.Node{Value: START1}, &graph.Node{Value: BB4})
correctGraph[1].InsertEdge(&graph.Node{Value: BB4}, &graph.Node{Value: BB5})
correctGraph[1].InsertEdge(&graph.Node{Value: BB5}, &graph.Node{Value: EXIT1})
correctGraph[1].InsertEdge(&graph.Node{Value: EXIT1}, &graph.Node{Value: START1})
// Test basic-blocks.
if err := VerifyBasicBlocks(basicBlocks, correctBasicBlocks); err != nil {
t.Fatal(err)
}
// Test control-flow graph.
if err := VerifyControlFlowGraphs(expectedGraphs[0], correctGraph[0]); err != nil {
t.Fatal(err)
}
if err := VerifyControlFlowGraphs(expectedGraphs[1], correctGraph[1]); err != nil {
t.Fatal(err)
}
}
func TestSwitchControlFlowGraph(t *testing.T) {
filePath := "./testcode/_switcher.go"
sourceFile, err := ioutil.ReadFile(filePath)
if err != nil {
t.Fatal(err)
}
basicBlocks, err := bblock.GetBasicBlocksFromSourceCode(filePath, sourceFile)
if err != nil {
t.Fatal(err)
}
expectedGraphs := cfgraph.GetControlFlowGraph(basicBlocks)
correctGraph := []*graph.Graph{
graph.NewGraph(), // func 'main'
graph.NewGraph(), // func 'integerToString'
}
START := bblock.NewBasicBlock(-1, bblock.START, 0)
EXIT := bblock.NewBasicBlock(-1, bblock.EXIT, 0)
// Function 'main'
BB0 := bblock.NewBasicBlock(0, bblock.FUNCTION_ENTRY, 8)
BB1 := bblock.NewBasicBlock(1, bblock.RETURN_STMT, 11)
// Function 'integerToString'
BB2 := bblock.NewBasicBlock(2, bblock.FUNCTION_ENTRY, 13)
BB3 := bblock.NewBasicBlock(3, bblock.SWITCH_STATEMENT, 14)
BB4 := bblock.NewBasicBlock(4, bblock.RETURN_STMT, 16)
BB5 := bblock.NewBasicBlock(5, bblock.RETURN_STMT, 18)
BB6 := bblock.NewBasicBlock(6, bblock.RETURN_STMT, 20)
BB7 := bblock.NewBasicBlock(7, bblock.RETURN_STMT, 22)
BB8 := bblock.NewBasicBlock(8, bblock.RETURN_STMT, 24)
BB0.AddSuccessorBlock(BB1)
BB2.AddSuccessorBlock(BB3)
BB3.AddSuccessorBlock(BB4, BB5, BB6, BB7, BB8)
correctBasicBlocks := []*bblock.BasicBlock{BB0, BB1, BB2, BB3, BB4, BB5, BB6, BB7, BB8}
//Test basic-blocks.
if err := VerifyBasicBlocks(basicBlocks, correctBasicBlocks); err != nil {
t.Fatal(err)
}
// Control flow graph for function 'main'.
correctGraph[0].InsertEdge(&graph.Node{Value: START}, &graph.Node{Value: BB0})
correctGraph[0].InsertEdge(&graph.Node{Value: BB0}, &graph.Node{Value: BB1})
correctGraph[0].InsertEdge(&graph.Node{Value: BB1}, &graph.Node{Value: EXIT})
correctGraph[0].InsertEdge(&graph.Node{Value: EXIT}, &graph.Node{Value: START})
// Control flow graph for function 'integerToString'.
correctGraph[1].InsertEdge(&graph.Node{Value: START}, &graph.Node{Value: BB2})
correctGraph[1].InsertEdge(&graph.Node{Value: BB2}, &graph.Node{Value: BB3})
correctGraph[1].InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: BB4})
correctGraph[1].InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: BB5})
correctGraph[1].InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: BB6})
correctGraph[1].InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: BB7})
correctGraph[1].InsertEdge(&graph.Node{Value: BB3}, &graph.Node{Value: BB8})
correctGraph[1].InsertEdge(&graph.Node{Value: BB4}, &graph.Node{Value: EXIT})
correctGraph[1].InsertEdge(&graph.Node{Value: BB5}, &graph.Node{Value: EXIT})
correctGraph[1].InsertEdge(&graph.Node{Value: BB6}, &graph.Node{Value: EXIT})
correctGraph[1].InsertEdge(&graph.Node{Value: BB7}, &graph.Node{Value: EXIT})
correctGraph[1].InsertEdge(&graph.Node{Value: BB8}, &graph.Node{Value: EXIT})
correctGraph[1].InsertEdge(&graph.Node{Value: EXIT}, &graph.Node{Value: START})
if err := VerifyControlFlowGraphs(expectedGraphs[0], correctGraph[0]); err != nil {
t.Fatal(err)
}
if err := VerifyControlFlowGraphs(expectedGraphs[1], correctGraph[1]); err != nil {
t.Fatal(err)
}
}