/*
 * Copyright 2011,  Unitils.org
 *
 * 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.unitils.io.annotation.handler;

import org.junit.Before;
import org.junit.Test;
import org.unitils.UnitilsJUnit4;
import org.unitils.core.UnitilsException;
import org.unitils.io.annotation.TempFile;
import org.unitils.io.temp.TempService;
import org.unitils.mock.Mock;
import org.unitils.mock.annotation.Dummy;

import java.io.File;

import static org.junit.Assert.assertSame;
import static org.junit.Assert.fail;

/**
 * @author Tim Ducheyne
 * @author Jeroen Horemans
 * @since 3.3
 */
public class TempFileAnnotationHandlerAfterTestMethodTest extends UnitilsJUnit4 {

    /* Tested object */
    private TempFileAnnotationHandler tempFileAnnotationHandler;

    private Mock<TempService> tempServiceMock;
    @Dummy
    private File testFile;


    @Before
    public void initialize() {
        tempFileAnnotationHandler = new TempFileAnnotationHandler(tempServiceMock.getMock(), true);
    }


    @Test
    public void cleanup() {
        TestClass testObject = new TestClass();
        testObject.tempFile = testFile;

        tempFileAnnotationHandler.afterTestMethod(testObject, null, null);

        tempServiceMock.assertInvoked().deleteTempFileOrDir(testFile);
    }

    @Test
    public void cleanupDisabled() {
        TestClass testObject = new TestClass();
        testObject.tempFile = testFile;

        tempFileAnnotationHandler = new TempFileAnnotationHandler(tempServiceMock.getMock(), false);
        tempFileAnnotationHandler.afterTestMethod(testObject, null, null);

        tempServiceMock.assertNotInvoked().deleteTempFileOrDir(null);
    }

    @Test
    public void noAnnotations() {
        NoAnnotationTestClass testObject = new NoAnnotationTestClass();

        tempFileAnnotationHandler.afterTestMethod(testObject, null, null);
        tempServiceMock.assertNotInvoked().deleteTempFileOrDir(null);
    }

    @Test
    public void exception() {
        NullPointerException exception = new NullPointerException();
        TestClass testObject = new TestClass();
        tempServiceMock.raises(exception).deleteTempFileOrDir(null);

        try {
            tempFileAnnotationHandler.afterTestMethod(testObject, null, null);
            fail("UnitilsException expected");

        } catch (UnitilsException e) {
            assertSame(exception, e.getCause());
        }
    }


    private static class TestClass {

        @TempFile("tempFile.tmp")
        protected File tempFile;
    }

    private static class NoAnnotationTestClass {

        protected File tempFile;
    }

}