From 1bb9b1e01484a314f7b17022a329951ae9a51281 Mon Sep 17 00:00:00 2001
From: DanB <danb@cgrates.org>
Date: Wed, 6 Jan 2016 13:50:22 +0100
Subject: [PATCH] Diameter test improvement

---
 agents/dmtagent_it_test.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/agents/dmtagent_it_test.go b/agents/dmtagent_it_test.go
index 2de7b88fb..9a449b0b0 100644
--- a/agents/dmtagent_it_test.go
+++ b/agents/dmtagent_it_test.go
@@ -320,8 +320,11 @@ func TestDmtAgentSendCCRTerminate(t *testing.T) {
 	if err := dmtClient.SendMessage(m); err != nil {
 		t.Error(err)
 	}
-	time.Sleep(time.Duration(100) * time.Millisecond)
+	time.Sleep(time.Duration(150) * time.Millisecond)
 	msg := dmtClient.ReceivedMessage()
+	if msg == nil {
+		t.Fatal("No answer to CCR terminate received")
+	}
 	if avps, err := msg.FindAVPsWithPath([]interface{}{"Granted-Service-Unit", "CC-Time"}, dict.UndefinedVendorID); err != nil {
 		t.Error(err)
 	} else if len(avps) == 0 {