com.splunk.shuttl.archiver.bucketlock.SimpleFileLockConstructorTest.java Source code

Introduction

Here is the source code for com.splunk.shuttl.archiver.bucketlock.SimpleFileLockConstructorTest.java

Source

// Copyright (C) 2011 Splunk Inc.
//
// Splunk Inc. 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 com.splunk.shuttl.archiver.bucketlock;

import static org.testng.AssertJUnit.*;

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.channels.FileChannel;

import org.apache.commons.io.IOUtils;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import com.splunk.shuttl.archiver.bucketlock.SimpleFileLock;
import com.splunk.shuttl.testutil.TUtilsFile;
import com.splunk.shuttl.testutil.TUtilsTestNG;

/**
 * Fixture: Created with constructor that takes {@link FileChannel} to make sure
 * that it is closed properly.
 */
@Test(groups = { "fast-unit" })
public class SimpleFileLockConstructorTest {

    SimpleFileLock simpleFileLock;
    FileChannel fileChannel;

    @BeforeMethod(groups = { "fast-unit" })
    public void setUp() {
        fileChannel = getFileChannel();
        simpleFileLock = new SimpleFileLock(fileChannel);
    }

    @AfterMethod(groups = { "fast-unit" })
    public void tearDown() {
        IOUtils.closeQuietly(fileChannel);
    }

    /**
     * @return
     */
    private FileChannel getFileChannel() {
        return getOutputStreamToTempFile().getChannel();
    }

    private FileOutputStream getOutputStreamToTempFile() {
        File file = TUtilsFile.createFile();
        try {
            return new FileOutputStream(file);
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            TUtilsTestNG.failForException("Could not open InputStream" + " for file: " + file, e);
            return null;
        }
    }

    @Test(groups = { "fast-unit" })
    public void closeLock_givenFileChannelToFailedBucketsLock_doesntCloseChannelWhenFileChannelIsNotOpen()
            throws IOException {
        simpleFileLock.closeLock();
        assertTrue(!fileChannel.isOpen());
    }

    public void closeLock_givenClosedFileChannel_doesNotThrowAnything() {
        IOUtils.closeQuietly(fileChannel);
        assertTrue(!fileChannel.isOpen());
        simpleFileLock.closeLock();
    }
}