How to handle multiple windows in Selenium?

package io.learn;

import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;

public class BaseTest {

    protected WebDriver driver;
    

    // Set up the ChromeDriver
    @BeforeMethod
    public void setup() {
        // Set the path to your chromedriver executable
        System.setProperty("webdriver.chrome.driver", "C:\\Users\\change the path of the ChromeDriver\\drivers\\chromedriver.exe");
        
        // Initialize the ChromeDriver
        driver = new ChromeDriver();
        
    }

    // Close the browser after each test
    @AfterMethod
    public void teardown() {
        if (driver != null) {
            driver.quit();
        }
    }
}

package io.learn.window;

import static org.assertj.core.api.Assertions.assertThat;

import org.openqa.selenium.WindowType;
import org.testng.annotations.Test;

import io.learn.BaseTest;

public class WindowHandlingTestsNewTab extends BaseTest {

    @Test
    void testNewTab() {
        // Navigate to an initial URL
        driver.get("https://bonigarcia.dev/selenium-webdriver-java/");
        String initialHandle = driver.getWindowHandle();

        // Open a new tab and navigate to another URL
        driver.switchTo().newWindow(WindowType.TAB);
        driver.get("https://bonigarcia.dev/selenium-webdriver-java/web-form.html");

        // Ensure there are two windows open (main window + new tab)
        assertThat(driver.getWindowHandles().size()).isEqualTo(2);

        // Switch back to the original window (main window)
        driver.switchTo().window(initialHandle);

        // Close the main window and verify only one window is left
        driver.close();
        assertThat(driver.getWindowHandles().size()).isEqualTo(1);
    }
}

package io.learn.window;


import io.learn.BaseTest;
import org.openqa.selenium.WindowType;
import org.testng.annotations.Test;

import static org.assertj.core.api.Assertions.assertThat;

public class WindowHandlingTestsNewWindow extends BaseTest {

    @Test
    void testNewWindow() {
        // Open the initial URL in the main window
        driver.get("https://bonigarcia.dev/selenium-webdriver-java/");
        String initialHandle = driver.getWindowHandle();

        // Open a new window and navigate to another URL
        driver.switchTo().newWindow(WindowType.WINDOW);
        driver.get("https://bonigarcia.dev/selenium-webdriver-java/web-form.html");

        // Ensure there are two windows open
        assertThat(driver.getWindowHandles()).hasSize(2);

        // Switch back to the original window (parent window)
        driver.switchTo().window(initialHandle);

        // Close the original window
        driver.close();
        assertThat(driver.getWindowHandles()).hasSize(1);
    }
}