Fixed generation for enum arguments with default values when avoiding the protected hack.

Also added an unit test.

2 files changed, 5 insertions, 1 deletions

diff --git a/tests/samplebinding/enum_test.py b/tests/samplebinding/enum_test.py
index 7d95f669c..6199a9b78 100644
--- a/tests/samplebinding/enum_test.py
+++ b/tests/samplebinding/enum_test.py
@@ -121,6 +121,11 @@ class EnumTest(unittest.TestCase):
         self.assertEqual(text, str(Event.ANY_EVENT))
         self.assertEqual(text, repr(Event.ANY_EVENT))
 
+    def testEnumArgumentWithDefaultValue(self):
+        '''Option enumArgumentWithDefaultValue(Option opt = UnixTime);'''
+        self.assertEqual(SampleNamespace.enumArgumentWithDefaultValue(), SampleNamespace.UnixTime)
+        self.assertEqual(SampleNamespace.enumArgumentWithDefaultValue(SampleNamespace.RandomNumber), SampleNamespace.RandomNumber)
+
 
 class MyEvent(Event):
     def __init__(self):
diff --git a/tests/samplebinding/sample_test.py b/tests/samplebinding/sample_test.py
index 16e652667..cc11dd552 100644
--- a/tests/samplebinding/sample_test.py
+++ b/tests/samplebinding/sample_test.py
@@ -26,7 +26,6 @@
 
 '''Test cases for libsample bindings module'''
 
-import sys
 import unittest
 
 import sample