JSONAuditLogMarchallerTest.java

/*
 * Copyright (c) 2015 EMC Corporation
 * All Rights Reserved
 */
package com.emc.storageos.api.service;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.net.URI;
import java.net.URISyntaxException;

import org.junit.Assert;

import org.codehaus.jackson.map.AnnotationIntrospector;
import org.codehaus.jackson.map.ObjectMapper;
import org.codehaus.jackson.map.exc.UnrecognizedPropertyException;
import org.codehaus.jackson.xc.JaxbAnnotationIntrospector;
import org.junit.Test;

import com.emc.storageos.api.service.impl.resource.utils.JSONAuditLogMarshaller;
import com.emc.storageos.api.service.impl.resource.utils.MarshallingExcetion;
import com.emc.storageos.db.client.model.AuditLog;

public class JSONAuditLogMarchallerTest {
    private static final String JsonTestOutputFile = "JSONSAuditLogMarshallerTest.json";

    @Test
    public void testJsonAuditLogMarshalling() throws URISyntaxException, IOException,
            MarshallingExcetion {

        deleteIfExists(JsonTestOutputFile);
        JSONAuditLogMarshaller jm = new JSONAuditLogMarshaller();
        AuditLog log = new AuditLog();
        log.setProductId("productId.1");
        log.setTenantId(new URI("http://tenant.1"));
        log.setUserId(new URI("http://user.1"));
        log.setServiceType("serviceType.1");
        log.setAuditType("auditType.1");
        log.setDescription("description.1");
        log.setOperationalStatus("operationalStatus.1");

        OutputStream output = new OutputStream() {
            private StringBuilder string = new StringBuilder();

            @Override
            public void write(int b) throws IOException {
                this.string.append((char) b);
            }

            public String toString() {
                return this.string.toString();
            }
        };

        BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
                output));
        jm.marshal(log, writer);
        writer.close();

        FileWriter fileWriter = new FileWriter(JsonTestOutputFile);
        fileWriter.write(output.toString());
        fileWriter.close();

        ObjectMapper mapper = null;
        mapper = new ObjectMapper();
        AnnotationIntrospector introspector = new JaxbAnnotationIntrospector();
        mapper.getDeserializationConfig().withAnnotationIntrospector(
                introspector);

        AuditLog auditLog = mapper.readValue(new File(JsonTestOutputFile),
                AuditLog.class);

        Assert.assertEquals("productId.1", auditLog.getProductId().toString());
        Assert.assertEquals("http://tenant.1", auditLog.getTenantId().toString());
        Assert.assertEquals("http://user.1", auditLog.getUserId().toString());
        Assert.assertEquals("serviceType.1", auditLog.getServiceType().toString());
        Assert.assertEquals("auditType.1", auditLog.getAuditType().toString());
        Assert.assertEquals("description.1", auditLog.getDescription().toString());
        Assert.assertEquals("operationalStatus.1", auditLog.getOperationalStatus().toString());

        deleteIfExists(JsonTestOutputFile);
    }

    @Test
    public void testJsonAuditLogMarshallingForNullLog() throws URISyntaxException, IOException,
            MarshallingExcetion {

        deleteIfExists(JsonTestOutputFile);
        JSONAuditLogMarshaller jm = new JSONAuditLogMarshaller();
        AuditLog log = null;

        OutputStream output = new OutputStream() {
            private StringBuilder string = new StringBuilder();

            @Override
            public void write(int b) throws IOException {
                this.string.append((char) b);
            }

            public String toString() {
                return this.string.toString();
            }
        };

        BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
                output));
        jm.header(writer);
        jm.marshal(log, writer);
        jm.tailer(writer);
        writer.close();

        FileWriter fileWriter = new FileWriter(JsonTestOutputFile);
        fileWriter.write(output.toString());
        fileWriter.close();

        ObjectMapper mapper = null;
        mapper = new ObjectMapper();
        AnnotationIntrospector introspector = new JaxbAnnotationIntrospector();
        mapper.getDeserializationConfig().withAnnotationIntrospector(
                introspector);

        try {
            @SuppressWarnings("unused")
            AuditLog auditLog = mapper.readValue(new File(JsonTestOutputFile),
                    AuditLog.class);
        } catch (UnrecognizedPropertyException e) {
            Assert.assertTrue(e.toString().contains("Unrecognized"));
        }

        deleteIfExists(JsonTestOutputFile);

    }

    @Test
    public void testJsonAuditLogMarshallingForIOExceptions() throws URISyntaxException, IOException,
            MarshallingExcetion {

        deleteIfExists(JsonTestOutputFile);
        JSONAuditLogMarshaller jm = new JSONAuditLogMarshaller();
        AuditLog log = new AuditLog();
        log.setProductId("productId.2");
        log.setTenantId(new URI("http://tenant.2"));
        log.setUserId(new URI("http://user.2"));
        log.setServiceType("serviceType.2");
        log.setAuditType("auditType.2");
        log.setDescription("description.2");
        log.setOperationalStatus("operationalStatus.2");

        OutputStream output = new OutputStream() {
            private StringBuilder string = new StringBuilder();

            @Override
            public void write(int b) throws IOException {
                this.string.append((char) b);
            }

            public String toString() {
                return this.string.toString();
            }
        };

        try {
            BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
                    output));
            writer.close();
            jm.header(writer);
        } catch (MarshallingExcetion e) {
            Assert.assertTrue(e.toString().contains("JSON head Streaming failed"));
        }

        try {
            BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
                    output));
            writer.close();
            jm.marshal(log, writer);
        } catch (MarshallingExcetion e) {
            Assert.assertTrue(e.toString().contains("JSON streaming failed"));
        }

        try {
            BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(
                    output));
            writer.close();
            jm.tailer(writer);
        } catch (MarshallingExcetion e) {
            Assert.assertTrue(e.toString().contains("JSON tail Streaming failed"));
        }
        deleteIfExists(JsonTestOutputFile);

    }

    private void deleteIfExists(String fname) {
        File f = new File(fname);

        if (f.exists()) {
            f.delete();
        }
    }
}