tests: excel: bigger coverage

vdjagilev · Feb 23, 2024 · 2da6952 · 2da6952
1 parent b428bf8
commit 2da6952
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 3 deletions.
diff --git a/formatter/formatter_excel.go b/formatter/formatter_excel.go
@@ -30,7 +30,6 @@ func (cd *CellData) writeCell(cell string, value string) error {
 
 // Format the data to Excel and output it to an Excel file
 func (f *ExcelFormatter) Format(td *TemplateData, templateContent string) (err error) {
-	// TODO: Write unit tests
 	file := excelize.NewFile()
 	sheetName := "Sheet1"
 

diff --git a/formatter/formatter_excel_test.go b/formatter/formatter_excel_test.go
@@ -56,13 +56,39 @@ func TestExcelFormatter_Format(t *testing.T) {
 							{
 								StartTime: 0,
 								EndTime:   0,
-								Port:      []Port{},
+								Port: []Port{
+									{
+										PortID:   80,
+										Protocol: "tcp",
+										Service: PortService{
+											Name: "http",
+										},
+									},
+									{
+										PortID:   443,
+										Protocol: "tcp",
+										Service: PortService{
+											Name: "https",
+										},
+									},
+								},
 								HostAddress: []HostAddress{
 									{
 										Address: "192.168.1.1",
 									},
 								},
-								HostNames: HostNames{},
+								HostNames: HostNames{
+									HostName: []HostName{
+										{
+											Name: "example.com",
+											Type: "",
+										},
+										{
+											Name: "www.example.com",
+											Type: "",
+										},
+									},
+								},
 								Status: HostStatus{
 									State: "up",
 								},