def test_init_partial(self):
doc = Document(id="test", content_type="test type")
self.assertEqual(doc.id, "test")
self.assertEqual(doc.content_type, "test type")
self.assertTrue(hasattr(doc, "content"))
self.assertTrue(hasattr(doc, "etag"))
self.assertTrue(hasattr(doc, "timestamp"))
def test_init_arg_exception_obj(self):
class Tester(object):
def __init__(self, id=None, bad_test="test"):
self.id = id
self.bad_test = bad_test
obj = Tester()
with self.assertRaises(AttributeError):
Document(obj)
def test_init_empty(self):
doc = Document()
self.assertIsInstance(doc, Document)
self.assertTrue(hasattr(doc, "id"))
self.assertIsNone(doc.id)
self.assertTrue(hasattr(doc, "content_type"))
self.assertIsNone(doc.content_type)
self.assertTrue(hasattr(doc, "content"))
self.assertIsNone(doc.content)
self.assertTrue(hasattr(doc, "etag"))
self.assertIsNone(doc.etag)
self.assertTrue(hasattr(doc, "timestamp"))
self.assertIsNone(doc.timestamp)
def test_init_all(self):
doc = Document(id="test",
content_type="test type",
content=bytearray("test bytearray", "utf-8"),
etag="test etag",
timestamp="2014-06-23T15:25:00-05:00")
self.assertEqual(doc.id, "test")
self.assertEqual(doc.content_type, "test type")
self.assertEqual(doc.content, bytearray("test bytearray", "utf-8"))
self.assertEqual(doc.etag, "test etag")
central = pytz.timezone("US/Central") # UTC -0500
dt = central.localize(datetime.datetime(2014, 6, 23, 15, 25))
self.assertEqual(doc.timestamp, dt)
def test_setters_none(self):
doc = Document()
doc.id = None
doc.content_type = None
doc.content = None
doc.etag = None
doc.timestamp = None
self.assertIsNone(doc.id)
self.assertIsNone(doc.content_type)
self.assertIsNone(doc.content)
self.assertIsNone(doc.etag)
self.assertIsNone(doc.timestamp)
def test_setters_none(self):
doc = Document()
doc.id = None
doc.content_type = None
doc.content = None
doc.etag = None
doc.timestamp = None
self.assertIsNone(doc.id)
self.assertIsNone(doc.content_type)
self.assertIsNone(doc.content)
self.assertIsNone(doc.etag)
self.assertIsNone(doc.timestamp)
def test_setters(self):
doc = Document()
doc.id = "test"
doc.content_type = "test type"
doc.content = bytearray("test bytearray", "utf-8")
doc.etag = "test etag"
doc.timestamp = "2014-06-23T15:25:00-05:00"
self.assertEqual(doc.id, "test")
self.assertEqual(doc.content_type, "test type")
self.assertEqual(doc.content, bytearray("test bytearray", "utf-8"))
self.assertEqual(doc.etag, "test etag")
central = pytz.timezone("US/Central") # UTC -0500
dt = central.localize(datetime.datetime(2014, 6, 23, 15, 25))
self.assertEqual(doc.timestamp, dt)
def test_init_arg_exception_dict(self):
d = {"bad_test": "test", "id": "ok"}
with self.assertRaises(AttributeError):
Document(d)
def test_init_kwarg_exception(self):
with self.assertRaises(AttributeError):
Document(bad_test="test")
def test_content_timestamp(self):
doc = Document()
dt = pytz.utc.localize(datetime.datetime.utcnow())
doc.timestamp = dt
self.assertEqual(doc.timestamp, dt)
def test_content_setter(self):
doc = Document()
doc.content = "test bytearray"
self.assertEqual(doc.content, bytearray("test bytearray", "utf-8"))
def test_content_timestamp(self):
doc = Document()
dt = pytz.utc.localize(datetime.datetime.utcnow())
doc.timestamp = dt
self.assertEqual(doc.timestamp, dt)
def test_content_setter(self):
doc = Document()
doc.content = "test bytearray"
self.assertEqual(doc.content, bytearray("test bytearray", "utf-8"))
