From 5b5daaf39764373a5b0e2c0d54796785b5a1427b Mon Sep 17 00:00:00 2001
From: Oliver Gierke <ogierke@pivotal.io>
Date: Tue, 31 May 2016 18:52:36 +0200
Subject: [PATCH] DATACMNS-831 - Polishing.
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Formatting. Fixed copyright header and author. Added missing assertions in AbstractMappingContext.addPersistentEntity(…).

Original pull request: #157.
---
 .../mapping/context/AbstractMappingContext.java     | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/src/main/java/org/springframework/data/mapping/context/AbstractMappingContext.java b/src/main/java/org/springframework/data/mapping/context/AbstractMappingContext.java
index 896aaf874..373899921 100644
--- a/src/main/java/org/springframework/data/mapping/context/AbstractMappingContext.java
+++ b/src/main/java/org/springframework/data/mapping/context/AbstractMappingContext.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2011-2015 by the original author(s).
+ * Copyright 2011-2016 by the original author(s).
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -65,6 +65,7 @@ import org.springframework.util.StringUtils;
  * @author Thomas Darimont
  * @author Tomasz Wysocki
  * @author Mark Paluch
+ * @author Mikael Klamra
  */
 public abstract class AbstractMappingContext<E extends MutablePersistentEntity<?, P>, P extends PersistentProperty<P>>
 		implements MappingContext<E, P>, ApplicationEventPublisherAware, InitializingBean {
@@ -274,7 +275,7 @@ public abstract class AbstractMappingContext<E extends MutablePersistentEntity<?
 	/**
 	 * Adds the given type to the {@link MappingContext}.
 	 *
-	 * @param type
+	 * @param type must not be {@literal null}.
 	 * @return
 	 */
 	protected E addPersistentEntity(Class<?> type) {
@@ -284,13 +285,17 @@ public abstract class AbstractMappingContext<E extends MutablePersistentEntity<?
 	/**
 	 * Adds the given {@link TypeInformation} to the {@link MappingContext}.
 	 *
-	 * @param typeInformation
+	 * @param typeInformation must not be {@literal null}.
 	 * @return
 	 */
 	protected E addPersistentEntity(TypeInformation<?> typeInformation) {
-		
+
+		Assert.notNull(typeInformation, "TypeInformation must not be null!");
+
 		try {
+
 			read.lock();
+
 			E persistentEntity = persistentEntities.get(typeInformation);
 
 			if (persistentEntity != null) {