Jelajahi Sumber

Images: fileName added to domain

Andrew Grant 2 bulan lalu
induk
melakukan
b7a89cb6cb

+ 12 - 0
src/main/java/scot/carricksoftware/grants/domains/images/Image.java

@@ -18,6 +18,10 @@ public class Image extends BaseEntity {
     @Column(name = "`name`")
     private String name;
 
+    @SuppressWarnings("JpaDataSourceORMInspection")
+    @Column(name= "`file_name`")
+    private String fileName;
+
     @SuppressWarnings("JpaDataSourceORMInspection")
     @Lob
     @Column(name = "`imageData`")
@@ -38,4 +42,12 @@ public class Image extends BaseEntity {
     public void setImageData(byte[] imageData) {
         this.imageData = imageData;
     }
+
+    public String getFileName() {
+        return fileName;
+    }
+
+    public void setFileName(String fileName) {
+        this.fileName = fileName;
+    }
 }

+ 12 - 0
src/test/java/scot/carricksoftware/grants/domains/images/ImageTest.java

@@ -57,5 +57,17 @@ class ImageTest {
         assertArrayEquals(string.getBytes(), image.getImageData());
     }
 
+    @Test
+    public void getFileNameTest() {
+       assertNull(image.getFileName());
+    }
+
+    @Test
+    public void setFileNameTest() {
+        String filename = GetRandomString();
+        image.setFileName(filename);
+        assertEquals(filename, image.getFileName());
+    }
+
 
 }