def test_search_sort_year(factory: Factory, db: Connection): for _ in range(5): factory.release(release_year=factory.rand_year(), conn=db) releases = release.search(sort=ReleaseSort.YEAR, asc=True, conn=db) # This is the Unknown Release (with a null year). Should be sorted last. assert releases[-1].release_year is None not_null_years = [rls.release_year for rls in releases[:-1]] assert None not in not_null_years assert not_null_years == sorted(not_null_years) # type: ignore
def test_search_sort_year(factory: Factory, db: Connection): # Unknown Release has no year. factory.track(release_id=1, conn=db) # Now create five track with a year. for _ in range(5): rls = factory.release(release_year=factory.rand_year(), conn=db) factory.track(release_id=rls.id, conn=db) out = track.search(sort=TrackSort.YEAR, asc=True, conn=db) years = [track.release(t, db).release_year for t in out] # Assert the track attached to Unknown Release has no year and is last. assert years[-1] is None not_null_years = years[:-1] assert not_null_years == sorted(not_null_years) # type: ignore