NewFileUploaderTest.java

/*
 * Copyright 2016 Red Hat, Inc. and/or its affiliates.
 *
 * 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.kie.workbench.common.screens.defaulteditor.client.editor;

import javax.enterprise.event.Event;

import com.google.gwtmockito.GwtMock;
import com.google.gwtmockito.GwtMockitoTestRunner;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.kie.workbench.common.services.shared.project.KieProjectService;
import org.kie.workbench.common.widgets.client.handlers.NewResourcePresenter;
import org.kie.workbench.common.widgets.client.handlers.NewResourceSuccessEvent;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.uberfire.backend.vfs.Path;
import org.uberfire.client.mvp.PlaceManager;
import org.uberfire.ext.widgets.common.client.common.BusyIndicatorView;
import org.uberfire.ext.widgets.core.client.editors.defaulteditor.DefaultEditorNewFileUpload;
import org.uberfire.mocks.CallerMock;
import org.uberfire.mocks.EventSourceMock;
import org.uberfire.mvp.Command;
import org.uberfire.workbench.events.NotificationEvent;
import org.uberfire.workbench.type.AnyResourceTypeDefinition;

import static org.junit.Assert.*;
import static org.mockito.Mockito.*;

@RunWith(GwtMockitoTestRunner.class)
public class NewFileUploaderTest {

    @Mock
    private PlaceManager placeManager;

    @GwtMock
    private DefaultEditorNewFileUpload options;

    private AnyResourceTypeDefinition resourceType = new AnyResourceTypeDefinition();

    @Mock
    private BusyIndicatorView busyIndicatorView;

    @Mock
    private org.guvnor.common.services.project.model.Package pkg;

    @Mock
    private Path pkgResourcesPath;

    @Mock
    private NewResourcePresenter presenter;

    private Event<NewResourceSuccessEvent> newResourceSuccessEventMock = spy( new EventSourceMock<NewResourceSuccessEvent>() {
        @Override
        public void fire( final NewResourceSuccessEvent event ) {
            //Do nothing. Default implementation throws an Exception
        }
    } );

    private Event<NotificationEvent> mockNotificationEvent = new EventSourceMock<NotificationEvent>() {
        @Override
        public void fire( final NotificationEvent event ) {
            //Do nothing. Default implementation throws an Exception
        }
    };

    @Mock
    private KieProjectService projectService;

    private NewFileUploader uploader;

    @Before
    public void setup() {
        uploader = new NewFileUploader( placeManager,
                                        options,
                                        resourceType,
                                        busyIndicatorView,
                                        new CallerMock<>( projectService ) ) {
            {
                super.notificationEvent = mockNotificationEvent;
                super.newResourceSuccessEvent = newResourceSuccessEventMock;
            }

            @Override
            String encode( final String uri ) {
                //Tests don't concern themselves with URI encoding
                return uri;
            }
        };
        when( projectService.resolveDefaultPath( pkg, "txt" ) ).thenReturn( pkgResourcesPath );
        when( pkgResourcesPath.toURI() ).thenReturn( "default://p0/src/main/resources" );
        when( options.getFormFileName() ).thenReturn( "file.txt" );
    }

    @Test
    public void testCreateFileNameWithExtension() {
        uploader.create( pkg,
                         "file.txt",
                         presenter );

        verify( projectService,
                times( 1 ) ).resolveDefaultPath( pkg, "txt" );
        verify( busyIndicatorView,
                times( 1 ) ).showBusyIndicator( any( String.class ) );
        verify( options,
                times( 1 ) ).setFileName( "file.txt" );
        verify( options,
                times( 1 ) ).upload( any( Command.class ),
                                     any( Command.class ) );
    }

    @Test
    public void testCreateFileNameWithoutExtension() {
        uploader.create( pkg,
                         "file",
                         presenter );

        verify( projectService,
                times( 1 ) ).resolveDefaultPath( pkg, "txt" );
        verify( busyIndicatorView,
                times( 1 ) ).showBusyIndicator( any( String.class ) );
        verify( options,
                times( 1 ) ).setFileName( "file.txt" );
        verify( options,
                times( 1 ) ).upload( any( Command.class ),
                                     any( Command.class ) );
    }

    @Test
    public void testCreateSuccess() {
        final ArgumentCaptor<Command> commandArgumentCaptor = ArgumentCaptor.forClass( Command.class );
        final ArgumentCaptor<Path> pathArgumentCaptor = ArgumentCaptor.forClass( Path.class );

        uploader.create( pkg,
                         "file",
                         presenter );

        verify( projectService,
                times( 1 ) ).resolveDefaultPath( pkg, "txt" );
        verify( busyIndicatorView,
                times( 1 ) ).showBusyIndicator( any( String.class ) );
        verify( options,
                times( 1 ) ).upload( commandArgumentCaptor.capture(),
                                     any( Command.class ) );

        //Emulate a successful upload
        final Command command = commandArgumentCaptor.getValue();
        assertNotNull( command );

        command.execute();

        verify( busyIndicatorView,
                times( 1 ) ).hideBusyIndicator();
        verify( presenter,
                times( 1 ) ).complete();
        verify( newResourceSuccessEventMock,
                times( 1 ) ).fire( any( NewResourceSuccessEvent.class ) );
        verify( placeManager,
                times( 1 ) ).goTo( pathArgumentCaptor.capture() );

        //Check navigation
        final Path routedPath = pathArgumentCaptor.getValue();
        assertEquals( "default://p0/src/main/resources/file.txt",
                      routedPath.toURI() );
    }

    @Test
    public void testCreateFailure() {
        final ArgumentCaptor<Command> commandArgumentCaptor = ArgumentCaptor.forClass( Command.class );

        uploader.create( pkg,
                         "file",
                         presenter );

        verify( projectService,
                times( 1 ) ).resolveDefaultPath( pkg, "txt" );
        verify( busyIndicatorView,
                times( 1 ) ).showBusyIndicator( any( String.class ) );
        verify( options,
                times( 1 ) ).upload( any( Command.class ),
                                     commandArgumentCaptor.capture() );

        //Emulate a successful upload
        final Command command = commandArgumentCaptor.getValue();
        assertNotNull( command );

        command.execute();

        verify( busyIndicatorView,
                times( 1 ) ).hideBusyIndicator();
        verify( presenter,
                never() ).complete();
        verify( placeManager,
                never() ).goTo( any( Path.class ) );
    }

}