Expose more internals for testing purposes.

Author: samuel-williams-shopify

lib/async/container/controller.rb

@@ -24,19 +24,30 @@ class Controller
 			# Initialize the controller.
 			# @parameter notify [Notify::Client] A client used for process readiness notifications.
 			def initialize(notify: Notify.open!, container_class: Container, graceful_stop: true)
-				@container = nil
+				@notify = notify
 				@container_class = container_class
+				@graceful_stop = graceful_stop
 
-				@notify = notify
+				@container = nil
 				@signals = {}
 
 				self.trap(SIGHUP) do
 					self.restart
 				end
-				
-				@graceful_stop = graceful_stop
 			end
 
+			# The notify client used by the controller.
+			attr :notify
+			
+			# The container class used by the controller.
+			attr :container_class
+			
+			# The graceful stop flag used by the controller.
+			attr :graceful_stop
+			
+			# The current container being managed by the controller.
+			attr :container
+			
 			# The state of the controller.
 			# @returns [String]
 			def state_string
@@ -60,9 +71,6 @@ def trap(signal, &block)
 				@signals[signal] = block
 			end
 
-			# The current container being managed by the controller.
-			attr :container
-			
 			# Create a policy for managing child lifecycle events.
 			# Can be overridden by a sub-class to provide a custom policy.
 			# @returns [Policy] The policy to use for the container.