1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
"""Tests for qr_code application."""
import os
from decimal import Decimal
from django.test import SimpleTestCase
from pydantic import ValidationError
from qr_code.qrcode.constants import (
DEFAULT_IMAGE_FORMAT,
DEFAULT_MODULE_SIZE,
DEFAULT_ERROR_CORRECTION,
DEFAULT_VERSION,
DEFAULT_ECI,
DEFAULT_BOOST_ERROR,
DEFAULT_ENCODING,
)
from qr_code.qrcode.serve import make_qr_code_url
from qr_code.qrcode.utils import QRCodeOptions
from qr_code.tests import TEST_TEXT, PNG_REF_SUFFIX, SVG_REF_SUFFIX
from qr_code.tests.utils import write_svg_content_to_file, get_resources_path, write_png_content_to_file
class TestApps(SimpleTestCase):
def test_apps_attributes(self):
from qr_code.apps import QrCodeConfig
self.assertEqual(QrCodeConfig.name, "qr_code")
self.assertEqual(QrCodeConfig.verbose_name, "Django QR Code")
class TestQRCodeOptions(SimpleTestCase):
def test_qr_code_options(self):
with self.assertRaises((TypeError, ValidationError)):
QRCodeOptions(foo="bar")
options = QRCodeOptions()
self.assertEqual(options.border, 4)
self.assertEqual(options.size, DEFAULT_MODULE_SIZE)
self.assertEqual(options.image_format, DEFAULT_IMAGE_FORMAT)
self.assertEqual(options.version, DEFAULT_VERSION)
self.assertEqual(options.error_correction, DEFAULT_ERROR_CORRECTION)
self.assertEqual(options.eci, DEFAULT_ECI)
self.assertEqual(options.boost_error, DEFAULT_BOOST_ERROR)
self.assertEqual(options.encoding.lower(), DEFAULT_ENCODING)
options = QRCodeOptions(image_format="invalid-image-format")
self.assertEqual(options.image_format, DEFAULT_IMAGE_FORMAT)
def test_kw_save(self):
options = QRCodeOptions(border=0, image_format="png", size=13)
self.assertDictEqual(options.kw_save(), {"border": 0, "dark": "black", "kind": "png", "light": "white", "scale": 13})
options = QRCodeOptions(border=0, image_format="svg", size=13)
self.assertDictEqual(
options.kw_save(), {"border": 0, "dark": "black", "kind": "svg", "light": "white", "scale": Decimal("1.3"), "unit": "mm"}
)
class TestWriteResourceData(SimpleTestCase):
resource_file_base_name = "TestWriteResourceData"
def test_write_svg(self):
response = self.client.get(make_qr_code_url(TEST_TEXT))
image_data = response.content.decode("utf-8")
write_svg_content_to_file(TestWriteResourceData.resource_file_base_name, image_data)
file_path_to_remove = os.path.join(get_resources_path(), TestWriteResourceData.resource_file_base_name + SVG_REF_SUFFIX)
os.remove(file_path_to_remove)
def test_write_png(self):
response = self.client.get(make_qr_code_url(TEST_TEXT))
image_data = response.content
write_png_content_to_file(TestWriteResourceData.resource_file_base_name, image_data)
file_path_to_remove = os.path.join(get_resources_path(), TestWriteResourceData.resource_file_base_name + PNG_REF_SUFFIX)
os.remove(file_path_to_remove)
|
