Hooks and Tags — Setup, Teardown, and Organization

package com.autopractice.hooks;

import io.cucumber.java.After;
import io.cucumber.java.Before;
import io.cucumber.java.Scenario;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;

public class Hooks {
    private static ThreadLocal<WebDriver> driverThread =
        new ThreadLocal<>();

    @Before
    public void setUp(Scenario scenario) {
        System.out.println("Starting: " + scenario.getName());
        WebDriver driver = new ChromeDriver();
        driver.manage().window().maximize();
        driverThread.set(driver);
    }

    @After
    public void tearDown(Scenario scenario) {
        WebDriver driver = driverThread.get();

        if (scenario.isFailed() && driver != null) {
            byte[] screenshot = ((TakesScreenshot) driver)
                .getScreenshotAs(OutputType.BYTES);
            scenario.attach(screenshot, "image/png",
                "Failure-" + scenario.getName());
        }

        if (driver != null) {
            driver.quit();
            driverThread.remove();
        }
    }

    public static WebDriver getDriver() {
        return driverThread.get();
    }
}

// Only runs before @banking scenarios
@Before("@banking")
public void setUpBanking() {
    getDriver().get("https://www.testerrank.com/banking");
}

// Only runs before @shopping scenarios
@Before("@shopping")
public void setUpShopping() {
    getDriver().get("https://www.testerrank.com/shopping");
}

// Control hook execution order (lower runs first)
@Before(order = 0)
public void globalSetUp() {
    // Runs first — browser setup
}

@Before(order = 1)
public void specificSetUp() {
    // Runs second — navigation
}

@banking @regression
Feature: Banking Dashboard

  @smoke @critical
  Scenario: Dashboard loads after login
    Given the user has logged in
    Then the dashboard should show the balance

  @regression
  Scenario: Transaction history displays
    Given the user has logged in
    Then recent transactions should be visible

  @wip
  Scenario: Download statement
    # Work in progress — not ready for CI
    Given the user has logged in
    Then the user can download a PDF statement

# Run only @smoke tests
mvn test -Dcucumber.filter.tags="@smoke"

# Run @smoke OR @regression
mvn test -Dcucumber.filter.tags="@smoke or @regression"

# Run @smoke AND @banking
mvn test -Dcucumber.filter.tags="@smoke and @banking"

# Run everything EXCEPT @wip
mvn test -Dcucumber.filter.tags="not @wip"

# Complex: (smoke or regression) but not wip
mvn test -Dcucumber.filter.tags="(@smoke or @regression) and not @wip"