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

Added test for find with Query

Browse Source
This commit is contained in:
alex-semenyuk 2015-10-14 16:29:38 +02:00
parent 919123ba34
commit 78f441d1b9
1 changed files with 42 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
spring-data-mongodb/src/test/java/org/baeldung/gridfs/GridFSIntegrationTest.java
View File

@ -31,7 +31,7 @@ import com.mongodb.BasicDBObject;
import com.mongodb.DBObject; import com.mongodb.DBObject;
import com.mongodb.gridfs.GridFSDBFile; import com.mongodb.gridfs.GridFSDBFile;
@ContextConfiguration(classes = MongoConfig.class) @ContextConfiguration("file:src/main/resources/mongoConfig.xml")
@RunWith(SpringJUnit4ClassRunner.class) @RunWith(SpringJUnit4ClassRunner.class)
public class GridFSIntegrationTest { public class GridFSIntegrationTest {
@ -51,7 +51,7 @@ public class GridFSIntegrationTest {
@Test @Test
public void whenStoringFileWithMetadata_thenFileAndMetadataAreStored() { public void whenStoringFileWithMetadata_thenFileAndMetadataAreStored() {
DBObject metaData = new BasicDBObject(); DBObject metaData = new BasicDBObject();
metaData.put("key", "value"); metaData.put("user", "alex");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
@ -75,7 +75,7 @@ public class GridFSIntegrationTest {
@Test @Test
public void givenFileWithMetadataExist_whenFindingFileById_thenFileWithMetadataIsFound() { public void givenFileWithMetadataExist_whenFindingFileById_thenFileWithMetadataIsFound() {
DBObject metaData = new BasicDBObject(); DBObject metaData = new BasicDBObject();
metaData.put("key", "value"); metaData.put("user", "alex");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
@ -108,19 +108,21 @@ public class GridFSIntegrationTest {
assertNotNull(gridFSDBFile.getChunkSize()); assertNotNull(gridFSDBFile.getChunkSize());
assertThat(gridFSDBFile.getContentType(), is("image/png")); assertThat(gridFSDBFile.getContentType(), is("image/png"));
assertThat(gridFSDBFile.getFilename(), is("test.png")); assertThat(gridFSDBFile.getFilename(), is("test.png"));
assertThat(gridFSDBFile.getMetaData().get("key"), is("value")); assertThat(gridFSDBFile.getMetaData().get("user"), is("alex"));
} }
@Test @Test
public void givenMetadataAndFilesExist_whenFindingAllFiles_thenFilesWithMetadataAreFound() { public void givenMetadataAndFilesExist_whenFindingAllFiles_thenFilesWithMetadataAreFound() {
DBObject metaData = new BasicDBObject(); DBObject metaDataUser1 = new BasicDBObject();
metaData.put("key", "value"); metaDataUser1.put("user", "alex");
DBObject metaDataUser2 = new BasicDBObject();
metaDataUser2.put("user", "david");
InputStream inputStream = null; InputStream inputStream = null;
try { try {
inputStream = new FileInputStream("src/main/resources/test.png"); inputStream = new FileInputStream("src/main/resources/test.png");
gridFsTemplate.store(inputStream, "test.png", "image/png", metaData); gridFsTemplate.store(inputStream, "test.png", "image/png", metaDataUser1);
gridFsTemplate.store(inputStream, "test.png", "image/png", metaData); gridFsTemplate.store(inputStream, "test.png", "image/png", metaDataUser2);
} catch (FileNotFoundException ex) { } catch (FileNotFoundException ex) {
logger.error("File not found", ex); logger.error("File not found", ex);
} finally { } finally {
@ -138,11 +140,41 @@ public class GridFSIntegrationTest {
assertNotNull(gridFSDBFiles); assertNotNull(gridFSDBFiles);
assertThat(gridFSDBFiles.size(), is(2)); assertThat(gridFSDBFiles.size(), is(2));
} }
@Test
public void givenMetadataAndFilesExist_whenFindingAllFilesOnQuery_thenFilesWithMetadataAreFoundOnQuery() {
DBObject metaDataUser1 = new BasicDBObject();
metaDataUser1.put("user", "alex");
DBObject metaDataUser2 = new BasicDBObject();
metaDataUser2.put("user", "david");
InputStream inputStream = null;
try {
inputStream = new FileInputStream("src/main/resources/test.png");
gridFsTemplate.store(inputStream, "test.png", "image/png", metaDataUser1);
gridFsTemplate.store(inputStream, "test.png", "image/png", metaDataUser2);
} catch (FileNotFoundException ex) {
logger.error("File not found", ex);
} finally {
if (inputStream != null) {
try {
inputStream.close();
} catch (IOException ex) {
logger.error("Failed to close", ex);
}
}
}
List<GridFSDBFile> gridFSDBFiles = gridFsTemplate.find(new Query(Criteria.where("metadata.user").is("alex")));
assertNotNull(gridFSDBFiles);
assertThat(gridFSDBFiles.size(), is(1));
}
@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("user", "alex");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {
@ -168,7 +200,7 @@ public class GridFSIntegrationTest {
@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("user", "alex");
InputStream inputStream = null; InputStream inputStream = null;
String id = ""; String id = "";
try { try {