for testing and deploying your application
for finding and fixing issues
for empowering human code reviews
#!/usr/bin/python
# -*- coding: utf-8 -*-
import os
import shutil
import sys
import unittest
from uuid import uuid4
from PIL import Image
from saucenao import run_application
class TestInit(unittest.TestCase):
SAUCENAO_MIN_WIDTH = 3
SAUCENAO_MIN_HEIGHT = 3
def setUp(self):
"""Constructor for unittest classes
:return:
"""
directory = str(uuid4())
os.mkdir(directory)
self.directory = os.path.abspath(directory)
self.generate_small_jpg()
def tearDown(self):
"""Destructor for unittest classes
shutil.rmtree(self.directory)
def generate_small_jpg(self):
"""Generate a rather small jpg file to upload faster(631 bytes)
file_path = str(uuid4()) + ".jpg"
im = Image.new("RGB", (self.SAUCENAO_MIN_WIDTH, self.SAUCENAO_MIN_HEIGHT))
im.save(os.path.join(self.directory, file_path), "JPEG")
return file_path
def test_run_application(self):
"""Test the run_application function
sys.argv = [sys.argv[0], '-d', self.directory]
run_application()
if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromTestCase(TestInit)
unittest.TextTestRunner(verbosity=2).run(suite)
DaRealFreak /
saucenao
