TFE_DeleteOp(op);
  if (TF_GetCode(status) != TF_OK) return nullptr;
  CHECK_EQ(0, num_retvals);
  TF_DeleteStatus(status);
  return var_handle;
}

TFE_Context* CreateContext(const std::string& serialized_server_def,
                           bool isolate_session_state,
                           int64_t init_timeout_in_ms) {
  TF_Status* status = TF_NewStatus();

// TODO(b/291142876) Simplify TFE_ContextSetServerDefWithTimeoutAndRetries and
// TFE_ContextUpdateServerDefWithTimeout to be simple wrappers around the same
// C++ function.
// Retries are used for CreateContext calls, which is used in
// ParameterServerStrategy initialization to be robust to worker preemption.
TF_CAPI_EXPORT extern void TFE_ContextSetServerDefWithTimeoutAndRetries(

        // Test encryption context
        NegotiateContextResponse enc = Smb2NegotiateResponse.createContext(EncryptionNegotiateContext.NEGO_CTX_ENC_TYPE);
        assertNotNull(enc);
        assertTrue(enc instanceof EncryptionNegotiateContext);

        // Test preauth context
        NegotiateContextResponse preauth = Smb2NegotiateResponse.createContext(PreauthIntegrityNegotiateContext.NEGO_CTX_PREAUTH_TYPE);
        assertNotNull(preauth);

        // Use separate contexts for different operations
        CIFSContext createContext = createFreshContext();
        String sourceFileName = "source_" + timestamp + ".txt";
        String targetFileName = "target_" + timestamp + ".txt";

        SmbFile sourceFile = new SmbFile(baseUrl + "shared/" + sourceFileName, createContext);

        // Create source file with content

            clientCreds = manager.createCredential(this.clientName, userLifetime, mechOid, GSSCredential.INITIATE_ONLY);
        } else {
            this.clientName = null;
        }

        this.gssContext = manager.createContext(this.serviceName, mechOid, clientCreds, contextLifetime);

        this.gssContext.requestAnonymity(false);
        this.gssContext.requestSequenceDet(false);
        this.gssContext.requestConf(false);

        lenient().when(gssManager.createName(anyString(), any(), any())).thenReturn(serviceName);
        lenient().when(gssManager.createName(anyString(), any())).thenReturn(serviceName);
        lenient().when(gssManager.createContext(any(), any(), any(), anyInt())).thenReturn(gssContext);

        // Create Kerb5Context with mocked dependencies
        ctx = new Kerb5Context("host.example", "cifs", null, 0, 0, null);
    }

    @AfterEach

        assertThrows(IllegalStateException.class, () -> {
            // This would normally be called during authentication
            auth.createContext(null, null, "host", null, false);
        });
    }

    /**
     * Test TTL with different constructor types
     */
    @Test