small refactor

Author: michalluzyna

src/main/java/com/capgemini/training/appointmentbooking/dataaccess/entity/BaseEntity.java

@@ -17,7 +17,8 @@ public class BaseEntity {
 
 	@Version
 	private int version;
-	
+
+	@Column(insertable = true, updatable = false)
 	private Date created;
 
 	@Column(name="LAST_UPDATED")

src/test/java/com/capgemini/training/appointmentbooking/dataaccess/entity/EntitySmokeTest.java

@@ -22,34 +22,18 @@ public class EntitySmokeTest {
 	void loadAllClasses() {
 
 		// given
-		Map<EntityName, Integer> classMap = Map.ofEntries( //
-				Map.entry(EntityName.USER, 8), //
-				Map.entry(EntityName.CLIENT, 4), //
-				Map.entry(EntityName.SPECIALIST, 4), //
-				Map.entry(EntityName.TREATMENT, 12), //
-				Map.entry(EntityName.APPOINTMENT, 20)
+		Map<Class<? extends BaseEntity>, Integer> classMap = Map.of( //
+				UserEntity.class, 8, //
+				ClientEntity.class, 4, //
+				SpecialistEntity.class, 4, //
+				TreatmentEntity.class, 12, //
+				AppointmentEntity.class, 20
 				);
 
 		// when //then
-		classMap.entrySet().stream() //
-				.forEach(entry -> {
-					List<?> result = em.createQuery("from " + entry.getKey().simpleName).getResultList();
-					assertThat(result).hasSize(entry.getValue());
-				});
-		
+		classMap.forEach((entityType, expectedCount) ->
+					assertThat(em.createQuery("from " + entityType.getSimpleName()).getResultList()).hasSize(expectedCount));
 	}
 
-	enum EntityName {
-
-		APPOINTMENT("AppointmentEntity"), CLIENT("ClientEntity"), SPECIALIST("SpecialistEntity"),
-		TREATMENT("TreatmentEntity"), USER("UserEntity");
-
-		private String simpleName;
-
-		private EntityName(String simpleName) {
-			this.simpleName = simpleName;
-		}
-
-	}
 
 }