iSharkFly-Docs/java-tutorials
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added assertion in GridFSTest

Browse Source
This commit is contained in:
coach88 2015-09-22 12:46:12 +03:00
parent f202857195
commit c6be2af18f
1 changed files with 16 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
spring-data-mongodb/src/test/java/org/baeldung/gridfs/GridFSTest.java
View File

@ -2,6 +2,7 @@ package org.baeldung.gridfs;
import static org.hamcrest.CoreMatchers.is; import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.Matchers.nullValue; import static org.hamcrest.Matchers.nullValue;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertThat; import static org.junit.Assert.assertThat;
@ -32,16 +33,16 @@ import com.mongodb.gridfs.GridFSDBFile;
@ContextConfiguration(classes = MongoConfig.class) @ContextConfiguration(classes = MongoConfig.class)
@RunWith(SpringJUnit4ClassRunner.class) @RunWith(SpringJUnit4ClassRunner.class)
public class GridFSTest { public class GridFSTest {
private final Logger logger = LoggerFactory.getLogger(getClass()); private final Logger logger = LoggerFactory.getLogger(getClass());
@Autowired @Autowired
private GridFsTemplate gridFsTemplate; private GridFsTemplate gridFsTemplate;
@After @After
public void tearDown() { public void tearDown() {
List<GridFSDBFile> fileList = gridFsTemplate.find(null); List<GridFSDBFile> fileList = gridFsTemplate.find(null);
for(GridFSDBFile file: fileList) { for (GridFSDBFile file : fileList) {
gridFsTemplate.delete(new Query(Criteria.where("filename").is(file.getFilename()))); gridFsTemplate.delete(new Query(Criteria.where("filename").is(file.getFilename())));
} }
} }
@ -52,7 +53,6 @@ public class GridFSTest {
metaData.put("key", "value"); metaData.put("key", "value");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
inputStream = new FileInputStream("src/main/resources/test.png"); inputStream = new FileInputStream("src/main/resources/test.png");
id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString(); id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString();
@ -77,7 +77,6 @@ public class GridFSTest {
metaData.put("key", "value"); metaData.put("key", "value");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
inputStream = new FileInputStream("src/main/resources/test.png"); inputStream = new FileInputStream("src/main/resources/test.png");
id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString(); id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString();
@ -92,9 +91,9 @@ public class GridFSTest {
} }
} }
} }
GridFSDBFile gridFSDBFile = gridFsTemplate.findOne(new Query(Criteria.where("_id").is(id))); GridFSDBFile gridFSDBFile = gridFsTemplate.findOne(new Query(Criteria.where("_id").is(id)));
assertNotNull(gridFSDBFile); assertNotNull(gridFSDBFile);
assertThat(gridFSDBFile.getFilename(), is("test.png")); assertThat(gridFSDBFile.getFilename(), is("test.png"));
assertThat(gridFSDBFile.getMetaData().get("key"), is("value")); assertThat(gridFSDBFile.getMetaData().get("key"), is("value"));
@ -121,21 +120,19 @@ public class GridFSTest {
} }
} }
} }
List<GridFSDBFile> gridFSDBFiles = gridFsTemplate.find(null); List<GridFSDBFile> gridFSDBFiles = gridFsTemplate.find(null);
assertNotNull(gridFSDBFiles); assertNotNull(gridFSDBFiles);
assertThat(gridFSDBFiles.size(), is(2)); assertThat(gridFSDBFiles.size(), is(2));
} }
@Test @Test
public void givenFileWithMetadataExist_whenDeletingFileById_thenFileWithMetadataIsDeleted() { public void givenFileWithMetadataExist_whenDeletingFileById_thenFileWithMetadataIsDeleted() {
DBObject metaData = new BasicDBObject(); DBObject metaData = new BasicDBObject();
metaData.put("key", "value"); metaData.put("key", "value");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
inputStream = new FileInputStream("src/main/resources/test.png"); inputStream = new FileInputStream("src/main/resources/test.png");
id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString(); id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString();
@ -150,19 +147,18 @@ public class GridFSTest {
} }
} }
} }
gridFsTemplate.delete(new Query(Criteria.where("_id").is(id))); gridFsTemplate.delete(new Query(Criteria.where("_id").is(id)));
assertThat(gridFsTemplate.findOne(new Query(Criteria.where("_id").is(id))), is(nullValue())); assertThat(gridFsTemplate.findOne(new Query(Criteria.where("_id").is(id))), is(nullValue()));
} }
@Test @Test
public void givenFileWithMetadataExist_whenGettingFileByResource_thenFileWithMetadataIsGotten() { public void givenFileWithMetadataExist_whenGettingFileByResource_thenFileWithMetadataIsGotten() {
DBObject metaData = new BasicDBObject(); DBObject metaData = new BasicDBObject();
metaData.put("key", "value"); metaData.put("key", "value");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
inputStream = new FileInputStream("src/main/resources/test.png"); inputStream = new FileInputStream("src/main/resources/test.png");
id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString(); id = gridFsTemplate.store(inputStream, "test.png", "image/png", metaData).getId().toString();
@ -177,10 +173,11 @@ public class GridFSTest {
} }
} }
} }
GridFsResource[] gridFsResource = gridFsTemplate.getResources("test*"); GridFsResource[] gridFsResource = gridFsTemplate.getResources("test*");
assertNotNull(gridFsResource); assertNotNull(gridFsResource);
assertEquals(gridFsResource.length, 1);
assertThat(gridFsResource[0].getFilename(), is("test.png")); assertThat(gridFsResource[0].getFilename(), is("test.png"));
} }
} }