✅ test: added new function for testing #14

pnguyen215 · Jan 14, 2024 · 5d2d3c6 · 5d2d3c6
1 parent 6b96227
commit 5d2d3c6
Show file tree

Hide file tree

Showing 4 changed files with 63 additions and 1 deletion.
diff --git a/example/ami_test.go b/example/ami_test.go
@@ -1,6 +1,8 @@
 package example
 
 import (
+	"fmt"
+	"log"
 	"testing"
 	"time"
 
@@ -44,3 +46,52 @@ func TestDialOut(t *testing.T) {
 func TestChanspy(t *testing.T) {
 
 }
+
+func TestGetSIPPeersStatus(t *testing.T) {
+	c, err := createConn()
+	if err != nil {
+		ami.D().Error(err.Error())
+		return
+	}
+	c.Core().AddSession()
+	peers, err := c.Core().GetSIPPeersStatus(c.Context())
+	if err != nil {
+		ami.D().Error(err.Error())
+		return
+	}
+	log.Println(fmt.Sprintf("SIP peer status: %v", ami.JsonString(peers[1])))
+}
+
+func TestGetQueueStatuses(t *testing.T) {
+	c, err := createConn()
+	if err != nil {
+		ami.D().Error(err.Error())
+		return
+	}
+	c.Core().AddSession()
+	c.Core().Dictionary.SetEnabledForceTranslate(true)
+	c.Core().Dictionary.AddKeyLinkTranslator("https://raw.githubusercontent.com/pnguyen215/gear-insights-free/master/ami.dictionaries.json")
+	peers, err := c.Core().GetQueueStatuses(c.Context(), "")
+	if err != nil {
+		ami.D().Error(err.Error())
+		return
+	}
+	log.Println(fmt.Sprintf("SIP queues status: %v", ami.JsonString(peers)))
+}
+
+func TestGetQueueSummary(t *testing.T) {
+	c, err := createConn()
+	if err != nil {
+		ami.D().Error(err.Error())
+		return
+	}
+	c.Core().AddSession()
+	c.Core().Dictionary.SetEnabledForceTranslate(true)
+	c.Core().Dictionary.AddKeyLinkTranslator("https://raw.githubusercontent.com/pnguyen215/gear-insights-free/master/ami.dictionaries.json")
+	peers, err := c.Core().GetQueueSummary(c.Context(), "")
+	if err != nil {
+		ami.D().Error(err.Error())
+		return
+	}
+	log.Println(fmt.Sprintf("SIP queues summary: %v", ami.JsonString(peers)))
+}
diff --git a/pkg/ami/ami.go b/pkg/ami/ami.go
@@ -28,6 +28,15 @@ func (c *AMI) Core() *AMICore {
 	return c.c
 }
 
+func (c *AMI) SetContext(value context.Context) *AMI {
+	c.ctx = value
+	return c
+}
+
+func (c *AMI) Context() context.Context {
+	return c.ctx
+}
+
 // Action sends an AMI action message to the Asterisk server.
 // If the action message does not have an ActionID, it adds one automatically.
 // The method returns true if the action message is successfully sent, otherwise false.
@@ -457,6 +466,7 @@ func serve(conn net.Conn, request AmiClient) (*AMI, error) {
 		return ins, err
 	}
 	ins.SetCore(c)
+	ins.SetContext(ctx)
 	ins.release(ctx)
 	return ins, nil
 }
diff --git a/pkg/ami/ami_core.go b/pkg/ami/ami_core.go
@@ -41,7 +41,7 @@ func (c *AMICore) SetDictionary(dictionary *AMIDictionary) *AMICore {
 	return c
 }
 
-func (c *AMICore) ResetUUID() *AMICore {
+func (c *AMICore) AddSession() *AMICore {
 	c.SetUUID(GenUUIDShorten())
 	c.socket.SetUUID(c.UUID)
 	return c

diff --git a/pkg/ami/ami_model.go b/pkg/ami/ami_model.go
@@ -27,6 +27,7 @@ type AmiClient struct {
 }
 
 type AMI struct {
+	ctx     context.Context
 	err     chan error
 	mutex   sync.RWMutex
 	conn    net.Conn