booxtream/tests.py at master · Gluejar/booxtream · GitHub

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
73
74
import unittest
import time
import urllib2
from tempfile import NamedTemporaryFile
from StringIO import StringIO
from django.conf import settings

class TestBooXtream(unittest.TestCase):
    def setUp(self):
        # get a small epub test file as a file-like object
        self.epub2file = NamedTemporaryFile(delete=False)
        test_file_content = urllib2.urlopen('http://www.hxa.name/articles/content/EpubGuide-hxa7241.epub')
        self.epub2file.write(test_file_content.read())
        self.epub2file.seek(0)
        self.textfile = NamedTemporaryFile(delete=False)
        self.textfile.write("bad text file")
        self.textfile.seek(0)


    def _makeOne(self):
        from . import BooXtream
        manager = BooXtream()
        return manager

    def test_booxtream_errors(self):
        if settings.LOCAL_TEST:
            return
        from .exceptions import BooXtreamError
        inst = self._makeOne()
        if not settings.BOOXTREAM_API_KEY:
            return
        with self.assertRaises(BooXtreamError) as cm:
            inst.platform()
        self.assertIn('expirydays not set', str(cm.exception))
        params = {
            'customername': 'Jane Test',
            'languagecode':'1043',
            'expirydays': 1,
            'downloadlimit': 3,
            'exlibris':1,
            'chapterfooter':1,
            'disclaimer':1,
            'referenceid':'bad_file_check'
            }
        with self.assertRaises(BooXtreamError) as cm:
            inst.platform(epubfile=self.textfile, **params)


    def test_booxtream_good(self):
        inst = self._makeOne()
        params = {
            'customeremailaddress':'jane@example.com',
            'customername': 'Jane Test',
            'languagecode':'1043',
            'expirydays': 1,
            'downloadlimit': 3,
            'exlibris':1,
            'chapterfooter':1,
            'disclaimer':1,
            }
        params['referenceid'] = 'order' + str(time.time())
        boox = inst.platform(epubfile=self.epub2file, **params)
        self.assertRegexpMatches(boox.download_link_epub, 'download.booxtream.com/')
        self.assertFalse(boox.expired)
        self.assertEqual(boox.downloads_remaining, 3)

        # make sure it works with an in-memory file
        self.epub2file.seek(0)
        in_mem_epub = StringIO()
        in_mem_epub.write(self.epub2file.read())
        in_mem_epub.seek(0)
        boox2 = inst.platform(epubfile=in_mem_epub, **params)
        self.assertRegexpMatches(boox2.download_link_epub, 'download.booxtream.com/')