Test: Extend database marshalling testcase

CityDoctorParent/CityDoctorModel/src/test/java/de/hft/stuttgart/citydoctor2/database/DataBaseHandlerTest.java

 package de.hft.stuttgart.citydoctor2.database;
 
+import de.hft.stuttgart.citydoctor2.check.CheckableUtilsVisitor;
 import de.hft.stuttgart.citydoctor2.datastructure.Building;
 import de.hft.stuttgart.citydoctor2.datastructure.CityDoctorModel;
+import de.hft.stuttgart.citydoctor2.datastructure.CityObject;
 import de.hft.stuttgart.citydoctor2.parser.CityGmlParseException;
 import de.hft.stuttgart.citydoctor2.parser.CityGmlParser;
 import de.hft.stuttgart.citydoctor2.parser.InvalidGmlFileException;
 import de.hft.stuttgart.citydoctor2.parser.ParserConfiguration;
+import de.hft.stuttgart.citydoctor2.utils.Copy;
+import de.hft.stuttgart.citydoctor2.utils.visitors.CityObjectCollector;
 import org.h2gis.functions.factory.H2GISDBFactory;
-import org.h2gis.utilities.wrapper.ConnectionWrapper;
 import org.junit.Before;
 import org.junit.BeforeClass;
 import org.junit.Test;
 
-import java.io.File;
 import java.sql.Connection;
-import java.sql.PreparedStatement;
 import java.sql.SQLException;
+import java.util.Set;
 
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertNotNull;
@@ -45,16 +47,28 @@ public class DataBaseHandlerTest {
         try (Connection con = H2GISDBFactory.openSpatialDataBase("cd_db")) {
             assertNotNull(con);
             ParserConfiguration config = new ParserConfiguration(8, false);
-            CityDoctorModel model = CityGmlParser.parseCityGmlFile("src/test/resources/SimpleSolid_SrefBS.gml", config);
+            CityDoctorModel model = CityGmlParser.parseCityGmlFile("src/test/resources/FZK_haus.gml", config);
             Building b = model.getBuildings().get(0);
+            Copy.copy(b);
             assertNotNull(b);
             assertEquals(0, DatabaseHandler.getFeatureCount());
             DatabaseHandler.marshallCityObject(b);
             assertEquals(1, DatabaseHandler.getFeatureCount());
 
-//            Building b2 = (Building) DatabaseHandler.unmarshallCityObject(b.getGmlId());
-//            assertNotNull(b2);
-//            assertEquals(1, DatabaseHandler.getFeatureCount());
+            CityObject b2 = DatabaseHandler.unmarshallCityObject(b.getGmlId());
+            assertNotNull(b2);
+            assertEquals(1, DatabaseHandler.getFeatureCount());
+            assertEquals(b.getGmlId(), b2.getGmlId());
+            assertEquals(b.getFeatureType(), b2.getFeatureType());
+
+            CityObjectCollector collector = new CityObjectCollector();
+            b.accept(collector);
+            Set<CityObject> coB = collector.getCityObjects();
+            collector = new CityObjectCollector();
+            b2.accept(collector);
+            Set<CityObject> coB2 = collector.getCityObjects();
+            assertEquals(coB.size(), coB2.size());
+
         } catch (SQLException e) {
             throw new RuntimeException(e);
         }