Router: fix failed ant builds in unit tests

i2p · Oct 18, 2023 · a23e09c · a23e09c
1 parent 00e3390
commit a23e09c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/router/java/test/junit/net/i2p/router/transport/SSUDemo.java b/router/java/test/junit/net/i2p/router/transport/SSUDemo.java
@@ -12,6 +12,7 @@
 import net.i2p.data.Hash;
 import net.i2p.data.router.RouterIdentity;
 import net.i2p.data.router.RouterInfo;
+import net.i2p.router.RouterContext;
 import net.i2p.data.i2np.DatabaseStoreMessage;
 import net.i2p.data.i2np.I2NPMessage;
 import net.i2p.data.i2np.I2NPMessageException;
@@ -181,7 +182,7 @@ private void loadPeers() {
     }
 
     private void peerRead(RouterInfo ri) {
-        RouterInfo old = _us.netDbSegmentor().store(ri.getIdentity().calculateHash(), ri);
+        RouterInfo old = _us.netDb().store(ri.getIdentity().calculateHash(), ri);
         if (old == null)
             newPeerRead(ri);
     }
@@ -314,7 +315,7 @@ public void runJob() {
             DatabaseStoreMessage m = (DatabaseStoreMessage)_msg;
             System.out.println("RECV: " + m);
             try {
-                _us.netDbSegmentor().store(m.getKey(), (RouterInfo) m.getEntry());
+                _us.netDb().store(m.getKey(), (RouterInfo) m.getEntry());
             } catch (IllegalArgumentException iae) {
                 iae.printStackTrace();
             }

diff --git a/router/java/test/junit/net/i2p/router/transport/udp/UDPFlooder.java b/router/java/test/junit/net/i2p/router/transport/udp/UDPFlooder.java
@@ -83,7 +83,7 @@ public void run() {
                     m.setMessageExpiration(_context.clock().now() + 10*1000);
                     m.setUniqueId(_context.random().nextLong(I2NPMessage.MAX_ID_VALUE));
                     if (true) {
-                        RouterInfo to = _context.netDbSegmentor().lookupRouterInfoLocally(peer.getRemotePeer());
+                        RouterInfo to = _context.netDb().lookupRouterInfoLocally(peer.getRemotePeer());
                         if (to == null)
                             continue;
                         OutNetMessage msg = new OutNetMessage(_context, m, m.getMessageExpiration(), 500, to);