def test_start_is_valid_format(self, mock_datetime, mock_parse): parse_datetime = mock.Mock() mock_parse.return_value = parse_datetime output = convert_start_to_datetime('2015-01-01') mock_parse.assert_called_once_with('2015-01-01') self.assertEqual(output, parse_datetime)
def test_start_is_none_returns_now(self, mock_datetime, mock_parse): mock_now = mock.Mock() mock_datetime.now = mock_now now = mock.Mock() mock_now.return_value = now output = convert_start_to_datetime(None) mock_now.assert_called_once_with() self.assertEqual(output, now)