from app.models import ( Playlists, Tracks, ) from app.helpers import get_file_metadata from app import playlistmodel from dbconfig import scoped_session test_tracks = [ "testdata/isa.mp3", "testdata/isa_with_gap.mp3", "testdata/loser.mp3", "testdata/lovecats-10seconds.mp3", "testdata/lovecats.mp3", "testdata/mom.mp3", "testdata/sitting.mp3", ] def create_model_with_tracks(session: scoped_session) -> "playlistmodel.PlaylistModel": playlist = Playlists(session, "test playlist") model = playlistmodel.PlaylistModel(playlist.id) for row in range(len(test_tracks)): track_path = test_tracks[row % len(test_tracks)] metadata = get_file_metadata(track_path) track = Tracks(session, **metadata) model.insert_track_row(row, track.id, f"{row=}") session.commit() return model def create_model_with_playlist_rows( session: scoped_session, rows: int ) -> "playlistmodel.PlaylistModel": playlist = Playlists(session, "test playlist") # Create a model model = playlistmodel.PlaylistModel(playlist.id) for row in range(rows): plr = model._insert_row(session, row) newrow = plr.plr_rownum plr.note = str(newrow) model.playlist_rows[newrow] = playlistmodel.PlaylistRowData(plr) session.commit() return model def test_11_row_playlist(monkeypatch, session): # Create multirow playlist monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) assert model.rowCount() == 11 assert max(model.playlist_rows.keys()) == 10 for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row def test_move_rows_test2(monkeypatch, session): # move row 3 to row 5 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([3], 5) # Check we have all rows and plr_rownums are correct for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row if row not in [3, 4, 5]: assert model.playlist_rows[row].note == str(row) elif row == 3: assert model.playlist_rows[row].note == str(4) elif row == 4: assert model.playlist_rows[row].note == str(5) elif row == 5: assert model.playlist_rows[row].note == str(3) def test_move_rows_test3(monkeypatch, session): # move row 4 to row 3 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([4], 3) # Check we have all rows and plr_rownums are correct for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row if row not in [3, 4]: assert model.playlist_rows[row].note == str(row) elif row == 3: assert model.playlist_rows[row].note == str(4) elif row == 4: assert model.playlist_rows[row].note == str(3) def test_move_rows_test4(monkeypatch, session): # move row 4 to row 2 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([4], 2) # Check we have all rows and plr_rownums are correct for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row if row not in [2, 3, 4]: assert model.playlist_rows[row].note == str(row) elif row == 2: assert model.playlist_rows[row].note == str(4) elif row == 3: assert model.playlist_rows[row].note == str(2) elif row == 4: assert model.playlist_rows[row].note == str(3) def test_move_rows_test5(monkeypatch, session): # move rows [1, 4, 5, 10] → 8 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([1, 4, 5, 10], 8) # Check we have all rows and plr_rownums are correct new_order = [] for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row new_order.append(int(model.playlist_rows[row].note)) assert new_order == [0, 2, 3, 6, 7, 8, 9, 1, 4, 5, 10] def test_move_rows_test6(monkeypatch, session): # move rows [3, 6] → 5 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([3, 6], 5) # Check we have all rows and plr_rownums are correct new_order = [] for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row new_order.append(int(model.playlist_rows[row].note)) assert new_order == [0, 1, 2, 4, 5, 3, 6, 7, 8, 9, 10] def test_move_rows_test7(monkeypatch, session): # move rows [3, 5, 6] → 8 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([3, 5, 6], 8) # Check we have all rows and plr_rownums are correct new_order = [] for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row new_order.append(int(model.playlist_rows[row].note)) assert new_order == [0, 1, 2, 4, 7, 8, 9, 10, 3, 5, 6] def test_move_rows_test8(monkeypatch, session): # move rows [7, 8, 10] → 5 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_playlist_rows(session, 11) model.move_rows([7, 8, 10], 5) # Check we have all rows and plr_rownums are correct new_order = [] for row in range(model.rowCount()): assert row in model.playlist_rows assert model.playlist_rows[row].plr_rownum == row new_order.append(int(model.playlist_rows[row].note)) assert new_order == [0, 1, 2, 3, 4, 7, 8, 10, 5, 6, 9] def test_insert_header_row_end(monkeypatch, session): # insert header row at end of playlist monkeypatch.setattr(playlistmodel, "Session", session) note_text = "test text" initial_row_count = 11 model = create_model_with_playlist_rows(session, initial_row_count) model.insert_header_row(None, note_text) assert model.rowCount() == initial_row_count + 1 prd = model.playlist_rows[model.rowCount() - 1] # Test against edit_role because display_role for headers is # handled differently (sets up row span) assert ( model.edit_role(model.rowCount(), playlistmodel.Col.NOTE.value, prd) == note_text ) def test_insert_header_row_middle(monkeypatch, session): # insert header row in middle of playlist monkeypatch.setattr(playlistmodel, "Session", session) note_text = "test text" initial_row_count = 11 insert_row = 6 model = create_model_with_playlist_rows(session, initial_row_count) model.insert_header_row(insert_row, note_text) assert model.rowCount() == initial_row_count + 1 prd = model.playlist_rows[insert_row] # Test against edit_role because display_role for headers is # handled differently (sets up row span) assert ( model.edit_role(model.rowCount(), playlistmodel.Col.NOTE.value, prd) == note_text ) def test_create_model_with_tracks(monkeypatch, session): monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_tracks(session) assert len(model.playlist_rows) == len(test_tracks) def test_timing_one_track(monkeypatch, session): START_ROW = 0 END_ROW = 2 monkeypatch.setattr(playlistmodel, "Session", session) model = create_model_with_tracks(session) model.insert_header_row(START_ROW, "start+") model.insert_header_row(END_ROW, "-") prd = model.playlist_rows[START_ROW] qv_value = model.display_role(START_ROW, playlistmodel.HEADER_NOTES_COLUMN, prd) assert qv_value.value() == "start [1 tracks, 4:23 unplayed]" # def test_edit_header(monkeypatch, session): # edit header row in middle of playlist # monkeypatch.setattr(playlistmodel, "Session", session) # note_text = "test text" # initial_row_count = 11 # insert_row = 6 # model = create_model_with_playlist_rows(session, initial_row_count) # model.insert_header_row(insert_row, note_text) # assert model.rowCount() == initial_row_count + 1 # prd = model.playlist_rows[insert_row] # # Test against edit_role because display_role for headers is # # handled differently (sets up row span) # assert ( # model.edit_role(model.rowCount(), playlistmodel.Col.NOTE.value, prd) # == note_text # )