mirror of
https://github.com/kennethreitz/records.git
synced 2026-06-05 06:46:17 +00:00
Chad Whitacre
e7da94b249
It's a common use-case to want one and only one result from a query. This adds a .one method to RecordCollections that is parallel to .all.
90 lines
2.5 KiB
Python
90 lines
2.5 KiB
Python
from collections import namedtuple
|
|
|
|
import records
|
|
|
|
from pytest import raises
|
|
|
|
|
|
IdRecord = namedtuple('IdRecord', 'id')
|
|
|
|
|
|
def check_id(i, row):
|
|
assert row.id == i
|
|
|
|
|
|
class TestRecordCollection:
|
|
def test_iter(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
for i, row in enumerate(rows):
|
|
check_id(i, row)
|
|
|
|
def test_next(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
for i in range(10):
|
|
check_id(i, next(rows))
|
|
|
|
def test_iter_and_next(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
i = enumerate(iter(rows))
|
|
check_id(*next(i)) # Cache first row.
|
|
next(rows) # Cache second row.
|
|
check_id(*next(i)) # Read second row from cache.
|
|
|
|
def test_multiple_iter(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
i = enumerate(iter(rows))
|
|
j = enumerate(iter(rows))
|
|
|
|
check_id(*next(i)) # Cache first row.
|
|
|
|
check_id(*next(j)) # Read first row from cache.
|
|
check_id(*next(j)) # Cache second row.
|
|
|
|
check_id(*next(i)) # Read second row from cache.
|
|
|
|
def test_slice_iter(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(10))
|
|
for i, row in enumerate(rows[:5]):
|
|
check_id(i, row)
|
|
for i, row in enumerate(rows):
|
|
check_id(i, row)
|
|
assert len(rows) == 10
|
|
|
|
|
|
# all
|
|
|
|
def test_all_returns_a_list_of_records(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(3))
|
|
assert rows.all() == [IdRecord(0), IdRecord(1), IdRecord(2)]
|
|
|
|
|
|
# one
|
|
|
|
def test_one_returns_a_single_record(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(1))
|
|
assert rows.one() == IdRecord(0)
|
|
|
|
def test_one_defaults_to_None(self):
|
|
rows = records.RecordCollection(iter([]))
|
|
assert rows.one() is None
|
|
|
|
def test_one_default_is_overridable(self):
|
|
rows = records.RecordCollection(iter([]))
|
|
assert rows.one('Cheese') == 'Cheese'
|
|
|
|
def test_one_raises_when_more_than_one(self):
|
|
rows = records.RecordCollection(IdRecord(i) for i in range(3))
|
|
raises(ValueError, rows.one)
|
|
|
|
|
|
class TestRecord:
|
|
|
|
def test_record_dir(self):
|
|
keys, values = ['id', 'name', 'email'], [1, '', '']
|
|
record = records.Record(keys, values)
|
|
_dir = dir(record)
|
|
for key in keys:
|
|
assert key in _dir
|
|
for key in dir(object):
|
|
assert key in _dir
|