2525import org .neo4j .gds .BaseTest ;
2626import org .neo4j .gds .GdsCypher ;
2727import org .neo4j .gds .NodeProjections ;
28- import org .neo4j .gds .ProcedureCallContextReturnColumns ;
2928import org .neo4j .gds .RelationshipProjections ;
3029import org .neo4j .gds .api .AlgorithmMetaDataSetter ;
3130import org .neo4j .gds .api .CloseableResourceRegistry ;
3231import org .neo4j .gds .api .DatabaseId ;
3332import org .neo4j .gds .api .NodeLookup ;
33+ import org .neo4j .gds .api .ProcedureReturnColumns ;
3434import org .neo4j .gds .api .TerminationMonitor ;
3535import org .neo4j .gds .catalog .GraphProjectProc ;
3636import org .neo4j .gds .compat .GraphDatabaseApiProxy ;
4747import org .neo4j .gds .test .TestMutateConfig ;
4848import org .neo4j .gds .transaction .DatabaseTransactionContext ;
4949import org .neo4j .graphdb .Transaction ;
50- import org .neo4j .internal .kernel .api .procs .ProcedureCallContext ;
5150
5251import java .util .List ;
5352import java .util .Map ;
@@ -74,13 +73,7 @@ void setup() throws Exception {
7473 .builder ()
7574 .databaseId (DatabaseId .of (db ))
7675 .dependencyResolver (GraphDatabaseApiProxy .dependencyResolver (db ))
77- .returnColumns (new ProcedureCallContextReturnColumns (new ProcedureCallContext (
78- 42 ,
79- new String [0 ],
80- false ,
81- "neo4j" ,
82- false
83- )))
76+ .returnColumns (ProcedureReturnColumns .EMPTY )
8477 .log (Neo4jProxy .testLog ())
8578 .taskRegistryFactory (EmptyTaskRegistryFactory .INSTANCE )
8679 .userLogRegistryFactory (EmptyUserLogRegistryFactory .INSTANCE )
0 commit comments