Add new unit tests on agents

This commit is contained in:
armirveliaj
2024-07-08 10:27:49 -04:00
committed by Dan Christian Bogos
parent 05f7981cae
commit a26c9bb29a
3 changed files with 67 additions and 0 deletions

View File

@@ -1579,3 +1579,11 @@ func TestFseventV1AuthorizeArgsNoCGRFlags(t *testing.T) {
t.Errorf("")
}
}
func TestFseventV1TerminateSessionArgs(t *testing.T) {
fsev := FSEvent{}
args := fsev.V1TerminateSessionArgs()
if !args.TerminateSession {
t.Errorf("Expected TerminateSession to be true, got false")
}
}

View File

@@ -1023,3 +1023,12 @@ func TestKamEventAsMapStringInterfaceUsageKey(t *testing.T) {
t.Errorf("expected %v, got %v", expected, result)
}
}
func TestKamEventMissingParameterUnsupportedEvent(t *testing.T) {
kev := KamEvent{
EVENT: "unsupported_event",
}
if result := kev.MissingParameter(); !result {
t.Errorf("Expected true for unsupported event, got false")
}
}

View File

@@ -1865,3 +1865,53 @@ func TestUpdateDnsRRHeaderRrtypeConversion(t *testing.T) {
t.Errorf("Expected Rrtype to be 28, got %d", v.Rrtype)
}
}
func TestLibDnsUpdateDnsAnswerARecord(t *testing.T) {
q := []dns.RR{
&dns.A{},
}
qType := dns.TypeA
qName := "cgrates.com"
path1 := []string{utils.DNSHdr, utils.DNSName}
value1 := "new.cgrates.com"
_, err1 := updateDnsAnswer(q, qType, qName, path1, value1, false)
if err1 != nil {
t.Errorf("Unexpected error for valid path: %v", err1)
}
updatedHeader := q[0].(*dns.A).Hdr
if updatedHeader.Name != value1 {
t.Errorf("Expected DNS header name to be %s, got %s", value1, updatedHeader.Name)
}
path2 := []string{utils.DNSA}
value2 := "192.168.2.1"
_, err2 := updateDnsAnswer(q, qType, qName, path2, value2, false)
if err2 != nil {
t.Errorf("Unexpected error for valid path: %v", err2)
}
updatedIP := q[0].(*dns.A).A.String()
if updatedIP != value2 {
t.Errorf("Expected IP address to be %s, got %s", value2, updatedIP)
}
path3 := []string{"invalid_path"}
_, err3 := updateDnsAnswer(q, qType, qName, path3, value1, false)
if err3 == nil {
t.Errorf("Expected error for invalid path, got nil")
}
}
func TestLibDnsUpdateDnsOptionCase1(t *testing.T) {
var q []dns.EDNS0
path := []string{}
value := "value"
newBranch := true
updatedQ, err := updateDnsOption(q, path, value, newBranch)
if err == nil {
t.Errorf("Unexpected error for valid path: %v", err)
}
if len(updatedQ) == 1 {
t.Errorf("Expected updatedQ length to be 1, got %d", len(updatedQ))
}
}