[#84] Added StopAll() invocation for sigterm and sigint os events. (#85)

* [#84] Added StopAll() invocation for sigterm and sigint os events.

* [#84] Fixed the test case.

Author: nandagopalan

lifecycle/simple_component.go

@@ -61,6 +61,7 @@ func (sc *SimpleComponent) Start() (err error) {
 			sc.CompState = Error
 		} else {
 			sc.CompState = Running
+
 		}
 		if sc.OnStateChange != nil {
 			sc.OnStateChange(Starting, sc.CompState)
@@ -274,5 +275,17 @@ func NewSimpleComponentManager() ComponentManager {
 		cMutex:     &sync.RWMutex{},
 		waitChan:   make(chan struct{}),
 	}
+	sigs := make(chan os.Signal, 1)
+
+	signal.Notify(sigs, syscall.SIGINT, syscall.SIGTERM)
+
+	go func(manager ComponentManager) {
+		sig := <-sigs
+		logger.ErrorF("Received signal: %v, Stopping all components", sig)
+		err := manager.StopAll()
+		if err != nil {
+			logger.ErrorF("Error stopping components: %v", err)
+		}
+	}(manager)
 	return manager
 }

managers/item_manager_test.go

@@ -52,7 +52,7 @@ func TestItemManager_Items_AfterUnregister(t *testing.T) {
 	manager.Unregister("item2")
 
 	items := manager.Items()
-	expectedItems := []int{1, 3}
+	assert.ListHas(t, 1, items)
+	assert.ListHas(t, 3, items)
 
-	assert.ElementsMatch(t, items, expectedItems...)
 }