@@ -4,7 +4,7 @@ import com.github.plume.oss.drivers.IDriver
4
4
import io .shiftleft .codepropertygraph .generated .NodeTypes .*
5
5
import io .shiftleft .codepropertygraph .generated .PropertyNames .*
6
6
import io .shiftleft .codepropertygraph .generated .nodes .*
7
- import io .shiftleft .codepropertygraph .generated .{DispatchTypes , EdgeTypes }
7
+ import io .shiftleft .codepropertygraph .generated .{Cpg , DispatchTypes , EdgeTypes }
8
8
import org .scalatest .matchers .should .Matchers
9
9
import org .scalatest .wordspec .AnyWordSpec
10
10
import org .scalatest .{BeforeAndAfter , BeforeAndAfterAll }
@@ -41,7 +41,7 @@ class PlumeDriverFixture(val driver: IDriver)
41
41
" overflowdb.BatchedUpdate.DiffGraph based changes" should {
42
42
43
43
" should reflect node additions in bulk transactions" in {
44
- val diffGraph = new DiffGraphBuilder
44
+ val diffGraph = Cpg .newDiffGraphBuilder
45
45
// Create some nodes
46
46
diffGraph.addNode(nodeToNodeCreate(m1)).addNode(nodeToNodeCreate(b1))
47
47
driver.bulkTx(diffGraph.build())
@@ -54,8 +54,8 @@ class PlumeDriverFixture(val driver: IDriver)
54
54
}
55
55
56
56
" should reflect edge additions in bulk transactions" in {
57
- val diffGraph1 = new DiffGraphBuilder
58
- val diffGraph2 = new DiffGraphBuilder
57
+ val diffGraph1 = Cpg .newDiffGraphBuilder
58
+ val diffGraph2 = Cpg .newDiffGraphBuilder
59
59
// Create some nodes
60
60
diffGraph1.addNode(nodeToNodeCreate(m1)).addNode(b1)
61
61
driver.bulkTx(diffGraph1)
0 commit comments