Skip to content

Commit 7b76b82

Browse files
committed
trim white-box queue sequencing tests
1 parent d00e617 commit 7b76b82

2 files changed

Lines changed: 0 additions & 60 deletions

File tree

invokeai/frontend/web/src/features/controlLayers/components/StagingArea/state.test.ts

Lines changed: 0 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -840,8 +840,6 @@ describe('StagingAreaApi', () => {
840840
api.$items.set([createMockQueueItem({ item_id: 1 })]);
841841
api.$lastStartedItemId.set(1);
842842
api.$lastCompletedItemId.set(1);
843-
api._seenItemStatusRanks.set(1, 2);
844-
api._seenItemStatusSequences.set(1, 3);
845843
api.$progressData.setKey(1, {
846844
itemId: 1,
847845
progressEvent: null,
@@ -858,8 +856,6 @@ describe('StagingAreaApi', () => {
858856
expect(api.$lastStartedItemId.get()).toBe(null);
859857
expect(api.$lastCompletedItemId.get()).toBe(null);
860858
expect(api.$progressData.get()).toEqual({});
861-
expect(api._seenItemStatusRanks.size).toBe(0);
862-
expect(api._seenItemStatusSequences.size).toBe(0);
863859
});
864860
});
865861

@@ -1179,32 +1175,6 @@ describe('StagingAreaApi', () => {
11791175
expect(api.$isPending.get()).toBe(false);
11801176
});
11811177

1182-
it('should prune seen item ordering when an item leaves the staging area', async () => {
1183-
await api.onItemsChangedEvent([
1184-
createMockQueueItem({
1185-
item_id: 1,
1186-
status: 'completed',
1187-
status_sequence: 2,
1188-
}),
1189-
]);
1190-
1191-
expect(api._seenItemStatusRanks.get(1)).toBe(2);
1192-
expect(api._seenItemStatusSequences.get(1)).toBe(2);
1193-
1194-
await api.onItemsChangedEvent([
1195-
createMockQueueItem({
1196-
item_id: 2,
1197-
status: 'pending',
1198-
status_sequence: 0,
1199-
}),
1200-
]);
1201-
1202-
expect(api._seenItemStatusRanks.has(1)).toBe(false);
1203-
expect(api._seenItemStatusSequences.has(1)).toBe(false);
1204-
expect(api._seenItemStatusRanks.get(2)).toBe(0);
1205-
expect(api._seenItemStatusSequences.get(2)).toBe(0);
1206-
});
1207-
12081178
it('should load all images from multiple canvas_output nodes', async () => {
12091179
const imageDTO1 = createMockImageDTO({ image_name: 'output1.png' });
12101180
const imageDTO2 = createMockImageDTO({ image_name: 'output2.png' });

tests/test_sqlite_migrator.py

Lines changed: 0 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -249,36 +249,6 @@ def test_migrator_runs_all_migrations_file(logger: Logger) -> None:
249249
# Must manually close else we get an error on Windows
250250
db._conn.close()
251251

252-
253-
def test_migration_28_adds_status_sequence_to_session_queue(memory_db_conn: sqlite3.Connection) -> None:
254-
from invokeai.app.services.shared.sqlite_migrator.migrations.migration_28 import Migration28Callback
255-
256-
cursor = memory_db_conn.cursor()
257-
cursor.execute(
258-
"""--sql
259-
CREATE TABLE IF NOT EXISTS session_queue (
260-
item_id INTEGER PRIMARY KEY,
261-
status TEXT NOT NULL DEFAULT 'pending'
262-
);
263-
"""
264-
)
265-
cursor.execute("INSERT INTO session_queue (item_id, status) VALUES (1, 'pending');")
266-
267-
Migration28Callback()(cursor)
268-
Migration28Callback()(cursor)
269-
270-
cursor.execute("PRAGMA table_info(session_queue);")
271-
columns = [row[1] for row in cursor.fetchall()]
272-
assert columns.count("status_sequence") == 1
273-
274-
cursor.execute("PRAGMA table_info(session_queue);")
275-
columns_by_name = {row[1]: row for row in cursor.fetchall()}
276-
assert "status_sequence" in columns_by_name
277-
278-
cursor.execute("SELECT status_sequence FROM session_queue WHERE item_id = 1;")
279-
assert cursor.fetchone()[0] == 0
280-
281-
282252
def test_migrator_backs_up_db(logger: Logger) -> None:
283253
with TemporaryDirectory() as tempdir:
284254
original_db_path = Path(tempdir) / "invokeai.db"

0 commit comments

Comments
 (0)