verify(mockAddress, times(1)).toInetAddress();
        }

        @Test
        void shouldReturnNullWhenWrappedIsNeitherInetAddressNorAddress() throws Exception {
            // UniAddress only accepts InetAddress, NbtAddress (which implements Address), or Address
            // Testing with NbtAddress which doesn't return anything from toInetAddress

    }

    @Test
    @DisplayName("toInetAddress should return valid InetAddress")
    void testToInetAddressContract() throws UnknownHostException {
        // Given
        InetAddress expectedInetAddress = InetAddress.getByName("127.0.0.1");
        when(mockAddress.toInetAddress()).thenReturn(expectedInetAddress);

        // When
        InetAddress inetAddress = mockAddress.toInetAddress();

        // Then

     *
     * @throws UnknownHostException if the host cannot be resolved
     *
     * @see jcifs.Address#toInetAddress()
     */
    @Override
    public InetAddress toInetAddress() throws UnknownHostException {
        if (this.addr instanceof Address) {
            return ((Address) this.addr).toInetAddress();
        }
        if (this.addr instanceof InetAddress) {
            return (InetAddress) this.addr;

    /**
     * Converts this address to an InetAddress.
     *
     * @return this address as an InetAddress
     * @throws UnknownHostException if the host cannot be resolved
     */
    InetAddress toInetAddress() throws UnknownHostException;

    /**
     * Guess called name to try for session establishment. These
     * methods are used by the smb package.
     *
     * @return guessed name
     */

        // Test toInetAddress (delegates to getInetAddress)
        mockName = new Name(mockConfig, "TESTHOST", 0x20, null);
        NbtAddress nbtAddress = new NbtAddress(mockName, testAddressInt, false, NbtAddress.H_NODE);
        InetAddress expectedInetAddress = InetAddress.getByName("192.168.1.1");
        assertEquals(expectedInetAddress, nbtAddress.toInetAddress());
    }

    @Test

    void testGetNodeStatus() throws UnknownHostException {
        // Given
        InetAddress realInetAddress = InetAddress.getByName("127.0.0.1");
        when(mockNetbiosAddress.toInetAddress()).thenReturn(realInetAddress);
        when(mockNetbiosAddress.unwrap(NbtAddress.class)).thenReturn(null);

        // When/Then - This will throw UnknownHostException quickly

     */

    public InetAddress getInetAddress() throws UnknownHostException {
        return InetAddress.getByName(getHostAddress());
    }

    @Override
    public InetAddress toInetAddress() throws UnknownHostException {
        return getInetAddress();
    }

    /**
     * Returns this IP adress as a {@link java.lang.String} in the form "%d.%d.%d.%d".
     *

     */
    private InetAddress discoverWitnessService() throws IOException {
        // Try the same server first
        InetAddress serverAddress = transport.getRemoteAddress().toInetAddress();

        if (isWitnessServiceAvailable(serverAddress)) {
            return serverAddress;
        }

        // Query for cluster witness service via DNS
        try {

        final NodeStatusRequest request = new NodeStatusRequest(this.transportContext.getConfig(),
                new Name(this.transportContext.getConfig(), NbtAddress.ANY_HOSTS_NAME, 0x00, null));
        request.addr = addr.toInetAddress();

        int n = this.transportContext.getConfig().getNetbiosRetryCount();
        while (n-- > 0) {
            try {