Skip to content

Commit

Permalink
Update TestODL2Importer
Browse files Browse the repository at this point in the history
  • Loading branch information
vbessonov committed Aug 23, 2021
1 parent 06d508a commit f778dc8
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion core
6 changes: 3 additions & 3 deletions tests/test_odl2.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,11 +27,11 @@
)
from core.model.configuration import ConfigurationFactory, ConfigurationStorage
from core.opds2_import import RWPMManifestParser
from core.tests.test_opds2_import import TestOPDS2Importer
from core.tests.test_opds2_import import OPDS2Test
from tests.test_odl import TestODLExpiredItemsReaper


class TestODL2Importer(TestOPDS2Importer):
class TestODL2Importer(OPDS2Test):
@staticmethod
def _get_delivery_mechanism_by_drm_scheme_and_content_type(
delivery_mechanisms, content_type, drm_scheme
Expand Down Expand Up @@ -61,7 +61,7 @@ def _get_delivery_mechanism_by_drm_scheme_and_content_type(

return None

def sample_opds(self, filename):
def sample_opds(self, filename, file_type="r"):
base_path = os.path.split(__file__)[0]
resource_path = os.path.join(base_path, "files", "odl2")
return open(os.path.join(resource_path, filename)).read()
Expand Down

0 comments on commit f778dc8

Please sign in to comment.