tests/init_provider.py

   1 import unittest
   2 from os import path
   3
   4 from manga_py import fs
   5 from manga_py.provider import Provider
   6 from manga_py.providers import get_provider
   7
   8 root_path = path.dirname(path.realpath(__file__))
   9
  10
  11 class TestInitProvider(unittest.TestCase):
  12
  13     # success
  14     def test_get_provider1(self):
  15         provider = get_provider('http://readmanga.live/manga/name/here')
  16         self.assertIsInstance(provider(), Provider)
  17
  18     # failed
  19     def test_get_provider2(self):
  20         provider = get_provider('http://example.org/manga/name/here')
  21         self.assertFalse(provider)
  22
  23     def test_root_path(self):
  24         self.assertEqual(path.realpath(fs.path_join(root_path, '..')), fs.root_path())
  25
  26     def test_file_name_query_remove1(self):
  27         name = '/addr/to/filename'
  28         self.assertEqual(
  29             name,
  30             fs.remove_file_query_params(name + '?query=params').replace('\\', '/')  # windows os patch
  31         )
  32
  33     def test_file_name_query_remove2(self):
  34         name = '/addr/to/filename/'
  35         self.assertEqual(
  36             name + 'image.png',
  37             fs.remove_file_query_params(name + '?query=params').replace('\\', '/')  # windows os patch
  38         )