55import static org .mockito .Mockito .verify ;
66import static org .mockito .Mockito .verifyZeroInteractions ;
77import static org .mockito .Mockito .when ;
8- import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .CONTEXT_SHUTTERED ;
9- import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .CONTEXT_UNSHUTTERED ;
10- import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .SHUTTERING_STARTED ;
11- import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .UNSHUTTERING_STARTED ;
8+ import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .SHUTTERED ;
9+ import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .UNSHUTTERED ;
10+ import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .SHUTTERING_IN_PROGRESS ;
11+ import static uk .gov .justice .services .management .shuttering .observers .shuttering .ContextShutteredState .UNSHUTTERING_IN_PROGRESS ;
1212
1313import uk .gov .justice .services .common .util .UtcClock ;
1414import uk .gov .justice .services .jmx .api .command .ShutterSystemCommand ;
1515import uk .gov .justice .services .jmx .api .command .UnshutterSystemCommand ;
1616import uk .gov .justice .services .management .shuttering .events .ShutteringRequestedEvent ;
1717import uk .gov .justice .services .management .shuttering .events .UnshutteringRequestedEvent ;
18- import uk .gov .justice .services .management .shuttering .observers .ContextShutteredStateObserver ;
18+ import uk .gov .justice .services .management .shuttering .observers .ShutteringStateRegistry ;
1919
2020import java .time .ZonedDateTime ;
2121
@@ -43,7 +43,7 @@ public class ShutteringSystemCommandHandlerTest {
4343 private Event <UnshutteringRequestedEvent > unshutteringRequestedEventFirer ;
4444
4545 @ Mock
46- private ContextShutteredStateObserver contextShutteredStateObserver ;
46+ private ShutteringStateRegistry shutteringStateRegistry ;
4747
4848 @ Mock
4949 private Logger logger ;
@@ -63,7 +63,7 @@ public void shouldFireEventOnShutteringRequested() throws Exception {
6363 final ZonedDateTime now = new UtcClock ().now ();
6464 final ShutterSystemCommand shutterSystemCommand = new ShutterSystemCommand ();
6565
66- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (CONTEXT_UNSHUTTERED );
66+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (UNSHUTTERED );
6767 when (clock .now ()).thenReturn (now );
6868
6969 shutteringSystemCommandHandler .onShutterRequested (shutterSystemCommand );
@@ -82,7 +82,7 @@ public void shouldFireEventOnUnshutteringRequested() throws Exception {
8282 final ZonedDateTime now = new UtcClock ().now ();
8383 final UnshutterSystemCommand unshutterSystemCommand = new UnshutterSystemCommand ();
8484
85- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (CONTEXT_SHUTTERED );
85+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (SHUTTERED );
8686 when (clock .now ()).thenReturn (now );
8787
8888 shutteringSystemCommandHandler .onUnshutterRequested (unshutterSystemCommand );
@@ -98,66 +98,66 @@ public void shouldFireEventOnUnshutteringRequested() throws Exception {
9898 @ Test
9999 public void shouldIgnoreShutteringRequestIfTheShutteredStateIsContextShuttered () throws Exception {
100100
101- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (CONTEXT_SHUTTERED );
101+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (SHUTTERED );
102102
103103 shutteringSystemCommandHandler .onShutterRequested (new ShutterSystemCommand ());
104104
105105 verifyZeroInteractions (shutteringRequestedEventFirer );
106- verify (logger ).info ("Ignoring command 'SHUTTER'. Context shuttered state is 'CONTEXT_SHUTTERED '" );
106+ verify (logger ).info ("Ignoring command 'SHUTTER'. Context shuttered state is 'SHUTTERED '" );
107107 }
108108
109109 @ Test
110110 public void shouldIgnoreShutteringRequestIfTheShutteredStateIsShutteringStarted () throws Exception {
111111
112- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (SHUTTERING_STARTED );
112+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (SHUTTERING_IN_PROGRESS );
113113
114114 shutteringSystemCommandHandler .onShutterRequested (new ShutterSystemCommand ());
115115
116116 verifyZeroInteractions (shutteringRequestedEventFirer );
117- verify (logger ).info ("Ignoring command 'SHUTTER'. Context shuttered state is 'SHUTTERING_STARTED '" );
117+ verify (logger ).info ("Ignoring command 'SHUTTER'. Context shuttered state is 'SHUTTERING_IN_PROGRESS '" );
118118 }
119119
120120 @ Test
121121 public void shouldIgnoreShutteringRequestIfTheShutteredStateIsUnshutteringStarted () throws Exception {
122122
123- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (UNSHUTTERING_STARTED );
123+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (UNSHUTTERING_IN_PROGRESS );
124124
125125 shutteringSystemCommandHandler .onShutterRequested (new ShutterSystemCommand ());
126126
127127 verifyZeroInteractions (shutteringRequestedEventFirer );
128- verify (logger ).info ("Ignoring command 'SHUTTER'. Context shuttered state is 'UNSHUTTERING_STARTED '" );
128+ verify (logger ).info ("Ignoring command 'SHUTTER'. Context shuttered state is 'UNSHUTTERING_IN_PROGRESS '" );
129129 }
130130
131131 @ Test
132132 public void shouldIgnoreUnshutteringRequestIfTheShutteredStateIsContextUnshuttered () throws Exception {
133133
134- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (CONTEXT_UNSHUTTERED );
134+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (UNSHUTTERED );
135135
136136 shutteringSystemCommandHandler .onUnshutterRequested (new UnshutterSystemCommand ());
137137
138138 verifyZeroInteractions (shutteringRequestedEventFirer );
139- verify (logger ).info ("Ignoring command 'UNSHUTTER'. Context shuttered state is 'CONTEXT_UNSHUTTERED '" );
139+ verify (logger ).info ("Ignoring command 'UNSHUTTER'. Context shuttered state is 'UNSHUTTERED '" );
140140 }
141141
142142 @ Test
143143 public void shouldIgnoreUnshutteringRequestIfTheShutteredStateIsShutteringStarted () throws Exception {
144144
145- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (SHUTTERING_STARTED );
145+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (SHUTTERING_IN_PROGRESS );
146146
147147 shutteringSystemCommandHandler .onUnshutterRequested (new UnshutterSystemCommand ());
148148
149149 verifyZeroInteractions (shutteringRequestedEventFirer );
150- verify (logger ).info ("Ignoring command 'UNSHUTTER'. Context shuttered state is 'SHUTTERING_STARTED '" );
150+ verify (logger ).info ("Ignoring command 'UNSHUTTER'. Context shuttered state is 'SHUTTERING_IN_PROGRESS '" );
151151 }
152152
153153 @ Test
154154 public void shouldIgnoreUnshutteringRequestIfTheShutteredStateIsUnshutteringStarted () throws Exception {
155155
156- when (contextShutteredStateObserver .getShutteredState ()).thenReturn (UNSHUTTERING_STARTED );
156+ when (shutteringStateRegistry .getShutteredState ()).thenReturn (UNSHUTTERING_IN_PROGRESS );
157157
158158 shutteringSystemCommandHandler .onUnshutterRequested (new UnshutterSystemCommand ());
159159
160160 verifyZeroInteractions (shutteringRequestedEventFirer );
161- verify (logger ).info ("Ignoring command 'UNSHUTTER'. Context shuttered state is 'UNSHUTTERING_STARTED '" );
161+ verify (logger ).info ("Ignoring command 'UNSHUTTER'. Context shuttered state is 'UNSHUTTERING_IN_PROGRESS '" );
162162 }
163163}
0 commit comments