class PollModelTestCase(Chai, TestCase):
def setUp(self):
super(PollModelTestCase, self).setUp()
self.subject = Poll()
def test_is_visable(self):
now = datetime.datetime.now()
dt = self.mock()
mock_dt = self.mock(models, 'datetime')
self.expect(mock_dt.now).returns(now)
self.subject.publish_date = now
assert self.subject.is_visable()
def test_is_not_visable(self):
now = datetime.datetime.now()
dt = self.mock()
mock_dt = self.mock(models, 'datetime')
self.expect(mock_dt.now).returns(now)
self.subject.publish_date = now - datetime.timedelta(days=10)
assert not self.subject.is_visable()
