From fec6ea1eba9cee325bbd9d82aa71b8ebf5ef90b0 Mon Sep 17 00:00:00 2001
From: jwijenbergh <jeroenwijenbergh@protonmail.com>
Date: Mon, 15 Aug 2022 14:41:28 +0200
Subject: Refactor: Use constants for state callbacks instead of strings

---
 wrappers/python/tests.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'wrappers/python/tests.py')

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)
 
-- 
cgit v1.2.3