Use the already defined function to wait for generate

[pyi2ncommon] / test / test_arnied_wrapper.py

diff --git a/test/test_arnied_wrapper.py b/test/test_arnied_wrapper.py
index e061150..bce22a8 100755 (executable)
--- a/test/test_arnied_wrapper.py
+++ b/test/test_arnied_wrapper.py
@@ -125,7 +125,7 @@ class ArniedWrapperTest(unittest.TestCase):
 
     def test_accept_license(self):
         """Test accepting license."""
-        DummyCmdOutputMapping.asserted_cmds = self.cmd_db[1:3]
+        DummyCmdOutputMapping.asserted_cmds = self.cmd_db[1:3] + self.cmd_db[8:11]
         arnied_wrapper.accept_licence()
         # make sure an error is ignored since license might
         # already be accepted