From 61bcc6d22dbe6b40b3348df36578b6de3693e7d9 Mon Sep 17 00:00:00 2001
From: Andrii Yurchuk <ay@mntw.re>
Date: Tue, 16 Mar 2021 11:57:41 +0100
Subject: [PATCH] Fix crash when there is no header and no data (#78)

---
 src/utils.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/utils.rs b/src/utils.rs
index 5bd3f74..1d40e8e 100644
--- a/src/utils.rs
+++ b/src/utils.rs
@@ -50,8 +50,9 @@ pub fn create_list_table<T: Entity>(
             }
         }
     }
-    let id_column = table.get_column_mut(0).unwrap();
-    id_column.set_constraint(ColumnConstraint::MinWidth(ID_COLUMN_WIDTH));
+    if let Some(id_column) = table.get_column_mut(0) {
+        id_column.set_constraint(ColumnConstraint::MinWidth(ID_COLUMN_WIDTH));
+    }
     table
 }
 
@@ -555,6 +556,15 @@ mod tests {
         );
     }
 
+    #[test]
+    fn test_create_list_table_no_header_no_data() {
+        let data: Vec<Patient> = vec![];
+        assert_eq!(
+            format_table(create_list_table(data, None, &PATIENTS_LIST_DICOM_TAGS)),
+            ""
+        );
+    }
+
     #[test]
     fn test_create_show_table_patient() {
         let patient = Patient {