From 505b147bd23e26d21932af2c3336c9f4607bbcce Mon Sep 17 00:00:00 2001
From: Tavian Barnes <tavianator@tavianator.com>
Date: Wed, 7 May 2014 21:09:20 -0400
Subject: core: Make DelayedError complain if cancel() is called too late.

---
 .../tavianator/sangria/core/DelayedErrorTest.java  | 29 +++++++++++++++++++++-
 1 file changed, 28 insertions(+), 1 deletion(-)

(limited to 'sangria-core/src/test')

diff --git a/sangria-core/src/test/java/com/tavianator/sangria/core/DelayedErrorTest.java b/sangria-core/src/test/java/com/tavianator/sangria/core/DelayedErrorTest.java
index 349dced..f27aea8 100644
--- a/sangria-core/src/test/java/com/tavianator/sangria/core/DelayedErrorTest.java
+++ b/sangria-core/src/test/java/com/tavianator/sangria/core/DelayedErrorTest.java
@@ -32,7 +32,7 @@ import static org.hamcrest.Matchers.*;
  * Tests for {@link DelayedError}.
  *
  * @author Tavian Barnes (tavianator@tavianator.com)
- * @version 1.0
+ * @version 1.1
  * @since 1.0
  */
 public class DelayedErrorTest {
@@ -83,4 +83,31 @@ public class DelayedErrorTest {
             }
         });
     }
+
+    @Test
+    public void testCancel() {
+        Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                DelayedError error = DelayedError.create(binder(), "Message");
+                error.cancel();
+            }
+        });
+    }
+
+    @Test
+    public void testLateCancel() {
+        final DelayedError[] errorHolder = new DelayedError[1];
+
+        Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                errorHolder[0] = DelayedError.create(binder(), "Message");
+                errorHolder[0].cancel();
+            }
+        });
+
+        thrown.expect(IllegalStateException.class);
+        errorHolder[0].cancel();
+    }
 }
-- 
cgit v1.2.3