diff options
| author | jwijenbergh <jeroenwijenbergh@protonmail.com> | 2022-08-15 14:41:28 +0200 |
|---|---|---|
| committer | jwijenbergh <jeroenwijenbergh@protonmail.com> | 2022-08-15 14:41:28 +0200 |
| commit | fec6ea1eba9cee325bbd9d82aa71b8ebf5ef90b0 (patch) | |
| tree | dec0894380a2de8721a1e7733d75053bcc5181fc /wrappers/python/tests.py | |
| parent | c5e85ba79d4d091af9873f1fb0e7415c3b17b9f8 (diff) | |
Refactor: Use constants for state callbacks instead of strings
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) |