Permalink
Browse files

#9 TI-1: Add test for undo after track with adjust hint

  • Loading branch information...
lauft committed Jul 24, 2018
1 parent ec5bf9f commit 3e17dae695e651713d8ff3d5b3e98866e6b27f1b
Showing with 34 additions and 0 deletions.
  1. +34 −0 test/undo.t
@@ -451,6 +451,40 @@ class TestUndo(TestCase):
expectedEnd=three_hours_before_utc,
expectedTags=["foo"])

def test_undo_track_with_adjust_hint(self):
"""Test undo of command 'track' with adjust hint"""
now_utc = datetime.now().utcnow()
one_hour_before_utc = now_utc - timedelta(hours=1)
two_hours_before_utc = now_utc - timedelta(hours=2)
three_hours_before_utc = now_utc - timedelta(hours=3)
four_hours_before_utc = now_utc - timedelta(hours=4)
six_hours_before_utc = now_utc - timedelta(hours=6)

self.t("track {:%Y%m%dT%H%M%SZ} - {:%Y%m%dT%H%M%SZ} foo".format(four_hours_before_utc, three_hours_before_utc))
self.t("track {:%Y%m%dT%H%M%SZ} - {:%Y%m%dT%H%M%SZ} bar".format(two_hours_before_utc, one_hour_before_utc))

self.t("track {:%Y%m%dT%H%M%SZ} - {:%Y%m%dT%H%M%SZ} xyz :adjust".format(six_hours_before_utc, now_utc))

j = self.t.export()
self.assertEqual(len(j), 1, msg="Expected 2 intervals before, got {}".format(len(j)))
self.assertClosedInterval(j[0],
expectedStart=six_hours_before_utc,
expectedEnd=now_utc,
expectedTags=["xyz"])

self.t("undo")

j = self.t.export()
self.assertEqual(len(j), 2, msg="Expected 2 interval afterwards, got {}".format(len(j)))
self.assertClosedInterval(j[0],
expectedStart=four_hours_before_utc,
expectedEnd=three_hours_before_utc,
expectedTags=["foo"])
self.assertClosedInterval(j[1],
expectedStart=two_hours_before_utc,
expectedEnd=one_hour_before_utc,
expectedTags=["bar"])

def test_undo_untag(self):
"""Test undo of command 'untag'"""
now_utc = datetime.now().utcnow()

0 comments on commit 3e17dae

Please sign in to comment.