68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
""" Unit tests for contribs """
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
import os
|
||
|
|
|
||
|
|
from gluon._compat import to_bytes
|
||
|
|
from gluon.storage import Storage
|
||
|
|
from gluon.contrib import fpdf as fpdf
|
||
|
|
from gluon.contrib import pyfpdf as pyfpdf
|
||
|
|
from gluon.contrib.appconfig import AppConfig
|
||
|
|
|
||
|
|
|
||
|
|
def setUpModule():
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
def tearDownModule():
|
||
|
|
if os.path.isfile('appconfig.json'):
|
||
|
|
os.unlink('appconfig.json')
|
||
|
|
|
||
|
|
|
||
|
|
class TestContribs(unittest.TestCase):
|
||
|
|
""" Tests the contrib package """
|
||
|
|
|
||
|
|
def test_fpdf(self):
|
||
|
|
""" Basic PDF test and sanity checks """
|
||
|
|
|
||
|
|
self.assertEqual(
|
||
|
|
fpdf.FPDF_VERSION, pyfpdf.FPDF_VERSION, 'version mistmatch')
|
||
|
|
self.assertEqual(fpdf.FPDF, pyfpdf.FPDF, 'class mistmatch')
|
||
|
|
|
||
|
|
pdf = fpdf.FPDF()
|
||
|
|
pdf.add_page()
|
||
|
|
pdf.compress = False
|
||
|
|
pdf.set_font('Arial', '', 14)
|
||
|
|
pdf.ln(10)
|
||
|
|
pdf.write(5, 'hello world')
|
||
|
|
pdf_out = pdf.output('', 'S')
|
||
|
|
|
||
|
|
self.assertTrue(to_bytes(fpdf.FPDF_VERSION) in pdf_out, 'version string')
|
||
|
|
self.assertTrue(to_bytes('hello world') in pdf_out, 'sample message')
|
||
|
|
|
||
|
|
def test_appconfig(self):
|
||
|
|
"""
|
||
|
|
Test for the appconfig module
|
||
|
|
"""
|
||
|
|
from gluon import current
|
||
|
|
s = Storage({'application': 'admin',
|
||
|
|
'folder': 'applications/admin'})
|
||
|
|
current.request = s
|
||
|
|
simple_config = '{"config1" : "abc", "config2" : "bcd", "config3" : { "key1" : 1, "key2" : 2} }'
|
||
|
|
with open('appconfig.json', 'w') as g:
|
||
|
|
g.write(simple_config)
|
||
|
|
myappconfig = AppConfig('appconfig.json')
|
||
|
|
self.assertEqual(myappconfig['config1'], 'abc')
|
||
|
|
self.assertEqual(myappconfig['config2'], 'bcd')
|
||
|
|
self.assertEqual(myappconfig.take('config1'), 'abc')
|
||
|
|
self.assertEqual(myappconfig.take('config3.key1', cast=str), '1')
|
||
|
|
# once parsed, can't be casted to other types
|
||
|
|
self.assertEqual(myappconfig.take('config3.key1', cast=int), '1')
|
||
|
|
|
||
|
|
self.assertEqual(myappconfig.take('config3.key2'), 2)
|
||
|
|
|
||
|
|
current.request = {}
|