def test_stop_end_time(): obj = TimeRecordFactory(end_time=None) assert obj.end_time is None end_time = datetime(2014, 10, 1, 18, 13, 32, tzinfo=pytz.utc) obj.stop(end_time) assert end_time == obj.end_time
def test_stop_already_stopped(): end_time = datetime(2014, 10, 1, 18, 13, 32, tzinfo=pytz.utc) obj = TimeRecordFactory(end_time=end_time) with pytest.raises(InvoiceError) as e: obj.stop() assert 'has already been stopped' in str(e.value)
def test_stop(): obj = TimeRecordFactory(end_time=None) assert obj.end_time is None obj.stop() assert obj.end_time is not None