1 files changed, 19 insertions, 0 deletions
diff --git a/tests/unit/test_csvcmp.py b/tests/unit/test_csvcmp.py
index c7ffe2f..984a61f 100644
--- a/tests/unit/test_csvcmp.py
+++ b/tests/unit/test_csvcmp.py
@@ -3,6 +3,7 @@ import pytest
 
 from gn3.csvcmp import csv_diff
 from gn3.csvcmp import fill_csv
+from gn3.csvcmp import get_allowable_sampledata_headers
 from gn3.csvcmp import remove_insignificant_edits
 from gn3.csvcmp import extract_strain_name
 
@@ -113,3 +114,21 @@ def test_extract_strain_name():
         extract_strain_name(csv_header="Strain Name,Value,SE,Count", data="BXD1,18,x,0")
         == "BXD1"
     )
+
+
+@pytest.mark.unit_test
+def test_get_allowable_csv_headers(mocker):
+    """Test that all the csv headers are fetched properly"""
+    mock_conn = mocker.MagicMock()
+    expected_values = [
+        "Strain Name", "Value", "SE", "Count",
+        "Condition", "Tissue", "Sex", "Age",
+        "Ethn.", "PMI (hrs)", "pH", "Color",
+    ]
+    with mock_conn.cursor() as cursor:
+        cursor.fetchall.return_value = (
+            ('Condition',), ('Tissue',), ('Sex',),
+            ('Age',), ('Ethn.',), ('PMI (hrs)',), ('pH',), ('Color',))
+        assert get_allowable_sampledata_headers(mock_conn) == expected_values
+        cursor.execute.assert_called_once_with(
+            "SELECT Name from CaseAttribute")