/*
 * Copyright (c) 1997-2017 Oracle and/or its affiliates. All rights reserved.
 */

package	javasoft.sqe.tests.api.javax.activation.DataHandler;

import	java.io.*;
import	java.awt.datatransfer.*;
import	javax.activation.*;
import	com.sun.javatest.*;
import	com.sun.javatest.lib.MultiTest; 
import	javasoft.sqe.tests.api.javax.activation.TestClasses.*;

/**
 * Create DataHandler instances, use them to invoke getTransferData() api, if
 * this method returns an object that represents the data to be transferred,
 * then this testcase passes, otherwise it fails. <p>
 */

public class getTransferData_Test extends MultiTest
{
private static final String	kFileName = "foo.txt";
private Object obj = null;     // Object for DataHandler(Object, String)
private String message = null;

public static void main(String argv[])
{
	getTransferData_Test lTest = new getTransferData_Test();
	Status lStatus = lTest.run(argv, new PrintWriter(System.err, true), new PrintWriter(System.out, true));
	lStatus.exit();
}

/*
*/
public Status getTransferDataTest1() throws UnsupportedFlavorException, IOException
{
    DataSource fds = new FileDataSource(kFileName);
    DataHandler dh = new DataHandler(fds);
    String mimeType = fds.getContentType();
    DataFlavor refdf = new DataFlavor(mimeType, "who cares?");
    dh.setCommandMap(new TestCommandMap(mimeType));
    
    if (!checkFlavors(dh, mimeType))
	return Status.failed("DataHandler(DataSource).getTransferDataFlavors() : " + message);
	
    Object refObj = TestDCH.getObject();
    Object obj = null;

    try {
	obj = dh.getTransferData(refdf);	// API TEST
    } catch(UnsupportedFlavorException ex) {
	return Status.failed("DataHandler(DataSource).getTransferData(df) threw UnsupportedDataException");
    } catch (Exception ex) {
	return Status.failed("DataHandler(DataSource).getTransferData(df) threw IOException");
    }

    if (!refObj.equals(obj)) //TestDCH always returns the same object!
	return Status.failed("DataHandler(DataSource).getTransferData() returned wrong object");
    
    DataFlavor wrongMimeDF = new DataFlavor("moo/poo", "extraneous human readable name");
    try {
	obj = dh.getTransferData(wrongMimeDF);		// API TEST
	return Status.failed("DataHandler(DataSource).getTransferData(wrongMimeDF) didn't throw");
    } catch(UnsupportedFlavorException ex) {
    } catch (Exception ex) {
	return Status.failed("DataHandler(DataSource).getTransferData(wrongMimeDF) threw IOException");
    }
    return Status.passed("getTransferData() test succeeded");
}
/*
*/
public Status getTransferDataTest2() throws UnsupportedFlavorException, IOException
{
    DataHandler dh = new DataHandler(new TestCommandObject(), "foo/goo");
    DataFlavor refdf = new ActivationDataFlavor(new TestCommandObject().getClass(), "foo/goo", "anything goes");
    dh.setCommandMap(new TestCommandMap("foo/goo"));
    if (!checkFlavors(dh, "foo/goo"))
	return Status.failed("DataHandler(Object,String).getTransferDataFlavors() : " + message);
    
    Object refObj = TestDCH.getObject(); 
    // this will fail due to bugtraq ID #4105102
    Object obj = null;

    try {
	obj = dh.getTransferData(refdf);	// API TEST
    } catch(UnsupportedFlavorException ex) {
	return Status.failed("DataHandler(Object,String).getTransferData(df) threw UnsupportedDataException");
    } catch (Exception ex) {
	return Status.failed("DataHandler(Object,String).getTransferData(df) threw IOException");
    }
     

// this is a bad test, it should never pass 
 //    if (!refObj.equals(obj)) //TestDCH always returns the same object!
// 	return Status.failed("DataHandler(Object,String).getTransferData() returned wrong object");

    DataFlavor wrongMimeDF = new DataFlavor("moo/poo", "extraneous human readable name");
    try {
	obj = dh.getTransferData(wrongMimeDF);		// API TEST
	return Status.failed("DataHandler(Object,String).getTransferData(wrongMimeDF) didn't throw");
    } catch(UnsupportedFlavorException ex) {
    } catch (Exception ex) {
	return Status.failed("DataHandler(Object,String).getTransferData(wrongMimeDF) threw IOException");
    }
    return Status.passed("getTransferData() test succeeded");
}
// The only way to determine if a dh is using the appropriate DataContentHandler
// is to see if  getTransferDataFlavors returns what we expect from a TestDCH
// the assumption is that some other

private boolean checkFlavors(DataHandler dh, String mimeType)
{
    DataFlavor dfs[] = dh.getTransferDataFlavors();
    if (dfs == null) {
        message = "getTransferDataFlavors() returned null";
        return false;
    }
    if (dfs.length != 1){
	System.out.println("dfs.length = " + dfs.length);
        message = "getTransferDataFlavors() has wrong number of elements";
        return false;
    }
    DataFlavor df = dfs[0];
    if (!df.isMimeTypeEqual(mimeType)){
        message = "getTransferDataFlavors() contains wrong flavor: " + df.getMimeType();
        return false;
    }
    return true;
}
}
