raises error when dates passed to query as long numbers

This commit is contained in:
discopatrick 2017-04-14 00:40:06 +01:00
parent 45bed808d3
commit 1744ca6e88
2 changed files with 4 additions and 3 deletions

View file

@ -561,7 +561,7 @@ class Period(object):
date = datetime.strptime(string, date_format)
except ValueError:
# Parsing failed.
return None
raise InvalidQueryArgumentTypeError(string, datetime)
precision = cls.precisions[ordinal]
return cls(date, precision)

View file

@ -21,7 +21,7 @@ from test import _common
from datetime import datetime
import unittest
import time
from beets.dbcore.query import _parse_periods, DateInterval, DateQuery
from beets.dbcore.query import _parse_periods, DateInterval, DateQuery, InvalidQueryArgumentTypeError
def _date(string):
@ -117,7 +117,8 @@ class DateQueryTest(_common.LibTestCase):
class DateQueryConstructTest(unittest.TestCase):
def test_long_numbers(self):
DateQuery('added', '1409830085..1412422089')
with self.assertRaises(InvalidQueryArgumentTypeError):
DateQuery('added', '1409830085..1412422089')
def test_too_many_components(self):
DateQuery('added', '12-34-56-78')