From 174d6063857fa2fd2d651ae6932b969616e94da1 Mon Sep 17 00:00:00 2001
From: Scott Talbert <swt@techie.net>
Date: Sat, 1 Jan 2022 17:24:37 -0500
Subject: [PATCH] Add tests for TaskBarIcon CreatePopupMenu and GetPopupMenu
 overrides

---
 unittests/test_taskbar.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

diff --git a/unittests/test_taskbar.py b/unittests/test_taskbar.py
index 20751271..02c51e8a 100644
--- a/unittests/test_taskbar.py
+++ b/unittests/test_taskbar.py
@@ -36,6 +36,37 @@ class taskbar_Tests(wtc.WidgetTestCase):
         wx.adv.wxEVT_TASKBAR_BALLOON_TIMEOUT
         wx.adv.wxEVT_TASKBAR_BALLOON_CLICK
 
+
+    # Test that the CreatePopupMenu override works
+    def test_taskbar3(self):
+        class MyTaskBarIcon(wx.adv.TaskBarIcon):
+            def __init__(self):
+                super().__init__()
+                self.flag = False
+            def CreatePopupMenu(self):
+                self.flag = True
+        tbi = MyTaskBarIcon()
+        evt = wx.adv.TaskBarIconEvent(wx.adv.wxEVT_TASKBAR_RIGHT_DOWN, tbi)
+        wx.PostEvent(tbi, evt)
+        self.myYield()
+        self.assertTrue(tbi.flag)
+
+
+    # Test that the GetPopupMenu override works
+    def test_taskbar4(self):
+        class MyTaskBarIcon(wx.adv.TaskBarIcon):
+            def __init__(self):
+                super().__init__()
+                self.flag = False
+            def GetPopupMenu(self):
+                self.flag = True
+        tbi = MyTaskBarIcon()
+        evt = wx.adv.TaskBarIconEvent(wx.adv.wxEVT_TASKBAR_RIGHT_DOWN, tbi)
+        wx.PostEvent(tbi, evt)
+        self.myYield()
+        self.assertTrue(tbi.flag)
+
+
 #---------------------------------------------------------------------------
 
 if __name__ == '__main__':