From 2750d91467143c874dc34515ee1c3913db289954 Mon Sep 17 00:00:00 2001
From: adragusin <adrian.dragusin@itsyscom.com>
Date: Mon, 18 Nov 2019 09:34:01 +0200
Subject: [PATCH] Finished tests for CGRRecordWriter

---
 utils/cgrrecordwriter_test.go | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/utils/cgrrecordwriter_test.go b/utils/cgrrecordwriter_test.go
index d7e92ddce..cffeae96f 100644
--- a/utils/cgrrecordwriter_test.go
+++ b/utils/cgrrecordwriter_test.go
@@ -34,19 +34,31 @@ func TestNewCgrIORecordWriter(t *testing.T) {
 	}
 }
 
+type writer2 struct{}
+
+func (*writer2) Write(p []byte) (n int, err error) { return 0, ErrNoMoreData }
+
 func TestWrite(t *testing.T) {
 	//empty check
 	args := new(bytes.Buffer)
 	rw := NewCgrIORecordWriter(args)
 	record := []string{"test1", "test2"}
-	rcv := rw.Write(record)
-	if rcv != nil {
-		t.Errorf("Expecting nil, received %+v", rcv)
+	err := rw.Write(record)
+	if err != nil {
+		t.Errorf("Expecting nil, received %+v", err)
 	}
 	eOut := "test1test2\n"
 	if !reflect.DeepEqual(eOut, args.String()) {
 		t.Errorf("Expected %q, received: %q", eOut, args.String())
 	}
+	//err check
+	args2 := &writer2{}
+	rw = NewCgrIORecordWriter(args2)
+	record = []string{"test1", "test2"}
+	err = rw.Write(record)
+	if err != ErrNoMoreData {
+		t.Errorf("Expecting %+v, received %v", ErrNoMoreData, err)
+	}
 
 }
 func TestCgrIORecordWriterFlush(t *testing.T) {