try {
            baos.write(session.getSessionKey());
            baos.write(channel.getLocalInterface().getAddress().getAddress());
            baos.write(channel.getRemoteInterface().getAddress().getAddress());
            baos.write(ByteBuffer.allocate(8).putLong(System.currentTimeMillis()).array());
        } catch (IOException e) {

        assertEquals(ipv4, newIPv4.getAddress());
        assertTrue(newIPv4.isIPv4());
        assertFalse(newIPv4.isIPv6());
        assertEquals(WitnessNotification.WitnessIPAddress.IPV4, newIPv4.getFlags());

        // Check IPv6 address
        WitnessNotification.WitnessIPAddress newIPv6 = newAddresses.get(1);
        assertEquals(ipv6, newIPv6.getAddress());
        assertFalse(newIPv6.isIPv4());

        SmbResourceLocatorImpl l1 = locator("smb://host/share?server=srv-from-query");
        assertSame(a1, l1.getAddress());

        // Query parameter 'address' builds UniAddress from IP
        SmbResourceLocatorImpl l2 = locator("smb://host/share?address=127.0.0.1");
        Address first = l2.getAddress();
        assertTrue(first instanceof UniAddress);

        // Host with root path -> possibleNTDomainOrWorkgroup=true

            if (sessionKey != null) {
                baos.write(sessionKey);
            }
            baos.write(channel.getLocalInterface().getAddress().getAddress());
            baos.write(channel.getRemoteInterface().getAddress().getAddress());
            baos.write(ByteBuffer.allocate(8).putLong(System.currentTimeMillis()).array());
        } catch (IOException e) {

        URL returned = loc.getURL();
        if (returned != null) {
            assertEquals(url, returned.toString());
        }

        // Test that getAddress returns null for our dummy implementation
        assertNull(loc.getAddress());
    }

    @Test
    @DisplayName("Test DFS referral data")
    void testDfsReferralData() throws Exception {
        String url = "smb://server/share/path/";

    }

    /**
     * Get the address the service is listening on
     *
     * @return the service address
     * @throws IOException if address cannot be determined
     */
    public InetAddress getAddress() throws IOException {
        if (serverSocket == null) {
            throw new IOException("Service not started");
        }
        return InetAddress.getLocalHost();
    }

    /**

    }

    @Test
    void getAddressReturnsOriginalObject() {
        InetAddress dummy = mock(InetAddress.class);
        UniAddress ua = new UniAddress(dummy);
        assertSame(dummy, ua.getAddress(), "getAddress should return the wrapped address");
    }

    @Test
    void getHostNameDelegatesToWrappedInetAddress() {
        InetAddress dummy = mock(InetAddress.class);

        NbtAddress[] addrEntry = (NbtAddress[]) addrEntryField.get(nameQueryResponse);
        assertNotNull(addrEntry[0], "NbtAddress should be created when address is non-zero");
        byte[] nodeAddress = addrEntry[0].getAddress();
        assertNotNull(nodeAddress, "Node address should not be null");
        // Note: Cannot test isGroupAddress() and getNodeType() without CIFSContext in unit test

    }

    /**
     * Returns the raw IP address of this NbtAddress. The result is in network
     * byte order: the highest order byte of the address is in getAddress()[0].
     *
     * @return a four byte array
     */
    public byte[] getAddress() {
        final byte[] addr = new byte[4];
        addr[0] = (byte) (this.address >>> 24 & 0xFF);
        addr[1] = (byte) (this.address >>> 16 & 0xFF);

                try {
                    addr = getAddress();
                } catch (final UnknownHostException uhe) {
                    throw new SmbException(url.toString(), uhe);
                }
                if (addr.getAddress() instanceof NbtAddress) {
                    final int code = ((NbtAddress) addr.getAddress()).getNameType();
                    if (code == 0x1d || code == 0x1b) {