From d2cd9ce383713babb26e4ce0bbcbe23723eac05a Mon Sep 17 00:00:00 2001
From: Christopher Dehner <christopher.dehner@outlook.com>
Date: Mon, 28 Aug 2023 15:31:42 -0400
Subject: [PATCH] add tests for VBA component files

---
 tests/test_closed.diff    | 33 +++++++++++++++++++++++++++++++++
 tests/test_new.diff       | 28 ++++++++++++++++++++++++++++
 tests/test_todo_parser.py |  8 +++++++-
 3 files changed, 68 insertions(+), 1 deletion(-)

diff --git a/tests/test_closed.diff b/tests/test_closed.diff
index d3ee0fb..fe7d092 100644
--- a/tests/test_closed.diff
+++ b/tests/test_closed.diff
@@ -359,3 +359,36 @@ index 0000000..a6c6cb0
 -<Label Text="ok"/>
 <Label Text="There you go"/>
 <Label Text="Thank you"/>
+diff --git a/tests/example.cls b/tests/example.cls
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.cls
++++ b/tests/example.cls
+@@ -1,2 +1,2 @@
+-' TODO: import library instead
+-Public Sub ReinventTheWheel()
++Dim helper as HelperLibrary
++Set helper = New HelperLibrary
+diff --git a/tests/example.bas b/tests/example.bas
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.bas
++++ b/tests/example.bas
+@@ -1,2 +1,1 @@
+-' TODO: simplify
+-Public Sub MakeThingsComplicated()
++Public Sub DoItSimpler()
+diff --git a/tests/example.frm b/tests/example.frm
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.frm
++++ b/tests/example.frm
+@@ -1,2 +1,1 @@
+-' TODO: remove feature to prevent legal issues
+-Public Sub Plagiarize()
++Public Sub MakeOriginalIdea()
+diff --git a/tests/example.sht b/tests/example.sht
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.sht
++++ b/tests/example.sht
+@@ -1,2 +1,1 @@
+-' TODO: switch to tea
+-Public Sub MakeCoffee()
++Public Sub MakeTea()
diff --git a/tests/test_new.diff b/tests/test_new.diff
index 4dd93f9..afed07c 100644
--- a/tests/test_new.diff
+++ b/tests/test_new.diff
@@ -403,3 +403,31 @@ index 0000000..a6c6cb0
 +<Label Text="why"/>
 +<!-- TODO: please -->
 +<Label Text="ok"/>
+diff --git a/tests/example.cls b/tests/example.cls
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.cls
++++ b/tests/example.cls
+@@ -1,1 +1,2 @@
++' TODO: import library instead
+ Public Sub ReinventTheWheel()
+diff --git a/tests/example.bas b/tests/example.bas
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.bas
++++ b/tests/example.bas
+@@ -1,1 +1,2 @@
++' TODO: simplify
+ Public Sub MakeThingsComplicated()
+diff --git a/tests/example.frm b/tests/example.frm
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.frm
++++ b/tests/example.frm
+@@ -1,1 +1,2 @@
++' TODO: remove feature to prevent legal issues
+ Public Sub Plagiarize()
+diff --git a/tests/example.sht b/tests/example.sht
+index 6b0c6cf..b37e70a 100644
+--- a/tests/example.sht
++++ b/tests/example.sht
+@@ -1,1 +1,2 @@
++' TODO: switch to tea
+ Public Sub MakeCoffee()
diff --git a/tests/test_todo_parser.py b/tests/test_todo_parser.py
index 0b0c1f8..045f95d 100644
--- a/tests/test_todo_parser.py
+++ b/tests/test_todo_parser.py
@@ -93,6 +93,9 @@ class NewIssueTests(unittest.TestCase):
     def test_xaml_issues(self):
         self.assertEqual(count_issues_for_file_type(self.raw_issues, 'xml'), 2)
 
+    def test_vba_issues(self):
+        self.asserEqual(count_issues_for_file_type(self.raw_issues, 'VBA'), 4)
+
 class ClosedIssueTests(unittest.TestCase):
     # Check for removed TODOs across the files specified.
     def setUp(self):
@@ -169,10 +172,13 @@ class ClosedIssueTests(unittest.TestCase):
 
     def test_nix_issues(self):
         self.assertEqual(count_issues_for_file_type(self.raw_issues, 'nix'), 2)
-        
+
     def test_xaml_issues(self):
         self.assertEqual(count_issues_for_file_type(self.raw_issues, 'xml'), 2)
 
+    def test_vba_issues(self):
+        self.assertEqual(count_issues_for_file_type(self.raw_issues, 'vba'), 4)
+
 class IgnorePatternTests(unittest.TestCase):
 
     def test_single_ignore(self):