def test_tasks(self):
s = Supplier()
s.name = 'Joe'
s.save()
p = Property()
p.save()
l = Location(property = p)
l.save()
a = Asset(location = l)
a.save()
t = Task(asset = a)
t.name = 'Mop floors'
t.frequency = 1
t.save()
q = Quote(supplier = s)
q.amount = 4
q.save()
self.assertEqual(0, len(q.tasks.all()))
q.tasks.add(t)
self.assertEqual(1, len(q.tasks.all()))
self.assertEqual
self.assertEqual("$4.00 to Mop floors (Joe)", str(q))
def test_frequency_object(self):
"""
Ensure that the .frequency setter works with Frequency objects, too.
"""
t = Task()
for value in [0, 1, 6, 12]:
t.frequency = Frequency(value)
self.assertEqual(value, t._frequency)
def test_frequency(self):
t = Task()
tests = {
0: "once",
1: "every month",
2: "every 2 months",
12: "every year",
13: "every 13 months",
24: "every 2 years"
}
for value, expected in tests.items():
t.frequency = value
self.assertEqual(expected, str(t.frequency))
def _create_task(row, asset):
if row['task'] == '':
return None
try:
task = Task.objects.get(
asset__pk=asset.pk,
name__iexact=row['task'])
except Task.DoesNotExist:
task = Task(name=row['task'], asset=asset)
task.frequency = Frequency.parse(row['frequency'])
task.last_done = _parse_last_done(row['last done'])
task.priority = Priority.parse(row['priority'])
if row['estimate'] != '':
task.estimate = float(row['estimate'])
task.save()
return task
def test_next_due_zero(self):
t = Task()
t.last_done = date(2012, 1, 1)
t.frequency = 0
self.assertIsNone(t.next_due)
def test_next_due_null(self):
t = Task()
t.last_done = None
t.frequency = 1
self.assertIsNone(t.next_due)
def test_next_due(self):
t = Task()
t.last_done = date(2012, 1, 1)
t.frequency = 1
self.assertEqual(date(2012, 2, 1), t.next_due)
