summaryrefslogtreecommitdiffstats
path: root/sangria-core/src/test/java/com/tavianator/sangria/core/PotentialAnnotationTest.java
blob: c4ccc3680661eb05ca9ba4de0f0758ff335c9ccf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
/****************************************************************************
 * Sangria                                                                  *
 * Copyright (C) 2014 Tavian Barnes <tavianator@tavianator.com>             *
 *                                                                          *
 * Licensed under the Apache License, Version 2.0 (the "License");          *
 * you may not use this file except in compliance with the License.         *
 * You may obtain a copy of the License at                                  *
 *                                                                          *
 * http://www.apache.org/licenses/LICENSE-2.0                               *
 *                                                                          *
 * Unless required by applicable law or agreed to in writing, software      *
 * distributed under the License is distributed on an "AS IS" BASIS,        *
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. *
 * See the License for the specific language governing permissions and      *
 * limitations under the License.                                           *
 ****************************************************************************/

package com.tavianator.sangria.core;

import java.lang.annotation.Annotation;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import javax.inject.Qualifier;

import com.google.inject.CreationException;
import com.google.inject.Key;
import com.google.inject.name.Names;
import org.hamcrest.Matcher;
import org.hamcrest.Matchers;
import org.junit.Test;

import static org.hamcrest.Matchers.*;
import static org.junit.Assert.*;

/**
 * Tests for {@link PotentialAnnotation}s.
 *
 * @author Tavian Barnes (tavianator@tavianator.com)
 * @version 1.1
 * @since 1.1
 */
public class PotentialAnnotationTest {
    @Retention(RetentionPolicy.RUNTIME)
    @Qualifier
    private @interface Simple {
    }

    private final PotentialAnnotation none = PotentialAnnotation.none();
    private final Annotation nameAnnotation = Names.named("name");

    @Test
    public void testHasAnnotation() {
        assertThat(none.hasAnnotation(), is(false));
        assertThat(none.annotatedWith(Simple.class).hasAnnotation(), is(true));
        assertThat(none.annotatedWith(nameAnnotation).hasAnnotation(), is(true));
    }

    @Test(expected = CreationException.class)
    public void testInvalidAnnotatedWithType() {
        none.annotatedWith(Simple.class)
                .annotatedWith(Simple.class);
    }

    @Test(expected = CreationException.class)
    public void testInvalidAnnotatedWithInstance() {
        none.annotatedWith(Names.named("name"))
                .annotatedWith(Names.named("name"));
    }

    @Test
    public void testGetKey() {
        assertThat(none.getKey(String.class),
                equalTo(new Key<String>() { }));
        assertThat(none.annotatedWith(Simple.class).getKey(String.class),
                equalTo(new Key<String>(Simple.class) { }));
        assertThat(none.annotatedWith(nameAnnotation).getKey(String.class),
                equalTo(new Key<String>(nameAnnotation) { }));
    }

    @Test
    public void testVisitor() {
        PotentialAnnotation.Visitor<String> visitor = new PotentialAnnotation.Visitor<String>() {
            @Override
            public String visitNoAnnotation() {
                return "none";
            }

            @Override
            public String visitAnnotationType(Class<? extends Annotation> annotationType) {
                assertThat((Object)annotationType, equalTo((Object)Simple.class));
                return "type";
            }

            @Override
            public String visitAnnotationInstance(Annotation annotation) {
                assertThat(annotation, equalTo(nameAnnotation));
                return "instance";
            }
        };

        assertThat(none.accept(visitor), equalTo("none"));
        assertThat(none.annotatedWith(Simple.class).accept(visitor), equalTo("type"));
        assertThat(none.annotatedWith(nameAnnotation).accept(visitor), equalTo("instance"));
    }

    @Test
    public void testToString() {
        assertThat(none.toString(),
                equalTo("[no annotation]"));
        assertThat(none.annotatedWith(Simple.class).toString(),
                equalTo("@com.tavianator.sangria.core.PotentialAnnotationTest.Simple"));
        assertThat(none.annotatedWith(nameAnnotation).toString(),
                equalTo("@com.google.inject.name.Named(value=name)"));
    }

    /**
     * Needed to avoid compilation error to to inferred type being anonymous class.
     */
    private static <T> Matcher<Key<T>> equalTo(Key<T> key) {
        return Matchers.equalTo(key);
    }

    private static <T> Matcher<T> equalTo(T object) {
        return Matchers.equalTo(object);
    }
}