org.pentaho.di.core.vfs.VfsCoreTest.java Source code

Introduction

Here is the source code for org.pentaho.di.core.vfs.VfsCoreTest.java

Source

/*! ******************************************************************************
 *
 * Pentaho Data Integration
 *
 * Copyright (C) 2002-2013 by Pentaho : http://www.pentaho.com
 *
 *******************************************************************************
 *
 * Licensed 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.pentaho.di.core.vfs;

import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;

import junit.framework.TestCase;

import org.apache.commons.vfs.FileObject;
import org.pentaho.di.core.Const;

public class VfsCoreTest extends TestCase {

    private static final String tmpDir = System.getProperty("java.io.tmpdir");
    private static final String content = "Just a small line of text\n";

    public void testWriteReadFile() throws Exception {
        // Write a text
        //
        FileObject tempFile = KettleVFS.createTempFile("prefix", "suffix", tmpDir);
        OutputStream outputStream = KettleVFS.getOutputStream(tempFile, false);
        OutputStreamWriter writer = new OutputStreamWriter(outputStream);
        writer.write(content);
        writer.close();
        outputStream.close();

        // Read it back...
        //
        InputStream inputStream = KettleVFS.getInputStream(tempFile);
        StringBuffer buffer = new StringBuffer();
        int c;
        while ((c = inputStream.read()) >= 0) {
            buffer.append((char) c);
        }
        inputStream.close();

        assertEquals(content, buffer.toString());

        // Now open the data as a regular file...
        //
        String url = tempFile.getName().getURI();
        String textFileContent = KettleVFS.getTextFileContent(url, Const.XML_ENCODING);

        // Now delete the file...
        //
        tempFile.delete();

        assertEquals(false, tempFile.exists());

        assertEquals(content, textFileContent);
    }
}