From d4494b9a9e57f314d4118f26a9d769ecb6dd1247 Mon Sep 17 00:00:00 2001
From: Trial97 <alexandru.tripon@itsyscom.com>
Date: Sun, 16 Dec 2018 11:34:06 +0200
Subject: [PATCH] Fix httpagent integration test

---
 agents/httpagent_it_test.go | 22 +++++++++-------------
 1 file changed, 9 insertions(+), 13 deletions(-)

diff --git a/agents/httpagent_it_test.go b/agents/httpagent_it_test.go
index 96f4ef0b1..2fe0d2d82 100644
--- a/agents/httpagent_it_test.go
+++ b/agents/httpagent_it_test.go
@@ -193,20 +193,16 @@ func testHAitAuth1001(t *testing.T) {
 	if err != nil {
 		t.Error(err)
 	}
-	var eXml []byte
-	if t := time.Now(); t.Weekday() != 6 && t.Weekday() != 7 { // Different rating plans for weekend
-		eXml = []byte(`<?xml version="1.0" encoding="UTF-8"?>
-<response>
-  <Allow>1</Allow>
-  <MaxDuration>6042</MaxDuration>
-</response>`)
-	} else {
-		eXml = []byte(`<?xml version="1.0" encoding="UTF-8"?>
-<response>
-  <Allow>1</Allow>
-  <MaxDuration>10800</MaxDuration>
-</response>`)
+	// var eXml []byte
+	maxDuration := 6042
+	if t := time.Now(); t.Weekday() == time.Saturday || t.Weekday() == time.Sunday { // Different rating plans for weekend
+		maxDuration = 10800
 	}
+	eXml := []byte(fmt.Sprintf(`<?xml version="1.0" encoding="UTF-8"?>
+<response>
+  <Allow>1</Allow>
+  <MaxDuration>%v</MaxDuration>
+</response>`, maxDuration))
 	if body, err := ioutil.ReadAll(rply.Body); err != nil {
 		t.Error(err)
 	} else if !reflect.DeepEqual(eXml, body) {