Improve O1 agent UT coverage

[ric-plt/o1.git] / agent / cmd / o1agent.go

diff --git a/agent/cmd/o1agent.go b/agent/cmd/o1agent.go
index 387f333..38b7341 100755 (executable)
--- a/agent/cmd/o1agent.go
+++ b/agent/cmd/o1agent.go
@@ -33,6 +33,7 @@ import (
 
 var Version string
 var Hash string
+var osExit = os.Exit
 
 type O1Agent struct {
        rmrReady  bool
@@ -73,7 +74,7 @@ func (o *O1Agent) Sighandler() {
 
        <-o.sigChan
        o.nbiClient.Stop()
-       os.Exit(1)
+       osExit(1)
 }
 
 func NewO1Agent() *O1Agent {