def test__export_as_csv(self):
t = DbObject(DBH.test)
t.tbl_.truncate()
t.db_.commit()
t.company_id = COMPANY_ID
t.name = NAME
t.number = NUMBER
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
self.assertTrue(t.add()) # Add succeeds
output = cStringIO.StringIO()
t.export_as_csv(out=output)
# String fields are double quoted.
self.assertEqual(
"""%s,%s,%s,"%s",%s,"%s","%s","%s","%s"\r\n"""
% (t.id, COMPANY_ID, NUMBER, NAME, AMOUNT, START_DATE, STATUS, t.created_on, t.updated_on),
output.getvalue(),
)
t.tbl_.truncate()
t.db_.commit()
return
def test____deepcopy__(self):
t = DbObject(DBH.test)
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
t2 = copy.deepcopy(t)
self.assertEqual(t, t2)
return
def _obj(self, number=0):
if number == 1:
t = DbObject(DBH.test)
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
self.assertTrue(t.add()) # Add test object
return t
if number == 2:
t2 = DbObject(DBH.test)
t2.company_id = COMPANY_ID
t2.number = NUMBER_2
t2.name = NAME_2
t2.amount = AMOUNT_2
t2.start_date = START_DATE_2
t2.status = STATUS_2
self.assertTrue(t2.add()) # Add test object 2
return t2
def initialize():
t = DbObject(DBH.test)
t.tbl_.truncate()
t.db_.commit()
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
self.assertTrue(t.add())
return t
def test__as_list(self):
t = DbObject(DBH.test)
# Uninitialized returns list of default values
self.assertEqual(t.as_list()[:7], [None, None, None, None, D("0.00"), None, None])
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
# Initialized returns expected list
self.assertEqual(t.as_list()[:7], [None, COMPANY_ID, NUMBER, NAME, AMOUNT, START_DATE, STATUS])
return
def test__as_dict(self):
t = DbObject(DBH.test)
as_d = t.as_dict()
del as_d["created_on"]
del as_d["updated_on"]
self.assertEqual(
as_d,
{
"id": None,
"company_id": None,
"number": None,
"name": None,
"amount": D("0.00"),
"start_date": None,
"status": None,
},
)
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
as_d = t.as_dict()
del as_d["created_on"]
del as_d["updated_on"]
self.assertEqual(
as_d,
{
"id": None,
"company_id": COMPANY_ID,
"number": NUMBER,
"name": NAME,
"amount": AMOUNT,
"start_date": START_DATE,
"status": STATUS,
},
)
return
def test__update(self):
def initialize():
t = DbObject(DBH.test)
t.tbl_.truncate()
t.db_.commit()
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
self.assertTrue(t.add())
return t
records = DbObject(DBH.test).set_.get()
self.assertEqual(len(records), 0)
t = initialize()
records = DbObject(DBH.test).set_.get()
self.assertEqual(len(records), 1)
self.assertEqual(records[0], t)
t2 = DbObject(DBH.test)
t2.company_id = COMPANY_ID_2
t2.number = NUMBER_2
t2.name = NAME_2
t2.amount = AMOUNT_2
t2.start_date = START_DATE_2
t2.status = STATUS_2
# Record is not in database, should add
self.assertTrue(t2.update())
records = DbObject(DBH.test).set_.get()
self.assertEqual(len(records), 2)
self.assertNotEqual(t.id, t2.id)
# Record is in database, should modify
self.assertTrue(t2.update())
records = DbObject(DBH.test).set_.get()
self.assertEqual(len(records), 2)
self.assertNotEqual(t.id, t2.id)
# Update record with identical key fields, should modify t
t = initialize()
t2 = DbObject(DBH.test)
t2.company_id = COMPANY_ID
t2.number = NUMBER_2
t2.name = NAME
t2.amount = AMOUNT_2
t2.start_date = START_DATE_2
t2.status = STATUS_2
self.assertTrue(t2.update(key_fields=["company_id", "name"]))
tests = t.set_.get()
self.assertEqual(len(tests), 1)
self.assertEqual(t.id, t2.id)
# Update record with non-identical key fields but matching id, should
# modify.
t = initialize()
t2 = DbObject(DBH.test)
t2.id = t.id
t2.company_id = COMPANY_ID_2
t2.number = NUMBER_2
t2.name = NAME_2
t2.amount = AMOUNT_2
t2.start_date = START_DATE_2
t2.status = STATUS_2
self.assertTrue(t2.update(key_fields=["company_id", "name"]))
tests = t.set_.get()
self.assertEqual(len(tests), 1)
self.assertEqual(t.id, t2.id)
self.assertEqual(tests[0].company_id, t2.company_id)
# Update record with non-identical key fields, no id, should add
t = initialize()
t2 = DbObject(DBH.test)
t2.id = None
t2.company_id = COMPANY_ID_2
t2.number = NUMBER_2
t2.name = NAME_2
t2.amount = AMOUNT_2
t2.start_date = START_DATE_2
t2.status = STATUS_2
self.assertTrue(t2.update(key_fields=["company_id", "name"]))
tests = t.set_.get()
self.assertEqual(len(tests), 2)
self.assertNotEqual(t.id, t2.id)
# Test with a key field not defined.
t = initialize()
t2 = DbObject(DBH.test)
t2.id = None
t2.company_id = COMPANY_ID_2
self.assertTrue(t2.update(key_fields=["company_id", "name"]))
tests = t.set_.get()
self.assertEqual(len(tests), 2)
self.assertNotEqual(t.id, t2.id)
t3 = DbObject(DBH.test)
t3.id = None
t3.company_id = COMPANY_ID_2
self.assertTrue(t3.update(key_fields=["company_id", "name"]))
tests = t.set_.get()
self.assertEqual(len(tests), 2)
self.assertEqual(t2.id, t3.id)
# Make sure it distinguishes between NULL and empty string.
t = initialize()
t2 = DbObject(DBH.test)
t2.id = None
t2.company_id = COMPANY_ID_2
t2.name = ""
self.assertTrue(t2.update(key_fields=["company_id", "name"]))
t3 = DbObject(DBH.test)
t3.id = None
t3.company_id = COMPANY_ID_2
self.assertTrue(t3.update(key_fields=["company_id", "name"]))
tests = t.set_.get()
self.assertEqual(len(tests), 3)
self.assertNotEqual(t2.id, t3.id)
t.tbl_.truncate()
t.db_.commit()
return
def test__add(self):
t = DbObject(DBH.test)
t.tbl_.truncate()
t.db_.commit()
t.company_id = COMPANY_ID
t.number = NUMBER
t.name = NAME
t.amount = AMOUNT
t.start_date = START_DATE
t.status = STATUS
self.assertTrue(t.add()) # Add succeeds
self.assertTrue(self.i_re.match(str(t.id))) # Object id field is set
# Creation and modified dates should be set.
self.assertTrue(isinstance(t.created_on, datetime.datetime))
self.assertTrue(isinstance(t.updated_on, datetime.datetime))
rows = t.db_.executesql(
"""
SELECT
id,
company_id,
number,
name,
amount,
start_date,
status,
created_on,
updated_on
FROM test;
"""
)
self.assertEqual(rows[0][0], t.id) # Id matches
self.assertEqual(rows[0][1], COMPANY_ID) # company_id matches
self.assertEqual(rows[0][2], NUMBER) # number matches
self.assertEqual(rows[0][3], NAME) # name matches
self.assertEqual(rows[0][4], AMOUNT) # amount matches
self.assertEqual(rows[0][5], START_DATE) # start_date matches
self.assertEqual(rows[0][6], STATUS) # status matches
# Creation and modified dates should be set.
self.assertTrue(isinstance(rows[0][7], datetime.datetime))
self.assertTrue(isinstance(rows[0][8], datetime.datetime))
old_id = t.id
self.assertTrue(t.add()) # Successive add succeeds
self.assertTrue(t.id != old_id) # Object has new id
self.assertEqual(t.db_(t.tbl_.id > 0).count(), 2) # Now two records
t.tbl_.truncate()
t.db_.commit()
# Test MySQL column default values.
t2 = DbObject(DBH.test)
self.assertEqual(t2.status, None) # Status initially null.
t2.add()
rows = t2.db_.executesql(
"""
SELECT
id,
company_id,
number,
name,
amount,
start_date,
status,
created_on,
updated_on
FROM test;
"""
)
self.assertEqual(rows[0][0], t2.id) # Id matches
self.assertEqual(rows[0][6], "a") # status matches default
self.assertEqual(t2.status, "a") # Instance is updated.
t2.tbl_.truncate()
t2.db_.commit()
return
