X-Git-Url: https://code.wpia.club/?a=blobdiff_plain;f=src%2Forg%2Fcacert%2Fgigi%2FdbObjects%2FAssurance.java;h=33dbf1ed5e26fcfbcf593214906b3332467aff7f;hb=af932253d612fbbbf1dcead6107df6fc53896282;hp=dbcc20b6c2f290c75b842e2e35c4784c37b6eb98;hpb=9def69bd08ea69eb27786d5b34f00e154e09e9f3;p=gigi.git

diff --git a/src/org/cacert/gigi/dbObjects/Assurance.java b/src/org/cacert/gigi/dbObjects/Assurance.java
index dbcc20b6..33dbf1ed 100644
--- a/src/org/cacert/gigi/dbObjects/Assurance.java
+++ b/src/org/cacert/gigi/dbObjects/Assurance.java
@@ -1,11 +1,12 @@
 package org.cacert.gigi.dbObjects;
 
+import org.cacert.gigi.database.DBEnum;
 import org.cacert.gigi.dbObjects.wrappers.DataContainer;
 
 @DataContainer
 public class Assurance {
 
-    public enum AssuranceType {
+    public enum AssuranceType implements DBEnum {
         FACE_TO_FACE("Face to Face Meeting"), TOPUP("TOPUP"), TTP_ASSISTED("TTP-Assisted"), NUCLEUS("Nucleus Bonus");
 
         private final String description;
@@ -17,6 +18,11 @@ public class Assurance {
         public String getDescription() {
             return description;
         }
+
+        @Override
+        public String getDBName() {
+            return description;
+        }
     }
 
     private int id;
@@ -33,7 +39,9 @@ public class Assurance {
 
     private String date;
 
-    public Assurance(int id, User from, Name to, String location, String method, int points, String date) {
+    private Country country;
+
+    public Assurance(int id, User from, Name to, String location, String method, int points, String date, Country country) {
         this.id = id;
         this.from = from;
         this.to = to;
@@ -41,6 +49,7 @@ public class Assurance {
         this.method = method;
         this.points = points;
         this.date = date;
+        this.country = country;
 
     }
 
@@ -72,4 +81,7 @@ public class Assurance {
         return date;
     }
 
+    public Country getCountry() {
+        return country;
+    }
 }