mirror of
https://github.com/scrapy/scrapy.git
synced 2025-02-22 15:03:46 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
import json
|
|
import unittest
|
|
import datetime
|
|
from decimal import Decimal
|
|
|
|
from twisted.internet import defer
|
|
|
|
from scrapy.utils.serialize import ScrapyJSONEncoder
|
|
from scrapy.http import Request, Response
|
|
|
|
|
|
class JsonEncoderTestCase(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.encoder = ScrapyJSONEncoder()
|
|
|
|
def test_encode_decode(self):
|
|
dt = datetime.datetime(2010, 1, 2, 10, 11, 12)
|
|
dts = "2010-01-02 10:11:12"
|
|
d = datetime.date(2010, 1, 2)
|
|
ds = "2010-01-02"
|
|
t = datetime.time(10, 11, 12)
|
|
ts = "10:11:12"
|
|
dec = Decimal("1000.12")
|
|
decs = "1000.12"
|
|
s = {'foo'}
|
|
ss = ['foo']
|
|
dt_set = {dt}
|
|
dt_sets = [dts]
|
|
|
|
for input, output in [('foo', 'foo'), (d, ds), (t, ts), (dt, dts),
|
|
(dec, decs), (['foo', d], ['foo', ds]), (s, ss),
|
|
(dt_set, dt_sets)]:
|
|
self.assertEqual(self.encoder.encode(input), json.dumps(output))
|
|
|
|
def test_encode_deferred(self):
|
|
self.assertIn('Deferred', self.encoder.encode(defer.Deferred()))
|
|
|
|
def test_encode_request(self):
|
|
r = Request("http://www.example.com/lala")
|
|
rs = self.encoder.encode(r)
|
|
self.assertIn(r.method, rs)
|
|
self.assertIn(r.url, rs)
|
|
|
|
def test_encode_response(self):
|
|
r = Response("http://www.example.com/lala")
|
|
rs = self.encoder.encode(r)
|
|
self.assertIn(r.url, rs)
|
|
self.assertIn(str(r.status), rs)
|