From 95dab97b1bc52f33e357a5560f3d3837de438db7 Mon Sep 17 00:00:00 2001
From: rob bygrave <robin.bygrave@gmail.com>
Date: Tue, 14 Aug 2018 14:16:08 +1200
Subject: [PATCH] No effective change - adjust test adding @Draft property to
 @Draftable Document

---
 .../java/org/tests/model/draftable/Document.java     | 12 ++++++++++++
 .../transaction/TestTransactionTryResources.java     |  1 +
 2 files changed, 13 insertions(+)

diff --git a/src/test/java/org/tests/model/draftable/Document.java b/src/test/java/org/tests/model/draftable/Document.java
index 5340ee4fe..2652b2237 100644
--- a/src/test/java/org/tests/model/draftable/Document.java
+++ b/src/test/java/org/tests/model/draftable/Document.java
@@ -1,6 +1,7 @@
 package org.tests.model.draftable;
 
 import io.ebean.Finder;
+import io.ebean.annotation.Draft;
 import io.ebean.annotation.DraftOnly;
 import io.ebean.annotation.Draftable;
 
@@ -25,6 +26,9 @@ public class Document extends BaseDomain {
 
   String body;
 
+  @Draft
+  boolean draft;
+
   @DraftOnly
   Timestamp whenPublish;
 
@@ -54,6 +58,14 @@ public class Document extends BaseDomain {
     this.body = body;
   }
 
+  public boolean isDraft() {
+    return draft;
+  }
+
+  public void setDraft(boolean draft) {
+    this.draft = draft;
+  }
+
   public Organisation getOrganisation() {
     return organisation;
   }
diff --git a/src/test/java/org/tests/transaction/TestTransactionTryResources.java b/src/test/java/org/tests/transaction/TestTransactionTryResources.java
index a9978a237..f53640dfd 100644
--- a/src/test/java/org/tests/transaction/TestTransactionTryResources.java
+++ b/src/test/java/org/tests/transaction/TestTransactionTryResources.java
@@ -34,6 +34,7 @@ public class TestTransactionTryResources extends BaseTestCase {
 
     Document document = Document.find.asDraft(doc.getId());
     assertThat(document).isNotNull();
+    assertThat(document.isDraft()).isTrue();
 
     // cleanup
     Ebean.delete(document);