From 93da6bac60ff9b86ed0a8324129b95fa7cc7525e Mon Sep 17 00:00:00 2001
From: zoltan-sule <z.sule@justimmo.at>
Date: Tue, 27 Nov 2018 11:04:35 +0100
Subject: [PATCH] Changes the type of the ownershipType from string to
 OwnershipType

---
 src/Entity/Realty/OwnershipType.php | 21 +++++++++++++++++++++
 src/Entity/Realty/Realty.php        |  6 +++---
 2 files changed, 24 insertions(+), 3 deletions(-)
 create mode 100644 src/Entity/Realty/OwnershipType.php

diff --git a/src/Entity/Realty/OwnershipType.php b/src/Entity/Realty/OwnershipType.php
new file mode 100644
index 0000000..4a3a3fa
--- /dev/null
+++ b/src/Entity/Realty/OwnershipType.php
@@ -0,0 +1,21 @@
+<?php
+
+namespace Justimmo\Api\Entity\Realty;
+
+use Justimmo\Api\Annotation as JUSTIMMO;
+use Justimmo\Api\Entity\Entity;
+use Justimmo\Api\Entity\Identifiable;
+use Justimmo\Api\Entity\Nameable;
+
+/**
+ * @JUSTIMMO\Entity(cacheKey="id")
+ */
+class OwnershipType implements Entity
+{
+    use Identifiable, Nameable;
+
+    public function __toString()
+    {
+        return (string) $this->getName();
+    }
+}
diff --git a/src/Entity/Realty/Realty.php b/src/Entity/Realty/Realty.php
index 1ff9c22..98aa670 100644
--- a/src/Entity/Realty/Realty.php
+++ b/src/Entity/Realty/Realty.php
@@ -117,8 +117,8 @@ class Realty implements Entity
     private $prices;
 
     /**
-     * @var string
-     * @JUSTIMMO\Column
+     * @var OwnershipType
+     * @JUSTIMMO\Relation(targetEntity="\Justimmo\Api\Entity\Realty\OwnershipType")
      */
     private $ownershipType;
 
@@ -448,7 +448,7 @@ public function getCategories()
     }
 
     /**
-     * @return string
+     * @return OwnershipType
      */
     public function getOwnershipType()
     {