fix unit tests

add-ons · Nov 14, 2019 · 0aa6f56 · 0aa6f56
1 parent bbfeba6
commit 0aa6f56
Show file tree

Hide file tree

Showing 8 changed files with 16 additions and 8 deletions.
diff --git a/test/test_apihelper.py b/test/test_apihelper.py
@@ -5,10 +5,10 @@
 
 from __future__ import absolute_import, division, print_function, unicode_literals
 import unittest
-from addon import kodi
 from apihelper import ApiHelper
 from data import CHANNELS
 from favorites import Favorites
+from kodiwrapper import KodiWrapper
 from resumepoints import ResumePoints
 from xbmcextra import kodi_to_ansi
 
@@ -21,6 +21,7 @@
 
 class ApiHelperTests(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _favorites = Favorites(kodi)
     _resumepoints = ResumePoints(kodi)
     _apihelper = ApiHelper(kodi, _favorites, _resumepoints)

diff --git a/test/test_favorites.py b/test/test_favorites.py
@@ -7,9 +7,9 @@
 from __future__ import absolute_import, division, print_function, unicode_literals
 import unittest
 from random import shuffle
-from addon import kodi
 from apihelper import ApiHelper
 from favorites import Favorites
+from kodiwrapper import KodiWrapper
 from resumepoints import ResumePoints
 
 xbmc = __import__('xbmc')
@@ -24,6 +24,7 @@
 
 class TestFavorites(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _favorites = Favorites(kodi)
     _resumepoints = ResumePoints(kodi)
     _apihelper = ApiHelper(kodi, _favorites, _resumepoints)

diff --git a/test/test_resumepoints.py b/test/test_resumepoints.py
@@ -6,9 +6,9 @@
 
 from __future__ import absolute_import, division, print_function, unicode_literals
 import unittest
-from addon import kodi
 from apihelper import ApiHelper
 from favorites import Favorites
+from kodiwrapper import KodiWrapper
 from resumepoints import ResumePoints
 
 xbmc = __import__('xbmc')
@@ -23,6 +23,7 @@
 
 class TestResumePoints(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _favorites = Favorites(kodi)
     _resumepoints = ResumePoints(kodi)
     _apihelper = ApiHelper(kodi, _favorites, _resumepoints)

diff --git a/test/test_search.py b/test/test_search.py
@@ -6,9 +6,9 @@
 
 from __future__ import absolute_import, division, print_function, unicode_literals
 import unittest
-from addon import kodi
 from apihelper import ApiHelper
 from favorites import Favorites
+from kodiwrapper import KodiWrapper
 from resumepoints import ResumePoints
 
 xbmc = __import__('xbmc')
@@ -20,6 +20,7 @@
 
 class TestSearch(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _favorites = Favorites(kodi)
     _resumepoints = ResumePoints(kodi)
     _apihelper = ApiHelper(kodi, _favorites, _resumepoints)

diff --git a/test/test_streamservice.py b/test/test_streamservice.py
@@ -14,8 +14,8 @@
 except ImportError:
     from urllib2 import HTTPError
 
-from addon import kodi
 from data import CHANNELS
+from kodiwrapper import KodiWrapper
 from streamservice import StreamService
 from tokenresolver import TokenResolver
 
@@ -34,6 +34,7 @@
 
 class StreamServiceTests(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _tokenresolver = TokenResolver(kodi)
     _streamservice = StreamService(kodi, _tokenresolver)
 

diff --git a/test/test_tokenresolver.py b/test/test_tokenresolver.py
@@ -5,7 +5,7 @@
 
 from __future__ import absolute_import, division, print_function, unicode_literals
 import unittest
-from addon import kodi
+from kodiwrapper import KodiWrapper
 from tokenresolver import TokenResolver
 
 xbmc = __import__('xbmc')
@@ -17,6 +17,7 @@
 
 class TokenResolverTests(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _tokenresolver = TokenResolver(kodi)
 
     def test_refresh_login(self):

diff --git a/test/test_tvguide.py b/test/test_tvguide.py
@@ -11,7 +11,7 @@
 import dateutil.tz
 from xbmcextra import kodi_to_ansi, uri_to_path
 
-from addon import kodi
+from kodiwrapper import KodiWrapper
 from tvguide import TVGuide
 
 xbmc = __import__('xbmc')
@@ -25,6 +25,7 @@
 
 class TestTVGuide(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _tvguide = TVGuide(kodi)
 
     def test_tvguide_date_menu(self):

diff --git a/test/test_vrtplayer.py b/test/test_vrtplayer.py
@@ -6,10 +6,10 @@
 from __future__ import absolute_import, division, print_function, unicode_literals
 import random
 import unittest
-from addon import kodi
 from apihelper import ApiHelper
 from data import CATEGORIES
 from favorites import Favorites
+from kodiwrapper import KodiWrapper
 from resumepoints import ResumePoints
 from vrtplayer import VRTPlayer
 
@@ -22,6 +22,7 @@
 
 class TestVRTPlayer(unittest.TestCase):
 
+    kodi = KodiWrapper(globals())
     _favorites = Favorites(kodi)
     _resumepoints = ResumePoints(kodi)
     _apihelper = ApiHelper(kodi, _favorites, _resumepoints)