feat: list command now handles number of DFD in output #29

threatcl · Feb 19, 2023 · f585d56 · f585d56
1 parent 08f20ee
commit f585d56
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/cmd/hcltm/list.go b/cmd/hcltm/list.go
@@ -192,11 +192,7 @@ func (c *ListCommand) Run(args []string) int {
 							bodyString = bodyString + "-"
 						}
 					case "Dfd":
-						if len(tm.DataFlowDiagrams) > 0 {
-							bodyString = bodyString + fmt.Sprintf("%t", true)
-						} else {
-							bodyString = bodyString + fmt.Sprintf("%t", false)
-						}
+						bodyString = bodyString + fmt.Sprintf("%d", len(tm.DataFlowDiagrams))
 					}
 				}
 

diff --git a/cmd/hcltm/list_test.go b/cmd/hcltm/list_test.go
@@ -160,15 +160,15 @@ func TestListRun(t *testing.T) {
 		{
 			"valid_dfd_exists",
 			"./testdata/tm3.hcl",
-			"true",
+			"1",
 			false,
 			0,
 			"-fields=dfd",
 		},
 		{
 			"no_valid_dfd_exists",
 			"./testdata/tm1.hcl",
-			"false",
+			"0",
 			false,
 			0,
 			"-fields=dfd",