Browse Source

Fix unit test mocking.

pull/133/head
Greg Holmes 9 years ago
parent
commit
468064de98

+ 4
- 0
awaycolours/test/com/dmdirc/addons/awaycolours/AwayColoursManagerTest.java View File

22
 
22
 
23
 package com.dmdirc.addons.awaycolours;
23
 package com.dmdirc.addons.awaycolours;
24
 
24
 
25
+import com.dmdirc.Channel;
25
 import com.dmdirc.ChannelClientProperty;
26
 import com.dmdirc.ChannelClientProperty;
26
 import com.dmdirc.DMDircMBassador;
27
 import com.dmdirc.DMDircMBassador;
27
 import com.dmdirc.config.ConfigBinder;
28
 import com.dmdirc.config.ConfigBinder;
54
     @Mock private ChannelUserAwayEvent awayEvent;
55
     @Mock private ChannelUserAwayEvent awayEvent;
55
     @Mock private ChannelUserBackEvent backEvent;
56
     @Mock private ChannelUserBackEvent backEvent;
56
     @Mock private ChannelClientInfo user;
57
     @Mock private ChannelClientInfo user;
58
+    @Mock private Channel channel;
57
     @Mock private Map<Object, Object> map;
59
     @Mock private Map<Object, Object> map;
58
     @Mock private ColourManager colourManager;
60
     @Mock private ColourManager colourManager;
59
     private AwayColoursManager instance;
61
     private AwayColoursManager instance;
69
         redColour = Colour.RED;
71
         redColour = Colour.RED;
70
         blackColour = Colour.BLACK;
72
         blackColour = Colour.BLACK;
71
         when(awayEvent.getUser()).thenReturn(user);
73
         when(awayEvent.getUser()).thenReturn(user);
74
+        when(awayEvent.getChannel()).thenReturn(channel);
72
         when(backEvent.getUser()).thenReturn(user);
75
         when(backEvent.getUser()).thenReturn(user);
76
+        when(backEvent.getChannel()).thenReturn(channel);
73
         when(user.getMap()).thenReturn(map);
77
         when(user.getMap()).thenReturn(map);
74
         when(config.getBinder()).thenReturn(binder);
78
         when(config.getBinder()).thenReturn(binder);
75
         when(binder.withDefaultDomain(anyString())).thenReturn(binder);
79
         when(binder.withDefaultDomain(anyString())).thenReturn(binder);

Loading…
Cancel
Save