def test_no_future_elements_created(self):
# current year
period = TinyDbPeriod()
elements = period.get_entries()
self.assertEqual(len(elements[DEFAULT_TABLE]), 0)
self.assertEqual(len(elements["recurrent"]), 0)
entry_id = period.add_entry(table_name="recurrent",
name="lunch",
value=-5,
frequency="daily",
start="01-01")
day_nr = dt.date.today().timetuple().tm_yday
elements = period.get_entries()
self.assertEqual(len(elements["recurrent"][entry_id]), day_nr)
class TinyDbPeriodStandardEntryTestCase(unittest.TestCase):
def setUp(self):
self.period = TinyDbPeriod(name=1901)
self.eid = self.period.add_entry(name="Bicycle",
value=-999.99,
date="1901-01-01")
def test_get_entries(self):
entries = self.period.get_entries(filters={"date": "01-"})
self.assertEqual("bicycle", entries[DEFAULT_TABLE][1]["name"])
def test_remove_entry(self):
response = self.period.remove_entry(eid=1)
self.assertEqual(0, len(self.period._db))
self.assertEqual(1, response)
def test_create_models_query_kwargs(self):
eid = self.period.add_entry(name="Xmas gifts",
value=500,
date="1901-12-23",
category="gifts")
standard_elements = self.period.get_entries(
filters={"date": "12"})[DEFAULT_TABLE]
self.assertEqual(len(standard_elements), 1)
self.assertEqual(standard_elements[eid]["name"], "xmas gifts")
standard_elements = self.period.get_entries(
filters={"category": None})[DEFAULT_TABLE]
self.assertEqual(len(standard_elements), 1)
standard_elements = self.period.get_entries(
filters={"category": "gi"})[DEFAULT_TABLE]
self.assertEqual(len(standard_elements), 1)
self.period.add_entry(name="hammer", value=-33, date="1901-12-20")
standard_elements = self.period.get_entries(filters={
"name": "xmas",
"date": "12"
})[DEFAULT_TABLE]
self.assertEqual(len(standard_elements), 1)
self.assertEqual(standard_elements[eid]["name"], "xmas gifts")
def test_category_cache(self):
self.period.add_entry(name="walmart",
value=-50.01,
category="groceries",
date="1901-02-02")
self.period.add_entry(name="walmart", value=-0.99, date="1901-02-03")
groceries_elements = self.period.get_entries(
filters={"category": "groceries"})
self.assertEqual(len(groceries_elements), 2)
self.assertEqual(
sum([
e["value"] for e in groceries_elements[DEFAULT_TABLE].values()
]), -51)
def test_remove_nonexisting_entry(self):
self.assertRaises(PeriodException, self.period.remove_entry, eid=0)
def test_add_rm_via_eid(self):
entry_name = "penguin sale"
entry_id = self.period.add_entry(name=entry_name,
value=1337,
date="1901-12-01")
nr_entries = len(self.period._db)
removed_entry_id = self.period.remove_entry(eid=entry_id)
self.assertEqual(removed_entry_id, entry_id)
self.assertEqual(len(self.period._db), nr_entries - 1)
self.assertEqual(
self.period._category_cache[entry_name][_DEFAULT_CATEGORY], 0)
def test_get_nonexisting_entry(self):
self.assertRaises(PeriodException, self.period.get_entry, eid=-1)
def test_add_entry_default_date(self):
name = "new backpack"
entry_id = self.period.add_entry(name=name, value=-49.95, date=None)
element = self.period.get_entry(entry_id)
self.assertEqual(element["date"],
dt.date.today().strftime(PERIOD_DATE_FORMAT))
self.period.remove_entry(eid=entry_id)
def test_update_standard_entry(self):
self.period.update_entry(eid=self.eid, value=-100)
element = self.period.get_entry(eid=self.eid)
self.assertEqual(element["value"], -100)
# kwargs with None-value should be ignored; they are passed e.g. by the
# flask_restful RequestParser
self.period.update_entry(eid=self.eid,
name="Trekking Bicycle",
value=None)
element = self.period.get_entry(eid=self.eid)
self.assertEqual(element["name"], "trekking bicycle")
self.assertEqual(self.period._category_cache["bicycle"],
Counter({_DEFAULT_CATEGORY: 0}))
self.assertEqual(self.period._category_cache["trekking bicycle"],
Counter({_DEFAULT_CATEGORY: 1}))
self.period.update_entry(eid=self.eid, category="Sports")
element = self.period.get_entry(eid=self.eid)
self.assertEqual(element["category"], "sports")
self.assertEqual(self.period._category_cache["trekking bicycle"],
Counter({
"sports": 1,
_DEFAULT_CATEGORY: 0
}))
# string-eids should be internally converted to int
self.period.update_entry(eid=str(self.eid),
name="MTB Tandem",
category="Fun",
value=-1000)
element = self.period.get_entry(eid=self.eid)
self.assertEqual(element["name"], "mtb tandem")
self.assertEqual(element["value"], -1000.0)
self.assertEqual(element["category"], "fun")
self.assertEqual(self.period._category_cache["trekking bicycle"],
Counter({
"sports": 0,
_DEFAULT_CATEGORY: 0
}))
self.assertEqual(self.period._category_cache["mtb tandem"],
Counter({"fun": 1}))
def test_update_nonexisting_entry(self):
self.assertRaises(PeriodException,
self.period.update_entry,
eid=0,
name="I shall fail")
def test_add_invalid_entry(self):
self.assertRaises(PeriodException,
self.period.add_entry,
name="I'm invalid",
date="1.1",
value="hundred")
def tearDown(self):
self.period.close()
class TinyDbPeriodRecurrentEntryTestCase(unittest.TestCase):
def setUp(self):
self.period = TinyDbPeriod(name=1901)
def test_recurrent_entries(self):
eid = self.period.add_entry(name="rent",
value=-500,
table_name="recurrent",
frequency="monthly",
start="10-01")
self.assertSetEqual({DEFAULT_TABLE, "recurrent"},
self.period._db.tables())
self.assertEqual(len(self.period._db.table("recurrent").all()), 1)
element = self.period._db.table("recurrent").all()[0]
recurrent_elements = list(
self.period._create_recurrent_elements(element))
self.assertEqual(len(recurrent_elements), 3)
rep_element_names = {e["name"] for e in recurrent_elements}
self.assertSetEqual(
rep_element_names,
{"rent, october", "rent, november", "rent, december"})
matching_elements = self.period.get_entries(
filters={"date": "11"})["recurrent"]
self.assertEqual(len(matching_elements), 1)
self.assertEqual(matching_elements[eid][0]["name"], "rent, november")
# the eid attribute is None because a new Element instance has been
# created in Period._create_recurrent_elements. The 'eid' entry
# however is 1 because the parent element is the first in the
# "recurrent" table
self.assertIsNone(matching_elements[eid][0].eid)
def test_recurrent_quarter_yearly_entries(self):
eid = self.period.add_entry(name="interest",
value=25,
table_name="recurrent",
frequency="quarter-yearly",
start="01-01")
element = self.period._db.table("recurrent").all()[0]
recurrent_elements = list(
self.period._create_recurrent_elements(element))
self.assertEqual(len(recurrent_elements), 4)
rep_element_names = {e["name"] for e in recurrent_elements}
self.assertSetEqual(
rep_element_names, {
"interest, january", "interest, april", "interest, july",
"interest, october"
})
recurrent_table_size = len(self.period._db.table("recurrent"))
self.period.remove_entry(eid=eid, table_name="recurrent")
self.assertEqual(len(self.period._db.table("recurrent")),
recurrent_table_size - 1)
def test_recurrent_bimonthly_entries(self):
eid = self.period.add_entry(name="interest",
value=25,
table_name="recurrent",
frequency="bimonthly",
start="01-08",
end="03-08")
recurrent_elements = self.period.get_entries()["recurrent"][eid]
self.assertEqual(len(recurrent_elements), 2)
self.assertEqual(recurrent_elements[0]["name"], "interest, january")
def test_recurrent_weekly_entries(self):
eid = self.period.add_entry(name="interest",
value=25,
table_name="recurrent",
frequency="weekly",
start="01-08",
end="01-14")
recurrent_elements = self.period.get_entries()["recurrent"][eid]
self.assertEqual(len(recurrent_elements), 1)
self.assertEqual(recurrent_elements[0]["name"], "interest, week 01")
def test_update_recurrent_entry(self):
eid = self.period.add_entry(name="interest",
value=25,
table_name="recurrent",
frequency="quarter-yearly",
start="01-01")
self.period.update_entry(eid=eid,
frequency="half-yearly",
start="03-01",
end="06-30",
table_name="recurrent")
entry = self.period.get_entry(eid=eid, table_name="recurrent")
self.assertEqual(entry["frequency"], "half-yearly")
self.assertEqual(entry["start"], "03-01")
self.assertEqual(entry["end"], "06-30")
recurrent_entries = self.period.get_entries()["recurrent"][eid]
self.assertEqual(len(recurrent_entries), 1)
self.assertEqual(recurrent_entries[0]["date"], "03-01")
def test_update_recurrent_entry_incorrectly(self):
eid = self.period.add_entry(name="interest",
value=25,
table_name="recurrent",
frequency="quarter-yearly",
start="01-01")
with self.assertRaises(PeriodException) as context:
self.period.update_entry(eid=eid,
end="Dec-24",
table_name="recurrent")
self.assertIn("end", str(context.exception))
def test_recurrent_yearly_entry(self):
eid = self.period.add_entry(name="Fee",
value=-100,
start="01-01",
table_name="recurrent",
frequency="yearly")
recurrent_entries = self.period.get_entries()["recurrent"][eid]
self.assertEqual(len(recurrent_entries), 1)
self.assertEqual(recurrent_entries[0]["date"], "01-01")
self.assertEqual(recurrent_entries[0]["name"], "fee")
def tearDown(self):
self.period.close()