// Security configuration
    private final SecureRandom secureRandom;
    private final int selectedHashAlgorithm;
    private final boolean enforceIntegrity;

    /**
     * Context for maintaining preauth integrity state per session.
     */
    public static class PreauthIntegrityContext {
        private final byte[] salt;
        private final int hashAlgorithm;
        private volatile byte[] currentHash;

     */
    @Test
    void testInit() throws ServletException {
        assertDoesNotThrow(() -> ntlmServlet.init(servletConfig));
        // Further assertions can be added here to check the internal state of the servlet if fields were protected/public
    }

    /**
     * Test the init method when CIFSException occurs.
     * Verifies that configuration properties are properly validated during initialization.
     */

        target.ensureOpen();
        target.close();
        verify(fileHandle, times(1)).close();
        verify(fileHandle, never()).release();
        verify(tree, atLeastOnce()).release();

        // Re-prepare state for second branch: not open but handle set
        when(fileHandle.isValid()).thenReturn(false);
        // Need to reopen logic to set handle again
        // Reset open flag by reconstructing target

						}()
					}
				}
				wg.Wait()
				sm.Unlock()
			}
			sTimer.Reset(siteResyncSaveInterval)
		case <-ctx.Done():
			return
		}
	}
}

// update overall site resync state
func (sm *siteResyncMetrics) updateState(s SiteResyncStatus) error {
	if !globalSiteReplicationSys.isEnabled() {
		return nil
	}
	sm.Lock()
	defer sm.Unlock()
	switch s.Status {
	case ResyncStarted:

     * Test that pre-auth hash is reset on error
     */
    @Test
    @DisplayName("Pre-auth integrity hash should reset on error")
    void testPreauthHashResetOnError() throws Exception {
        // Setup invalid SMB2 state to trigger error
        setPrivateField(transport, "smb2", false);

        // Try to update preauth hash - should fail and reset

        assertEquals("server.example.com", testImplementation.getServer());
    }

    @Test
    @DisplayName("Test resolve in DFS flag toggles")
    void testResolveInDfsToggle() {
        // Initial state should be false
        assertFalse(testImplementation.isResolveInDfs());

        // Toggle to true
        testImplementation.setResolveInDfs(true);
        assertTrue(testImplementation.isResolveInDfs());

}

func resetGlobalEndpoints() {
	globalEndpoints = EndpointServerPools{}
}

func resetGlobalIsErasure() {
	globalIsErasure = false
}

// reset global heal state
func resetGlobalHealState() {
	// Init global heal state
	if globalAllHealState == nil {
		globalAllHealState = newHealState(GlobalContext, false)
	} else {
		globalAllHealState.Lock()
		for _, v := range globalAllHealState.healSeqMap {

        NtTransNotifyChange notify3 = new NtTransNotifyChange(mockConfig, 0x3333, FILE_NOTIFY_CHANGE_ATTRIBUTES, false);

        // Verify each instance maintains its own state
        byte[] dst1 = new byte[8];
        byte[] dst2 = new byte[8];
        byte[] dst3 = new byte[8];

        notify1.writeSetupWireFormat(dst1, 0);
        notify2.writeSetupWireFormat(dst2, 0);

            result.setRoles(toStringArray(source.get("roles")));
            result.setRoomNumber(DfTypeUtil.toString(source.get("roomNumber")));
            result.setState(DfTypeUtil.toString(source.get("state")));
            result.setStreet(DfTypeUtil.toString(source.get("street")));
            result.setSurname(DfTypeUtil.toString(source.get("surname")));

Use multiple Pydantic models and inherit freely for each case.