Function test_add_multiple_columns

xtop/tests/test_tui_basic.py:321–360

Test 4: Adding multiple columns one at a time

()

@pytest.mark.asyncio
async def test_add_multiple_columns():
    """Test 4: Adding multiple columns one at a time"""
    wrapper = XtopTUIWrapper(
        datadir=XCAPTURE_DATADIR,
        initial_group_cols=['state']
    )
    app = wrapper.create_app()
    
    async with app.run_test(size=(120, 40)) as pilot:
        await pilot.pause()
        await wait_for_table_load(pilot)
        
        # Get initial column count
        initial_cols, _ = get_table_data(app)
        initial_count = len(initial_cols)
        print(f"Starting with {initial_count} columns: {initial_cols}")
        
        # Add first column (USERNAME)
        await select_single_column_in_grouping(pilot, 'USERNAME')
        await pilot.pause()
        await wait_for_table_load(pilot)
        
        # Check column count increased
        cols_after_first, _ = get_table_data(app)
        if len(cols_after_first) <= initial_count:
            raise AssertionError(f"Failed to add first column. Before: {initial_cols}, After: {cols_after_first}")
        
        print(f"After first addition: {len(cols_after_first)} columns")
        
        # Add second column (EXE)
        await select_single_column_in_grouping(pilot, 'EXE')
        await pilot.pause()
        await wait_for_table_load(pilot)
        
        # Check column count increased again
        final_cols, _ = get_table_data(app)
        if len(final_cols) <= len(cols_after_first):
            raise AssertionError(f"Failed to add second column. Before: {cols_after_first}, After: {final_cols}")
        
        print(f"✓ Test 4 passed: Added multiple columns. Final: {len(final_cols)} columns")


@pytest.mark.asyncio