mirror of
https://github.com/scrapy/scrapy.git
synced 2025-02-24 20:43:50 +00:00
45 lines
1.3 KiB
Python
45 lines
1.3 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"
|
|
|
|
for input, output in [('foo', 'foo'), (d, ds), (t, ts), (dt, dts),
|
|
(dec, decs), (['foo', d], ['foo', ds])]:
|
|
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)
|