From d4fa62cea9ea40c06dc400c07838b170e868a25b Mon Sep 17 00:00:00 2001
From: Simon Vieille <simon@deblan.fr>
Date: Thu, 19 May 2016 13:41:56 +0200
Subject: [PATCH] Legend attribute

---
 src/Deblan/Csv/Csv.php | 2 ++
 tests/CsvTest.php      | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/Deblan/Csv/Csv.php b/src/Deblan/Csv/Csv.php
index db3c9d5..1054f20 100644
--- a/src/Deblan/Csv/Csv.php
+++ b/src/Deblan/Csv/Csv.php
@@ -63,6 +63,8 @@ class Csv
     {
         $this->setHasLegend(true);
 
+        $this->legend = $values;
+
         $this->addLine($values, 0);
     }
 
diff --git a/tests/CsvTest.php b/tests/CsvTest.php
index 0120fff..4ae789c 100644
--- a/tests/CsvTest.php
+++ b/tests/CsvTest.php
@@ -17,7 +17,10 @@ class CsvTest extends \PHPUnit_Framework_TestCase
 
         $this->assertEquals(false, $csv->getHasLegend());
 
-        $csv->setLegend(array('bim', 'bam'));
+        $csv->setLegend($a = array('bim', 'bam'));
+
+        $this->assertEquals($a, $csv->getLegend());
+        
         $csv->addLine(array('foo', 'bar'));
 
         $this->assertEquals(true, $csv->getHasLegend());