summaryrefslogtreecommitdiff
path: root/scripts/testing/cleanup-tests.sh
blob: e4289a790f0c01f506ee5ed36ce86040431b25e3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
#!/bin/bash
# Clean up old test VMs and artifacts
# Author: Craig Jennings <craigmartinjennings@gmail.com>
# License: GNU GPLv3

set -e

# Get script directory
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PROJECT_ROOT="$(cd "$SCRIPT_DIR/../.." && pwd)"

# Source utilities
source "$SCRIPT_DIR/lib/logging.sh"
source "$SCRIPT_DIR/lib/vm-utils.sh"

# Parse arguments
KEEP_LAST=5
FORCE=false

while [[ $# -gt 0 ]]; do
    case $1 in
        --keep)
            KEEP_LAST="$2"
            shift 2
            ;;
        --force)
            FORCE=true
            shift
            ;;
        *)
            echo "Usage: $0 [--keep N] [--force]"
            echo "  --keep N  Keep last N test results (default: 5)"
            echo "  --force   Skip confirmation prompts"
            exit 1
            ;;
    esac
done

# Initialize logging
LOGFILE="/tmp/cleanup-tests-$(date +'%Y%m%d-%H%M%S').log"
init_logging "$LOGFILE"

section "Cleaning Up Test Artifacts"

# Find all test VMs
step "Finding test VMs"
TEST_VMS=$(virsh --connect qemu:///system list --all | grep "archsetup-test-" | awk '{print $2}' || true)

if [ -z "$TEST_VMS" ]; then
    info "No test VMs found"
else
    VM_COUNT=$(echo "$TEST_VMS" | wc -l)
    info "Found $VM_COUNT test VM(s)"

    if ! $FORCE; then
        echo ""
        echo "$TEST_VMS"
        echo ""
        read -p "Destroy these VMs? [y/N] " -n 1 -r
        echo ""
        if [[ ! $REPLY =~ ^[Yy]$ ]]; then
            info "Skipping VM cleanup"
        else
            for vm in $TEST_VMS; do
                step "Destroying VM: $vm"
                if vm_is_running "$vm"; then
                    virsh --connect qemu:///system destroy "$vm" >> "$LOGFILE" 2>&1
                fi
                virsh --connect qemu:///system undefine "$vm" --nvram >> "$LOGFILE" 2>&1 || true
                success "VM destroyed: $vm"
            done
        fi
    else
        for vm in $TEST_VMS; do
            step "Destroying VM: $vm"
            if vm_is_running "$vm"; then
                virsh --connect qemu:///system destroy "$vm" >> "$LOGFILE" 2>&1
            fi
            virsh --connect qemu:///system undefine "$vm" --nvram >> "$LOGFILE" 2>&1 || true
            success "VM destroyed: $vm"
        done
    fi
fi

# Clean up test disk images
section "Cleaning Up Disk Images"

step "Finding test disk images"
if [ -d "$PROJECT_ROOT/vm-images" ]; then
    TEST_DISKS=$(find "$PROJECT_ROOT/vm-images" -name "archsetup-test-*.qcow2" 2>/dev/null || true)

    if [ -z "$TEST_DISKS" ]; then
        info "No test disk images found"
    else
        DISK_COUNT=$(echo "$TEST_DISKS" | wc -l)
        DISK_SIZE=$(du -ch $TEST_DISKS | tail -1 | awk '{print $1}')
        info "Found $DISK_COUNT test disk image(s) totaling $DISK_SIZE"

        if ! $FORCE; then
            echo ""
            echo "$TEST_DISKS"
            echo ""
            read -p "Delete these disk images? [y/N] " -n 1 -r
            echo ""
            if [[ ! $REPLY =~ ^[Yy]$ ]]; then
                info "Skipping disk cleanup"
            else
                echo "$TEST_DISKS" | while read disk; do
                    rm -f "$disk"
                done
                success "Test disk images deleted"
            fi
        else
            echo "$TEST_DISKS" | while read disk; do
                rm -f "$disk"
            done
            success "Test disk images deleted"
        fi
    fi
fi

# Clean up old test results
section "Cleaning Up Test Results"

if [ ! -d "$PROJECT_ROOT/test-results" ]; then
    info "No test results directory"
else
    step "Finding test result directories"
    TEST_RESULTS=$(find "$PROJECT_ROOT/test-results" -maxdepth 1 -type d -name "20*" 2>/dev/null | sort -r || true)

    if [ -z "$TEST_RESULTS" ]; then
        info "No test results found"
    else
        RESULT_COUNT=$(echo "$TEST_RESULTS" | wc -l)
        info "Found $RESULT_COUNT test result directory(ies)"

        if [ $RESULT_COUNT -le $KEEP_LAST ]; then
            info "Keeping all results (count <= $KEEP_LAST)"
        else
            TO_DELETE=$(echo "$TEST_RESULTS" | tail -n +$((KEEP_LAST + 1)))
            DELETE_COUNT=$(echo "$TO_DELETE" | wc -l)
            info "Keeping last $KEEP_LAST, deleting $DELETE_COUNT old result(s)"

            if ! $FORCE; then
                echo ""
                echo "Will delete:"
                echo "$TO_DELETE"
                echo ""
                read -p "Delete these test results? [y/N] " -n 1 -r
                echo ""
                if [[ ! $REPLY =~ ^[Yy]$ ]]; then
                    info "Skipping results cleanup"
                else
                    echo "$TO_DELETE" | while read dir; do
                        rm -rf "$dir"
                    done
                    success "Old test results deleted"
                fi
            else
                echo "$TO_DELETE" | while read dir; do
                    rm -rf "$dir"
                done
                success "Old test results deleted"
            fi
        fi
    fi
fi

section "Cleanup Complete"

info "Log file: $LOGFILE"