Function test_add_single_column

xtop/tests/test_tui_basic.py:286–317

Test 3: Adding a column to existing STATE grouping

()

@pytest.mark.asyncio
async def test_add_single_column():
    """Test 3: Adding a column to existing STATE grouping"""
    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 columns
        initial_cols, _ = get_table_data(app)
        print(f"Initial columns: {initial_cols}")
        
        # Try to add USERNAME column (more likely to work in grouping)
        await select_single_column_in_grouping(pilot, 'USERNAME')
        
        # Wait for query to execute
        await pilot.pause()
        await wait_for_table_load(pilot)
        
        # Get new columns
        new_cols, _ = get_table_data(app)
        print(f"New columns after adding USERNAME: {new_cols}")
        
        # Check if we have more columns than before
        if len(new_cols) > len(initial_cols):
            print("✓ Test 3 passed: Successfully added a column to grouping")
        else:
            raise AssertionError(f"Failed to add column. Initial: {initial_cols}, After: {new_cols}")


@pytest.mark.asyncio