--- /dev/null 2019-02-07 20:54:51.336000000 +0300 +++ new/test/jdk/jfr/event/io/IOHelper.java 2019-02-08 18:33:30.248272920 +0300 @@ -0,0 +1,129 @@ +/* + * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved. + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. + * + * This code is free software; you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 2 only, as + * published by the Free Software Foundation. Oracle designates this + * particular file as subject to the "Classpath" exception as provided + * by Oracle in the LICENSE file that accompanied this code. + * + * This code is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License + * version 2 for more details (a copy is included in the LICENSE file that + * accompanied this code). + * + * You should have received a copy of the GNU General Public License version + * 2 along with this work; if not, write to the Free Software Foundation, + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. + * + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA + * or visit www.oracle.com if you need additional information or have any + * questions. + */ + +package jdk.jfr.event.io; + +import static jdk.test.lib.Asserts.assertEquals; +import static jdk.test.lib.Asserts.assertTrue; + +import java.util.List; +import java.util.stream.Collectors; + +import jdk.jfr.event.io.IOEvent.EventType; + +import jdk.jfr.consumer.RecordedEvent; +import jdk.test.lib.jfr.Events; + + +// Helper class to match actual RecordedEvents to expected events. +public class IOHelper { + + public static void verifyEqualsInOrder(List events, List expectedEvents) throws Throwable { + List actualEvents = getTestEvents(events, expectedEvents); + try { + assertEquals(actualEvents.size(), expectedEvents.size(), "Wrong number of events."); + for (int i = 0; i < actualEvents.size(); ++i) { + assertEquals(actualEvents.get(i), expectedEvents.get(i), "Wrong event at pos " + i); + } + } catch (Throwable t) { + logEvents(actualEvents, expectedEvents); + throw t; + } + } + + + public static void verifyEquals(List events, List expectedEvents) throws Throwable { + List actualEvents = getTestEvents(events, expectedEvents); + try { + assertEquals(actualEvents.size(), expectedEvents.size(), "Wrong number of events"); + assertTrue(actualEvents.containsAll(expectedEvents), "Not all expected events received"); + assertTrue(expectedEvents.containsAll(actualEvents), "Received unexpected events"); + } catch (Throwable t) { + logEvents(actualEvents, expectedEvents); + throw t; + } + } + + + private static List getTestEvents(List events, List expectedEvents) throws Throwable { + // Log all events + for (RecordedEvent event : events) { + String msg = event.getEventType().getName(); + boolean isSocket = IOEvent.EVENT_SOCKET_READ.equals(msg) || IOEvent.EVENT_SOCKET_WRITE.equals(msg); + boolean isFile = IOEvent.EVENT_FILE_FORCE.equals(msg) || IOEvent.EVENT_FILE_READ.equals(msg) || IOEvent.EVENT_FILE_WRITE.equals(msg); + boolean isFileReadOrWrite = IOEvent.EVENT_FILE_READ.equals(msg) || IOEvent.EVENT_FILE_WRITE.equals(msg); + boolean isRead = IOEvent.EVENT_FILE_READ.equals(msg) || IOEvent.EVENT_SOCKET_READ.equals(msg); + if (isFile) { + msg += " : " + Events.assertField(event, "path").getValue(); + } else if (isSocket) { + msg += " - " + Events.assertField(event, "host").getValue(); + msg += "." + Events.assertField(event, "address").getValue(); + msg += "." + Events.assertField(event, "port").getValue(); + } + if (isSocket || isFileReadOrWrite) { + String field = isRead ? "bytesRead" : "bytesWritten"; + msg += " : " + Events.assertField(event, field).getValue(); + } + System.out.println(msg); + } + + return events.stream() + .filter(event -> isTestEvent(event, expectedEvents)) + .map(event -> IOEvent.createTestEvent(event)) + .collect(Collectors.toList()); + } + + // A recording may contain extra events that are not part of the test. + // This function filters out events that not belong to the test. + public static boolean isTestEvent(RecordedEvent event, List testEvents) { + EventType eventType = IOEvent.getEventType(event); + if (eventType == EventType.UnknownEvent) { + return false; + } + + // Only care about threads in the expected threads. + final String threadName = event.getThread().getJavaName(); + if (testEvents.stream().noneMatch(te -> te.thread.equals(threadName))) { + return false; + } + + // Only care about files and sockets in expected events. + final String address = IOEvent.getEventAddress(event); + if (testEvents.stream().noneMatch(te -> te.address.equals(address))) { + return false; + } + return true; + } + + private static void logEvents(List actualEvents, List expectedEvents) { + for (int i = 0; i < actualEvents.size(); ++i) { + System.out.println("actual event[" + i + "] = " + actualEvents.get(i)); + } + for (int i = 0; i < expectedEvents.size(); ++i) { + System.out.println("expected event[" + i + "] = " + expectedEvents.get(i)); + } + } + +}