Selenium Java框架对接消息队列消费消息方案咨询

Let's break down why your integration isn't working and get it sorted out. The core issue here is that RabbitMQ's basicConsume is asynchronous, but your Selenium test lifecycle runs synchronously. When your @BeforeTest method finishes executing, the test starts immediately—often before the message is actually received and processed. Here's how to fix this step by step:

1. Synchronize Asynchronous Message Consumption with CountDownLatch #

We'll use a CountDownLatch to make the @BeforeTest thread wait until the message is received and parsed. This ensures your test only starts once you have the data you need.

First, update your test setup code:

// Declare these as class-level static variables to access them in @AfterTest
private static Connection connInterface;
private static Channel mqChannel;
private static TestData testData; // Your custom class to hold parsed JSON data

@BeforeTest
public static void initializeTestBaseSetup() throws Exception, IOException, TimeoutException {
    CountDownLatch messageLatch = new CountDownLatch(1); // Wait for 1 message
    ConnectionFactory conFactory = new ConnectionFactory();
    
    // Explicitly set connection params (avoid relying on defaults for consistency)
    conFactory.setUsername("guest");
    conFactory.setPassword("guest");
    conFactory.setVirtualHost("/");
    conFactory.setHost("localhost");
    conFactory.setPort(5672);

    connInterface = conFactory.newConnection();
    mqChannel = connInterface.createChannel();
    mqChannel.queueDeclare("MyQueue", false, false, false, null);

    mqChannel.basicConsume("MyQueue", false, (consumerTag, message) -> {
        try {
            String messageBody = new String(message.getBody(), StandardCharsets.UTF_8);
            System.out.println("Message received: " + messageBody);

            // Parse JSON into your custom data class
            ObjectMapper objectMapper = new ObjectMapper();
            // Fix the trailing comma in your JSON (common issue in your sample)
            objectMapper.configure(DeserializationFeature.FAIL_ON_TRAILING_COMMA, false);
            testData = objectMapper.readValue(messageBody, TestData.class);

            // Manually acknowledge the message after successful processing
            mqChannel.basicAck(message.getEnvelope().getDeliveryTag(), false);
            messageLatch.countDown(); // Release the waiting thread
        } catch (Exception e) {
            System.err.println("Failed to process message: " + e.getMessage());
            // Reject the message and requeue it if needed
            mqChannel.basicNack(message.getEnvelope().getDeliveryTag(), false, true);
            messageLatch.countDown(); // Don't hang the test indefinitely
            throw new RuntimeException("Message processing failed", e);
        }
    }, consumerTag -> {});

    // Wait up to 30 seconds for the message—adjust timeout as needed
    if (!messageLatch.await(30, TimeUnit.SECONDS)) {
        throw new TimeoutException("Timed out waiting for message from MyQueue");
    }
}

// Clean up MQ resources after tests finish
@AfterTest
public static void cleanUpMQResources() throws IOException, TimeoutException {
    if (mqChannel != null && mqChannel.isOpen()) {
        mqChannel.close();
    }
    if (connInterface != null && connInterface.isOpen()) {
        connInterface.close();
    }
}

2. Create a Data Class for Parsed JSON #

Define a simple POJO to map your JSON data—this makes it easy to access values in your Selenium tests:

public class TestData {
    private String NUMBER;
    private String Type;
    private String User;
    private String Email;

    // Getters and setters (or use Lombok for brevity)
    public String getNUMBER() { return NUMBER; }
    public void setNUMBER(String NUMBER) { this.NUMBER = NUMBER; }
    public String getType() { return Type; }
    public void setType(String type) { Type = type; }
    public String getUser() { return User; }
    public void setUser(String user) { User = user; }
    public String getEmail() { return Email; }
    public void setEmail(String email) { Email = email; }
}

3. Use the Parsed Data in Your Tests #

Now you can access the testData variable in your test methods to drive your Selenium actions:

@Test
public void runAutomatedTest() {
    // Example: Use the parsed data to fill a form
    driver.findElement(By.id("number-input")).sendKeys(testData.getNUMBER());
    driver.findElement(By.id("email-input")).sendKeys(testData.getEmail());
    // ... rest of your test logic
}

Key Improvements & Fixes #

• Synchronous Waiting: The CountDownLatch ensures your test doesn't start until the message is fully processed.
• Manual Message Acknowledgment: We switched from autoAck=true to manual acking—this prevents message loss if processing fails.
• JSON Parsing Fix: The FAIL_ON_TRAILING_COMMA configuration handles the extra comma in your sample JSON (which would normally cause parsing errors).
• Resource Cleanup: The @AfterTest method closes MQ connections/channels to avoid resource leaks.

内容的提问来源于stack exchange，提问作者Riidonesh