def test_timesheet_url(self): with mock.patch('timepiece.templatetags.timepiece_tags.reverse') \ as reverse: reverse.return_value = "Boo" retval = tags.timesheet_url('project', 27, None) self.assertEqual('Boo?', retval) self.assertEqual('view_project_timesheet', reverse.call_args[0][0]) self.assertEqual((27,), reverse.call_args[1]['args'])
def test_timesheet_url2(self): with mock.patch('timepiece.templatetags.timepiece_tags.reverse')\ as reverse: reverse.return_value = "Boo" dt = datetime.date(2013, 1, 10) retval = tags.timesheet_url('user', 13, dt) self.assertEqual('Boo?year=2013&month=1', retval) self.assertEqual('view_user_timesheet', reverse.call_args[0][0]) self.assertEqual((13,), reverse.call_args[1]['args'])