Skip to content

Commit

Permalink
Add test to verify that a resource is provided that returns a filename.
Browse files Browse the repository at this point in the history
This resolves BAOBAB-2730, #43.
  • Loading branch information
johanmynhardt committed Sep 7, 2017
1 parent 6613fc4 commit ba5ed7c
Show file tree
Hide file tree
Showing 2 changed files with 48 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
package com.afrozaar.wordpress.wpapi.v2.util;

import org.springframework.core.io.ByteArrayResource;

public class FilenameWrapperByteArrayResource extends ByteArrayResource {
final String fileName;

public FilenameWrapperByteArrayResource(byte[] byteArray, String fileName) {
super(byteArray);
this.fileName = fileName;
}

public FilenameWrapperByteArrayResource(byte[] byteArray, String description, String fileName) {
super(byteArray, description);
this.fileName = fileName;
}

@Override
public String getFilename() {
return fileName;
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -2,11 +2,18 @@

import static org.assertj.core.api.Assertions.assertThat;

import static org.junit.Assert.fail;

import com.afrozaar.wordpress.wpapi.v2.config.ClientConfig;
import com.afrozaar.wordpress.wpapi.v2.config.ClientFactory;
import com.afrozaar.wordpress.wpapi.v2.exception.WpApiParsedException;
import com.afrozaar.wordpress.wpapi.v2.model.Post;
import com.afrozaar.wordpress.wpapi.v2.model.builder.MediaBuilder;
import com.afrozaar.wordpress.wpapi.v2.model.builder.PostBuilder;
import com.afrozaar.wordpress.wpapi.v2.model.builder.TitleBuilder;
import com.afrozaar.wordpress.wpapi.v2.util.FilenameWrapperByteArrayResource;

import org.springframework.core.io.ByteArrayResource;

import org.junit.Test;
import org.slf4j.Logger;
Expand Down Expand Up @@ -46,4 +53,23 @@ public void PostFieldProcessingTest() {

}

@Test(expected = NullPointerException.class)
public void CreateMedia_MustFallOverWhenResourceDoesNotReturnFilename() throws WpApiParsedException {
wordpress.createMedia(MediaBuilder.aMedia().build(), new ByteArrayResource(new byte[0]));
}

@Test
public void CreateMedia_MustNotFallOverWhenResourceReturnsFileName() throws WpApiParsedException {



try {
wordpress.createMedia(MediaBuilder.aMedia().build(), new FilenameWrapperByteArrayResource(new byte[0], "myfile.png"));
} catch (RuntimeException e) {
if (e instanceof NullPointerException) {
fail("Did not expect a nullpointer exception");
}
}
}

}

0 comments on commit ba5ed7c

Please sign in to comment.