StoryLoaderBehaviour.java

package org.jbehave.core.io;

import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.Matchers.equalTo;
import static org.hamcrest.Matchers.is;
import static org.jbehave.core.io.LoadFromRelativeFile.intellijProjectStoryFilePath;
import static org.jbehave.core.io.LoadFromRelativeFile.intellijProjectTestStoryFilePath;
import static org.jbehave.core.io.LoadFromRelativeFile.mavenModuleStoryFilePath;
import static org.jbehave.core.io.LoadFromRelativeFile.mavenModuleTestStoryFilePath;

import java.io.IOException;
import java.io.InputStream;
import java.net.MalformedURLException;
import java.net.URISyntaxException;

import org.hamcrest.Matchers;
import org.jbehave.core.io.LoadFromRelativeFile.StoryFilePath;
import org.jbehave.core.io.stories.MyPendingStory;
import org.junit.Test;

public class StoryLoaderBehaviour {

    @Test
    public void shouldLoadStoryFromClasspath() {
        // Given
        String storyPath = "org/jbehave/core/io/stories/my_pending_story";
        String storyAsText = "Given my step";

        // When
        StoryLoader loader = new LoadFromClasspath();
        String loadedStoryAsText = loader.loadStoryAsText(storyPath);
        
        // Then
		assertThat(loadedStoryAsText, equalTo(storyAsText));

    }

    @Test(expected = StoryResourceNotFound.class)
    public void shouldNotLoadStoryFromClasspathIfNotFound() {

        StoryLoader loader = new LoadFromClasspath(StoryLoaderBehaviour.class);
        loader.loadStoryAsText("inexistent.story");

    }

    @Test(expected = InvalidStoryResource.class)
    public void shouldNotLoadStoryFropmClasspathIfClassloaderNotValid() {

        StoryLoader loader = new LoadFromClasspath(new InvalidClassLoader());
        assertThat(loader.toString(), Matchers.containsString(InvalidClassLoader.class.getName()));
        loader.loadStoryAsText("inexistent.story");

    }

    static class InvalidClassLoader extends ClassLoader {

        @Override
        public InputStream getResourceAsStream(String name) {
            return new InputStream() {

                public int available() throws IOException {
                    return 1;
                }

                @Override
                public int read() throws IOException {
                    throw new IOException("invalid");
                }

            };
        }
    }


    @Test
    public void shouldLoadStoryFromURL() {
        String storyPath = CodeLocations.codeLocationFromClass(this.getClass()) + "org/jbehave/core/io/stories/my_pending_story";
        String storyAsText = "Given my step";
 
        // When
        StoryLoader loader = new LoadFromURL();
        String loadedStoryAsText = loader.loadStoryAsText(storyPath);
        
        // Then
		assertThat(loadedStoryAsText, equalTo(storyAsText));
    }

    @Test(expected = InvalidStoryResource.class)
    public void shouldNotLoadStoryFromURLIfNotFound() {
        String storyPath = CodeLocations.codeLocationFromClass(this.getClass()) + "inexistent_story";

        // When
        StoryLoader loader = new LoadFromURL();
        loader.loadStoryAsText(storyPath);
        
        // Then fail as expected

    }

    @Test
    public void shouldLoadStoryFromRelativeFilePaths() {
        // Given
        String storyPath = "org/jbehave/core/io/stories/MyPendingStory.txt";
        String storyAsText = "Given my step";

        // When
        StoryLoader loader = new LoadFromRelativeFile(CodeLocations.codeLocationFromClass(MyPendingStory.class),
                mavenModuleStoryFilePath("src/main/java"),
                mavenModuleTestStoryFilePath("src/test/java"),
                intellijProjectStoryFilePath("src/main/java"),
                intellijProjectTestStoryFilePath("src/test/java"));
        
        // Then
        assertThat(loader.loadStoryAsText(storyPath), equalTo(storyAsText));

    }

    @Test
    public void shouldLoadStoryFromDefaultRelativeFilePaths() {
        // Given
        String storyPath = "org/jbehave/core/io/stories/MyPendingStory.txt";
        String storyAsText = "Given my step";

        // When
        StoryLoader loader = new LoadFromRelativeFile(CodeLocations.codeLocationFromClass(MyPendingStory.class));

        // Then
        assertThat(loader.loadStoryAsText(storyPath), equalTo(storyAsText));

    }
    
    @Test(expected=StoryResourceNotFound.class)
    public void shouldNotLoadStoryFromRelativeFileWhenNoPathsAreProvided() {
        // Given
        String storyPath = "org/jbehave/core/io/stories/MyPendingStory.txt";

        // When
        StoryLoader loader = new LoadFromRelativeFile(CodeLocations.codeLocationFromClass(MyPendingStory.class), new StoryFilePath[]{});

        // Then fail as expected
        loader.loadStoryAsText(storyPath);

    }
    
    @Test(expected=StoryResourceNotFound.class)
    public void shouldNotLoadStoryFromRelativeFileWhenNotFound() {
        // Given
        String storyPath = "org/jbehave/core/io/stories/MyInexistentStory";

        // When
        StoryLoader loader = new LoadFromRelativeFile(CodeLocations.codeLocationFromClass(MyPendingStory.class));

        // Then fail as expected
        loader.loadStoryAsText(storyPath);

    }

    @Test(expected=InvalidStoryResource.class)
    public void shouldNotLoadStoryFromRelativeFileWhenPathInvalid() {
        // Given
        String storyPath = null;

        // When
        LoadFromRelativeFile loader = new LoadFromRelativeFile(CodeLocations.codeLocationFromClass(MyPendingStory.class));

        // Then fail as expected
        loader.loadContent(storyPath);

    }

    @Test
    public void shouldLoadStoryFromRelativeFilePathsWithSpace() throws MalformedURLException, URISyntaxException {
        shouldWorkForPath("/org/jbehave/core/io/stories/foldername has spaces");

    }

    @Test
    public void shouldLoadStoryFromRelativeFilePathsWithBizarreName() throws MalformedURLException, URISyntaxException {
        shouldWorkForPath("/org/jbehave/core/io/stories/[mage_hg]");
    }

    private void shouldWorkForPath(String path) throws URISyntaxException, MalformedURLException {
        // Given
        String storyPath = "MyPendingStory.txt";
        String storyAsText = "Given my step";
        java.net.URL url = CodeLocations.codeLocationFromClass(MyPendingStory.class);
        java.io.File folderWithSpacesInName = new java.io.File(url.toURI().getPath() + path);
        java.net.URL urlThatHasEscapedSpaces = folderWithSpacesInName.toURI().toURL();
        assertThat(folderWithSpacesInName.exists(), is(true));

        // When
        StoryLoader loader = new LoadFromRelativeFile(urlThatHasEscapedSpaces);

        // Then
        assertThat(loader.loadStoryAsText(storyPath), equalTo(storyAsText));
    }


}