klemek
32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
from unittest import TestCase
|
|
from meme_otron import utils
|
|
|
|
|
|
class TestUtilsLang(TestCase):
|
|
def test_find_nearest(self):
|
|
self.assertEqual("test", utils.find_nearest("tost", ["test", "example", "what"]))
|
|
self.assertIsNone(utils.find_nearest("unknown", ["test", "example", "what"], threshold=2))
|
|
self.assertEqual("test", utils.find_nearest("unknown", ["test", "example", "what"], threshold=200))
|
|
|
|
|
|
class TestUtilsWeb(TestCase):
|
|
def test_read_web_file(self):
|
|
out, err = utils.read_web_file("http:invalid.url")
|
|
self.assertIsNone(out)
|
|
self.assertEqual('Invalid URL', err)
|
|
out, err = utils.read_web_file("http://unknown.domain/")
|
|
self.assertIsNone(out)
|
|
self.assertEqual('Could not connect to server', err)
|
|
out, err = utils.read_web_file("http://httpbin.org/status/418")
|
|
self.assertIsNone(out)
|
|
self.assertEqual('Could not connect: HTTP Error 418: I\'M A TEAPOT', err)
|
|
out, err = utils.read_web_file("http://httpbin.org/bytes/1024", max_file_size=1000)
|
|
self.assertIsNone(out)
|
|
self.assertEqual('File too big', err)
|
|
# out, err = utils.read_web_file("http://httpbin.org/delay/1", timeout=0.1)
|
|
# self.assertIsNone(out)
|
|
# self.assertEqual('Could not connect to server', err)
|
|
out, err = utils.read_web_file("http://httpbin.org/base64/dGVzdA==")
|
|
self.assertIsNone(err)
|
|
self.assertEqual('test', out.decode("utf-8"))
|