| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package org.apache.nifi.minifi.commons.status; |
| |
| import org.junit.Test; |
| |
| import java.io.ByteArrayInputStream; |
| import java.io.ByteArrayOutputStream; |
| import java.io.IOException; |
| import java.io.ObjectInputStream; |
| import java.io.ObjectOutputStream; |
| import java.io.Serializable; |
| |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addConnectionStatus; |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addControllerServiceStatus; |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addExpectedRemoteProcessGroupStatus; |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addInstanceStatus; |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addProcessorStatus; |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addReportingTaskStatus; |
| import static org.apache.nifi.minifi.commons.status.util.StatusReportPopulator.addSystemDiagnosticStatus; |
| import static org.junit.Assert.assertEquals; |
| |
| public class TestStatusReport { |
| |
| @Test |
| public void verifySerializableFullyPopulated() throws IOException, ClassNotFoundException { |
| FlowStatusReport original = new FlowStatusReport(); |
| |
| addControllerServiceStatus(original, true, true, true, true); |
| addInstanceStatus(original, true, true, true, true); |
| addSystemDiagnosticStatus(original, true, true, true, true, true); |
| addReportingTaskStatus(original, true, true, true, true); |
| addConnectionStatus(original, true, true); |
| addProcessorStatus(original, true, true, true, true, true); |
| addExpectedRemoteProcessGroupStatus(original, true, true, true, true, true, true); |
| |
| byte[] byteArrayCopy = serialize(original); |
| FlowStatusReport copy = unSerialize(byteArrayCopy, FlowStatusReport.class); |
| |
| assertEquals(original, copy); |
| } |
| |
| @Test |
| public void verifySerializableSomeNull() throws IOException, ClassNotFoundException { |
| FlowStatusReport original = new FlowStatusReport(); |
| |
| addControllerServiceStatus(original, true, true, true, true); |
| addInstanceStatus(original, true, true, true, true); |
| addSystemDiagnosticStatus(original, true, true, true, true, true); |
| addProcessorStatus(original, true, true, true, true, true); |
| addExpectedRemoteProcessGroupStatus(original, true, true, true, true, true, true); |
| |
| byte[] byteArrayCopy = serialize(original); |
| FlowStatusReport copy = unSerialize(byteArrayCopy, FlowStatusReport.class); |
| |
| assertEquals(original, copy); |
| } |
| |
| private static <T extends Serializable> byte[] serialize(T obj) throws IOException { |
| ByteArrayOutputStream baos = new ByteArrayOutputStream(); |
| ObjectOutputStream oos = new ObjectOutputStream(baos); |
| oos.writeObject(obj); |
| oos.close(); |
| return baos.toByteArray(); |
| } |
| |
| private static <T extends Serializable> T unSerialize(byte[] b, Class<T> cl) throws IOException, ClassNotFoundException { |
| ByteArrayInputStream bais = new ByteArrayInputStream(b); |
| ObjectInputStream ois = new ObjectInputStream(bais); |
| Object o = ois.readObject(); |
| return cl.cast(o); |
| } |
| } |