1
0
Fork 0
mirror of https://gitlab.com/SIGBUS/nyaa.git synced 2024-10-31 23:25:58 +00:00
nyaa/tests/test_backend.py
2019-02-25 12:27:09 +01:00

57 lines
2 KiB
Python

import unittest
from nyaa import backend
class TestBackend(unittest.TestCase):
# def setUp(self):
# self.db, nyaa.app.config['DATABASE'] = tempfile.mkstemp()
# nyaa.app.config['TESTING'] = True
# self.app = nyaa.app.test_client()
# with nyaa.app.app_context():
# nyaa.db.create_all()
#
# def tearDown(self):
# os.close(self.db)
# os.unlink(nyaa.app.config['DATABASE'])
def test_replace_utf8_values(self):
test_dict = {
'hash': '2346ad27d7568ba9896f1b7da6b5991251debdf2',
'title.utf-8': '¡hola! ¿qué tal?',
'filelist.utf-8': [
'Español 101.mkv',
'ру́сский 202.mp4'
]
}
expected_dict = {
'hash': '2346ad27d7568ba9896f1b7da6b5991251debdf2',
'title': '¡hola! ¿qué tal?',
'filelist': [
'Español 101.mkv',
'ру́сский 202.mp4'
]
}
self.assertTrue(backend._replace_utf8_values(test_dict))
self.assertDictEqual(test_dict, expected_dict)
def test_replace_invalid_xml_chars(self):
self.assertEqual(backend.sanitize_string('ayy\x08lmao'), 'ayy\uFFFDlmao')
self.assertEqual(backend.sanitize_string('ayy\x0clmao'), 'ayy\uFFFDlmao')
self.assertEqual(backend.sanitize_string('ayy\uD8FFlmao'), 'ayy\uFFFDlmao')
self.assertEqual(backend.sanitize_string('ayy\uFFFElmao'), 'ayy\uFFFDlmao')
self.assertEqual(backend.sanitize_string('\x08ayy\x0clmao'), '\uFFFDayy\uFFFDlmao')
self.assertEqual(backend.sanitize_string('ayy\x08\x0clmao'), 'ayy\uFFFD\uFFFDlmao')
self.assertEqual(backend.sanitize_string('ayy\x08\x08lmao'), 'ayy\uFFFD\uFFFDlmao')
self.assertEqual(backend.sanitize_string('ぼくのぴこ'), 'ぼくのぴこ')
@unittest.skip('Not yet implemented')
def test_handle_torrent_upload(self):
pass
if __name__ == '__main__':
unittest.main()