From 194c52db2d608705ea923a09745e0b9d37735cf0 Mon Sep 17 00:00:00 2001
From: DanB <danb@cgrates.org>
Date: Sun, 10 Nov 2013 16:55:52 +0100
Subject: [PATCH] Adding test units for CDR CSV exporter

---
 cdrexporter/csv_test.go | 16 +++++++++++-----
 test.sh                 |  6 +++++-
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/cdrexporter/csv_test.go b/cdrexporter/csv_test.go
index d7181f477..544b30df2 100644
--- a/cdrexporter/csv_test.go
+++ b/cdrexporter/csv_test.go
@@ -22,16 +22,22 @@ import (
 	"bytes"
 	"strings"
 	"testing"
+	"time"
+	"github.com/cgrates/cgrates/utils"
 )
 
-func TestCsv(t *testing.T) {
+func TestCsvCdrWriter(t *testing.T) {
 	writer := &bytes.Buffer{}
-	csvCdrWriter := NewCsvCdrWriter(writer)
-	csvCdrWriter.Write(map[string]string{"First": "test", "Second": "the", "Third": "cdr"})
+	csvCdrWriter := NewCsvCdrWriter(writer, 4, []string{"extra3", "extra1"})
+	ratedCdr := &utils.RatedCDR{ CgrId: utils.FSCgrId("dsafdsaf"), AccId:"dsafdsaf", CdrHost:"192.168.1.1", ReqType:"rated", Direction:"*out", Tenant:"cgrates.org",  
+			TOR:"call", Account:"1001", Subject:"1001", Destination:"1002", AnswerTime:time.Unix(1383813746,0).UTC(), Duration:10, 
+			ExtraFields:map[string]string{"extra1":"val_extra1", "extra2":"val_extra2", "extra3":"val_extra3"}, Cost:1.01,
+			}
+	csvCdrWriter.Write(ratedCdr)
 	csvCdrWriter.Close()
-	expected := "test,the,cdr"
+	expected := "b18944ef4dc618569f24c27b9872827a242bad0c,dsafdsaf,192.168.1.1,rated,*out,cgrates.org,call,1001,1001,1002,2013-11-07 08:42:26 +0000 UTC,10,1.0100,val_extra3,val_extra1"
 	result := strings.TrimSpace(writer.String())
 	if result != expected {
-		t.Errorf("Expected %s was %s.", expected, result)
+		t.Errorf("Expected %s received %s.", expected, result)
 	}
 }
diff --git a/test.sh b/test.sh
index 17b5ad4c2..2ac429a0c 100755
--- a/test.sh
+++ b/test.sh
@@ -9,6 +9,7 @@ go test -i github.com/cgrates/fsock
 go test -i github.com/cgrates/cgrates/cdrs
 go test -i github.com/cgrates/cgrates/utils
 go test -i github.com/cgrates/cgrates/history
+go test -i github.com/cgrates/cgrates/cdrexporter
 
 go test github.com/cgrates/cgrates/engine
 en=$?
@@ -28,5 +29,8 @@ go test github.com/cgrates/fsock
 fs=$?
 go test github.com/cgrates/cgrates/history
 hs=$?
+go test github.com/cgrates/cgrates/cdrexporter
+cdre=$?
 
-exit $en && $sm && $cfg && $bl && $cr && $md && $cdr && $fs && $ut && $hs
+
+exit $en && $sm && $cfg && $bl && $cr && $md && $cdr && $fs && $ut && $hs && $cdre