diff options
Diffstat (limited to 'wrappers/python/tests.py')
| -rw-r--r-- | wrappers/python/tests.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/wrappers/python/tests.py b/wrappers/python/tests.py index cbec370..5cf5c25 100644 --- a/wrappers/python/tests.py +++ b/wrappers/python/tests.py @@ -2,6 +2,7 @@ import unittest import eduvpn_common.main as eduvpn +from eduvpn_common.state import State, StateType import webbrowser import sys import os @@ -20,7 +21,7 @@ class ConfigTests(unittest.TestCase): # This can throw an exception _eduvpn.register() - @_eduvpn.event.on("OAuth_Started", eduvpn.StateType.Enter) + @_eduvpn.event.on(State.OAUTH_STARTED, StateType.Enter) def oauth_initialized(old_state, url): login_eduvpn(url) |