From 72d5d0281a9465ec9cccdf7d555ba7a8edb6a72b Mon Sep 17 00:00:00 2001
From: Marcel <mraaijmakers@gmail.com>
Date: Fri, 13 Jan 2012 14:34:34 +0100
Subject: [PATCH] use !== to check field visibility use class constants

---
 lib/Doctrine/ORM/Tools/EntityGenerator.php | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/Doctrine/ORM/Tools/EntityGenerator.php b/lib/Doctrine/ORM/Tools/EntityGenerator.php
index f81f63bb3..36ebc52ca 100644
--- a/lib/Doctrine/ORM/Tools/EntityGenerator.php
+++ b/lib/Doctrine/ORM/Tools/EntityGenerator.php
@@ -316,7 +316,7 @@ public function <methodName>()
      */
     public function setFieldVisibility($visibility)
     {
-        if ($visibility != 'private' && $visibility != 'protected') {
+        if ($visibility !== EntityGenerator::FIELD_VISIBLE_PRIVATE && $visibility !== EntityGenerator::FIELD_VISIBLE_PROTECTED) {
             throw new \InvalidArgumentException('Invalid provided visibilty (only private and protected are allowed): ' . $visibility);
         }