tensorflow::unwrap(ctx)->SetExecutorForThread(executor->executor());
}

TFE_Executor* TFE_ContextGetExecutorForThread(TFE_Context* ctx) {
  return new TFE_Executor(&tensorflow::unwrap(ctx)->Executor());
}

void TFE_HostAddressSpace(TFE_Context* ctx, TF_Buffer* buf) {
  auto address_space = tensorflow::DeviceNameUtils::AddressSpace(
      tensorflow::unwrap(ctx)->HostCPUParsedName());

void TF_DeleteOutputList(TF_OutputList* o) { delete unwrap(o); }
void TF_OutputListSetNumOutputs(TF_OutputList* o, int num_outputs,
                                TF_Status* s) {
  unwrap(o)->expected_num_outputs = num_outputs;
  unwrap(o)->outputs.clear();
  unwrap(o)->outputs.resize(num_outputs);
}
int TF_OutputListNumOutputs(TF_OutputList* o) {
  return unwrap(o)->outputs.size();
}

        when(smbTree.unwrap(CustomSmbTree.class)).thenReturn(null);

        CustomSmbTree result = smbTree.unwrap(CustomSmbTree.class);

        assertEquals(null, result, "Should return null when type cannot be unwrapped");
        verify(smbTree).unwrap(CustomSmbTree.class);
    }

    /**
     * Test for unwrap() method throwing exception.
     * Verifies that unwrap can throw exceptions when appropriate.
     */
    @Test

        // Configure the mock to throw DcerpcException when unwrap is called
        doThrow(new DcerpcException("Test unwrap exception")).when(dcerpcSecurityProvider).unwrap(mockNdrBuffer);

        // Assert that calling unwrap throws DcerpcException
        assertThrows(DcerpcException.class, () -> dcerpcSecurityProvider.unwrap(mockNdrBuffer));

        // Verify that the unwrap method was called exactly once

  tensorflow::unwrap(ctx)->StartStep();
}

void TFE_ContextEndStep(TFE_Context* ctx) {
  tensorflow::unwrap(ctx)->EndStep();
}

const TFE_OpAttrs* TFE_OpGetAttrs(const TFE_Op* op) {
  return tensorflow::wrap(tensorflow::unwrap(op)->GetOpAttrs());
}

void TFE_OpAddAttrs(TFE_Op* op, const TFE_OpAttrs* attrs) {
  tensorflow::unwrap(op)->AddAttrs(tensorflow::unwrap(attrs));
}

        jcifs.SID mockSid2 = mock(jcifs.SID.class);
        jcifs.SID mockSid3 = mock(jcifs.SID.class);
        when(mockSid1.unwrap(sid_t.class)).thenReturn(mockSidT);
        when(mockSid2.unwrap(sid_t.class)).thenReturn(mockSidT);
        when(mockSid3.unwrap(sid_t.class)).thenReturn(mockSidT);
        testSids = new jcifs.SID[] { mockSid1, mockSid2, mockSid3 };

        // Act

     */
    boolean isStale();

    /**
     * Unwraps this handle to the specified type.
     *
     * @param <T> the type to unwrap to
     * @param type the class of the type to unwrap to
     * @return unwrapped instance
     */
    <T extends SmbPipeHandle> T unwrap(Class<T> type);

                assertTrue(SmbSession.class.isAssignableFrom((Class<?>) unwrapMethod.getReturnType()),
                        "unwrap method should return SmbSession or subtype");
            }, "unwrap(Class) method should be properly defined");
        }

        @Test
        @DisplayName("Should handle unwrap with SmbSession type")

                                                          TF_Status* s) {
  return wrap(static_cast<AbstractTensorHandle*>(unwrap(t)));
}

TFE_TensorHandle* TF_AbstractTensorGetEagerTensor(TF_AbstractTensor* at,
                                                  TF_Status* s) {
  auto handle = dyn_cast<ImmediateExecutionTensorHandle>(unwrap(at));
  if (!handle) {
    string msg = absl::StrCat("Not an eager tensor handle.",

        // Act & Assert
        // unwrap to interface itself should return same instance
        CredentialsInternal unwrapped = creds.unwrap(CredentialsInternal.class);
        assertNotNull(unwrapped, "unwrap to CredentialsInternal returns instance");
        assertSame(creds, unwrapped, "unwrap returns the same object instance");