1 files changed, 269 insertions, 0 deletions
diff --git a/sangria-contextual/src/test/java/com/tavianator/sangria/contextual/ContextSensitiveBinderTest.java b/sangria-contextual/src/test/java/com/tavianator/sangria/contextual/ContextSensitiveBinderTest.java
new file mode 100644
index 0000000..24f073e
--- /dev/null
+++ b/sangria-contextual/src/test/java/com/tavianator/sangria/contextual/ContextSensitiveBinderTest.java
@@ -0,0 +1,269 @@
+/*********************************************************************
+ * Sangria                                                           *
+ * Copyright (C) 2014 Tavian Barnes <tavianator@tavianator.com>      *
+ *                                                                   *
+ * This library is free software. It comes without any warranty, to  *
+ * the extent permitted by applicable law. You can redistribute it   *
+ * and/or modify it under the terms of the Do What The Fuck You Want *
+ * To Public License, Version 2, as published by Sam Hocevar. See    *
+ * the COPYING file or http://www.wtfpl.net/ for more details.       *
+ *********************************************************************/
+
+package com.tavianator.sangria.contextual;
+
+import javax.inject.Inject;
+import javax.inject.Named;
+
+import com.google.inject.AbstractModule;
+import com.google.inject.CreationException;
+import com.google.inject.Guice;
+import com.google.inject.Injector;
+import com.google.inject.Key;
+import com.google.inject.MembersInjector;
+import com.google.inject.Provider;
+import com.google.inject.ProvisionException;
+import com.google.inject.TypeLiteral;
+import com.google.inject.name.Names;
+import com.google.inject.spi.InjectionPoint;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+
+import static org.hamcrest.Matchers.*;
+import static org.junit.Assert.*;
+
+/**
+ * Tests for {@link ContextSensitiveBinder}.
+ *
+ * @author Tavian Barnes (tavianator@tavianator.com)
+ * @version 1.0
+ * @since 1.0
+ */
+public class ContextSensitiveBinderTest {
+    public @Rule ExpectedException thrown = ExpectedException.none();
+
+    private static class SelfProvider implements ContextSensitiveProvider<String> {
+        @Override
+        public String getInContext(InjectionPoint injectionPoint) {
+            return injectionPoint.getDeclaringType().getRawType().getSimpleName();
+        }
+
+        @Override
+        public String getInUnknownContext() {
+            return "<unknown>";
+        }
+    }
+
+    private static class HasSelf {
+        @Inject @Named("self") String self;
+        @Inject @Named("self") Provider<String> selfProvider;
+    }
+
+    @Test
+    public void testProviderClass() {
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(SelfProvider.class);
+            }
+        });
+
+        HasSelf hasSelf = injector.getInstance(HasSelf.class);
+        assertThat(hasSelf.self, equalTo("HasSelf"));
+        assertThat(hasSelf.selfProvider.get(), equalTo("<unknown>"));
+    }
+
+    @Test
+    public void testProviderTypeLiteral() {
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(new TypeLiteral<SelfProvider>() { });
+            }
+        });
+
+        HasSelf hasSelf = injector.getInstance(HasSelf.class);
+        assertThat(hasSelf.self, equalTo("HasSelf"));
+        assertThat(hasSelf.selfProvider.get(), equalTo("<unknown>"));
+    }
+
+    @Test
+    public void testProviderKey() {
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                bind(SelfProvider.class)
+                        .annotatedWith(Names.named("self"))
+                        .to(SelfProvider.class);
+
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(new Key<SelfProvider>(Names.named("self")) { });
+            }
+        });
+
+        HasSelf hasSelf = injector.getInstance(HasSelf.class);
+        assertThat(hasSelf.self, equalTo("HasSelf"));
+        assertThat(hasSelf.selfProvider.get(), equalTo("<unknown>"));
+    }
+
+    @Test
+    public void testProviderInstance() {
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(new SelfProvider());
+            }
+        });
+
+        HasSelf hasSelf = injector.getInstance(HasSelf.class);
+        assertThat(hasSelf.self, equalTo("HasSelf"));
+        assertThat(hasSelf.selfProvider.get(), equalTo("<unknown>"));
+    }
+
+    @Test
+    public void testDeDuplication() {
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder contextualBinder = ContextSensitiveBinder.create(binder());
+                contextualBinder
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(SelfProvider.class);
+                contextualBinder
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(SelfProvider.class);
+            }
+        });
+        HasSelf hasSelf = injector.getInstance(HasSelf.class);
+        assertThat(hasSelf.self, equalTo("HasSelf"));
+        assertThat(hasSelf.selfProvider.get(), equalTo("<unknown>"));
+    }
+
+    private static class RequiredContextProvider implements ContextSensitiveProvider<String> {
+        @Override
+        public String getInContext(InjectionPoint injectionPoint) {
+            return injectionPoint.getDeclaringType().getRawType().getSimpleName();
+        }
+
+        @Override
+        public String getInUnknownContext() {
+            throw new IllegalStateException("@Named(\"self\") injection not supported here");
+        }
+    }
+
+    @Test
+    public void testContextRequired() {
+        thrown.expect(ProvisionException.class);
+
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(RequiredContextProvider.class);
+            }
+        });
+
+        HasSelf hasSelf = injector.getInstance(HasSelf.class);
+        assertThat(hasSelf.self, equalTo("HasSelf"));
+        hasSelf.selfProvider.get();
+    }
+
+    private static class Recursive {
+        @Inject HasSelf hasSelf;
+        String self;
+    }
+
+    private static class RecursiveProvider implements ContextSensitiveProvider<Recursive> {
+        @Inject MembersInjector<Recursive> membersInjector;
+
+        @Override
+        public Recursive getInContext(InjectionPoint injectionPoint) {
+            Recursive result = new Recursive();
+            membersInjector.injectMembers(result);
+            result.self = injectionPoint.getDeclaringType().getRawType().getSimpleName();
+            return result;
+        }
+
+        @Override
+        public Recursive getInUnknownContext() {
+            Recursive result = new Recursive();
+            membersInjector.injectMembers(result);
+            result.self = "<unknown>";
+            return result;
+        }
+    }
+
+    private static class HasRecursive {
+        @Inject Recursive recursive;
+    }
+
+    @Test
+    public void testRecursiveProvision() {
+        Injector injector = Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder contextualBinder = ContextSensitiveBinder.create(binder());
+                contextualBinder
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"))
+                        .toContextSensitiveProvider(SelfProvider.class);
+                contextualBinder
+                        .bind(Recursive.class)
+                        .toContextSensitiveProvider(new RecursiveProvider());
+            }
+        });
+
+        HasRecursive hasRecursive = injector.getInstance(HasRecursive.class);
+        assertThat(hasRecursive.recursive.self, equalTo("HasRecursive"));
+        assertThat(hasRecursive.recursive.hasSelf.self, equalTo("HasSelf"));
+
+        Recursive recursive = injector.getInstance(Recursive.class);
+        assertThat(recursive.self, equalTo("<unknown>"));
+        assertThat(recursive.hasSelf.self, equalTo("HasSelf"));
+    }
+
+    @Test
+    public void testIncompleteEdsl1() {
+        thrown.expect(CreationException.class);
+        thrown.expectMessage("Missing call to toContextSensitiveProvider() for java.lang.String");
+
+        Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class);
+            }
+        });
+    }
+
+    @Test
+    public void testIncompleteEdsl2() {
+        thrown.expect(CreationException.class);
+        thrown.expectMessage("Missing call to toContextSensitiveProvider() for java.lang.String annotated with "
+                + "@com.google.inject.name.Named(value=self)");
+
+        Guice.createInjector(new AbstractModule() {
+            @Override
+            protected void configure() {
+                ContextSensitiveBinder.create(binder())
+                        .bind(String.class)
+                        .annotatedWith(Names.named("self"));
+            }
+        });
+    }
+}