Переглянути джерело

Update tests.

pull/616/head
Chris Smith 8 роки тому
джерело
коміт
7c24c5dfa1
1 змінених файлів з 33 додано та 3 видалено
  1. 33
    3
      test/com/dmdirc/ui/messages/EventFormatterTest.java

+ 33
- 3
test/com/dmdirc/ui/messages/EventFormatterTest.java Переглянути файл

56
 
56
 
57
         when(templateProvider.getFormat(ChannelMessageEvent.class))
57
         when(templateProvider.getFormat(ChannelMessageEvent.class))
58
                 .thenReturn(Optional.of(
58
                 .thenReturn(Optional.of(
59
-                        EventFormat.create("Template {{channel}} meep", Optional.empty())));
59
+                        EventFormat.create(
60
+                                "Template {{channel}} meep",
61
+                                Optional.empty(),
62
+                                Optional.empty(),
63
+                                Optional.empty())));
60
         when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "channel"))
64
         when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "channel"))
61
                 .thenReturn(Optional.of("MONKEY"));
65
                 .thenReturn(Optional.of("MONKEY"));
62
 
66
 
63
         assertEquals("Template MONKEY meep", formatter.format(messageEvent).orElse(null));
67
         assertEquals("Template MONKEY meep", formatter.format(messageEvent).orElse(null));
64
     }
68
     }
65
 
69
 
70
+    @Test
71
+    public void testBeforeAndAfterFormat() {
72
+        messageEvent = new ChannelMessageEvent(channel, null, null);
73
+
74
+        when(templateProvider.getFormat(ChannelMessageEvent.class))
75
+                .thenReturn(Optional.of(
76
+                        EventFormat.create(
77
+                                "Template {{channel}} meep",
78
+                                Optional.of("Before!"),
79
+                                Optional.of("After!"),
80
+                                Optional.empty())));
81
+        when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "channel"))
82
+                .thenReturn(Optional.of("MONKEY"));
83
+
84
+        assertEquals(
85
+                "Before!\nTemplate MONKEY meep\nAfter!",
86
+                formatter.format(messageEvent).orElse(null));
87
+    }
88
+
66
     @Test
89
     @Test
67
     public void testFormatWithFunction() {
90
     public void testFormatWithFunction() {
68
         messageEvent = new ChannelMessageEvent(channel, null, null);
91
         messageEvent = new ChannelMessageEvent(channel, null, null);
69
 
92
 
70
         when(templateProvider.getFormat(ChannelMessageEvent.class))
93
         when(templateProvider.getFormat(ChannelMessageEvent.class))
71
                 .thenReturn(Optional.of(
94
                 .thenReturn(Optional.of(
72
-                        EventFormat.create("Template {{channel|lowercase}} meep",
95
+                        EventFormat.create(
96
+                                "Template {{channel|lowercase}} meep",
97
+                                Optional.empty(),
98
+                                Optional.empty(),
73
                                 Optional.empty())));
99
                                 Optional.empty())));
74
         when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "channel"))
100
         when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "channel"))
75
                 .thenReturn(Optional.of("MONKEY"));
101
                 .thenReturn(Optional.of("MONKEY"));
84
 
110
 
85
         when(templateProvider.getFormat(ChannelMessageEvent.class))
111
         when(templateProvider.getFormat(ChannelMessageEvent.class))
86
                 .thenReturn(Optional.of(
112
                 .thenReturn(Optional.of(
87
-                        EventFormat.create("Template {{message}} meep", Optional.empty())));
113
+                        EventFormat.create(
114
+                                "Template {{message}} meep",
115
+                                Optional.empty(),
116
+                                Optional.empty(),
117
+                                Optional.empty())));
88
         when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "message"))
118
         when(propertyManager.getProperty(messageEvent, ChannelMessageEvent.class, "message"))
89
                 .thenReturn(Optional.of("{{channel}}"));
119
                 .thenReturn(Optional.of("{{channel}}"));
90
 
120
 

Завантаження…
Відмінити
Зберегти