org.evosuite.runtime.mock.java.io.NativeMockedIO Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of evosuite-runtime Show documentation
There is a newer version: 1.0.6
/**
 * Copyright (C) 2010-2016 Gordon Fraser, Andrea Arcuri and EvoSuite
 * contributors
 *
 * This file is part of EvoSuite.
 *
 * EvoSuite is free software: you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3.0 of the License, or
 * (at your option) any later version.
 *
 * EvoSuite 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
 * Lesser Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with EvoSuite. If not, see .
 */
package org.evosuite.runtime.mock.java.io;

import java.io.IOException;
import java.util.concurrent.atomic.AtomicInteger;

import org.evosuite.runtime.vfs.FSObject;
import org.evosuite.runtime.vfs.VFile;
import org.evosuite.runtime.vfs.VirtualFileSystem;

/**
 * This class is used to mock native methods regarding I/O, and 
 * it also provide support functions used by different I/O mocks
 * 
 * 
 * All methods declaring IOException might throw it depending on simulation / test data 
 * 
 * @author arcuri
 *
 */
public class NativeMockedIO {

	
	public static VFile getFileForReading(String path){
		FSObject target = VirtualFileSystem.getInstance().findFSObject(path);
		if(target==null || target.isDeleted() || target.isFolder() || !target.isReadPermission()){
			return null;
		}
		return (VFile) target;
	}
	
	public static int read(String path, AtomicInteger position) throws IOException{
		VFile vf = NativeMockedIO.getFileForReading(path);
		if(vf==null){
			throw new MockIOException();
		}
		
		VirtualFileSystem.getInstance().throwSimuledIOExceptionIfNeeded(path);
		
		int b = vf.read(position.getAndIncrement());
				
		return b; 
	}

	
	public static VFile getFileForWriting(String path){
		FSObject target = VirtualFileSystem.getInstance().findFSObject(path);
		if(target==null || target.isDeleted() || target.isFolder() || !target.isWritePermission()){
			return null;
		}
		return (VFile) target;
	}	
	
	
	public static void writeBytes(String path, AtomicInteger position, byte b[], int off, int len)
			throws IOException{
		
		VFile vf = NativeMockedIO.getFileForWriting(path);
		if(vf==null){
			throw new MockIOException();
		}
				
		VirtualFileSystem.getInstance().throwSimuledIOExceptionIfNeeded(path);
				
		int written = vf.writeBytes(position.get(),b, off, len);
		if(written==0){
			throw new MockIOException("Error in writing to file");
		}
		position.addAndGet(written);
	}

	
	public static int size(String path) throws IOException{
		VFile vf = NativeMockedIO.getFileForReading(path);
		if(vf==null){
			throw new MockIOException();
		}
		
		VirtualFileSystem.getInstance().throwSimuledIOExceptionIfNeeded(path);
		
		return vf.getDataSize();
	}
	
	
	public static void setLength(String path, AtomicInteger position, long newLength) throws IOException{
		if(newLength < 0){
			throw new MockIOException("Negative position: "+newLength);
		}
		if(newLength > Integer.MAX_VALUE){
			throw new MockIOException("Virtual file system does not handle files larger than  "+Integer.MAX_VALUE+" bytes");
		}
		
		VFile vf = NativeMockedIO.getFileForWriting(path);
		if(vf==null){
			throw new MockIOException();
		}
		
		VirtualFileSystem.getInstance().throwSimuledIOExceptionIfNeeded(path);
		
		vf.setLength((int)newLength);
		
		if(position.get() > newLength){
			position.set((int)newLength);
		}		
	}
}