summaryrefslogtreecommitdiffstats
path: root/bitbake/lib/toaster/tests/functional/utils.py
blob: 7269fa1805c07bc3f7268e12c7661d69cb1fe1a3 (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
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# BitBake Toaster UI tests implementation
#
# Copyright (C) 2023 Savoir-faire Linux
#
# SPDX-License-Identifier: GPL-2.0-only


from time import sleep
from selenium.common.exceptions import NoSuchElementException, StaleElementReferenceException, TimeoutException
from selenium.webdriver.common.by import By

from orm.models import Build


def wait_until_build(test_instance, state):
    timeout = 60
    start_time = 0
    build_state = ''
    while True:
        try:
            if start_time > timeout:
                raise TimeoutException(
                    f'Build did not reach {state} state within {timeout} seconds'
                )
            last_build_state = test_instance.driver.find_element(
                By.XPATH,
                '//*[@id="latest-builds"]/div[1]//div[@class="build-state"]',
            )
            build_state = last_build_state.get_attribute(
                'data-build-state')
            state_text = state.lower().split()
            if any(x in str(build_state).lower() for x in state_text):
                return str(build_state).lower()
            if 'failed' in str(build_state).lower():
                break
        except NoSuchElementException:
            continue
        except TimeoutException:
            break
        start_time += 1
        sleep(1) # take a breath and try again

def wait_until_build_cancelled(test_instance):
    """ Cancel build take a while sometime, the method is to wait driver action
        until build being cancelled
    """
    timeout = 30
    start_time = 0
    build = None
    while True:
        try:
            if start_time > timeout:
                raise TimeoutException(
                    f'Build did not reach cancelled state within {timeout} seconds'
                )
            last_build_state = test_instance.driver.find_element(
                By.XPATH,
                '//*[@id="latest-builds"]/div[1]//div[@class="build-state"]',
            )
            build_state = last_build_state.get_attribute(
                'data-build-state')
            if 'failed' in str(build_state).lower():
                break
            if 'cancelling' in str(build_state).lower():
                # Change build state to cancelled
                if not build:  # get build object only once
                    build = Build.objects.last()
                    build.outcome = Build.CANCELLED
                    build.save()
            if 'cancelled' in str(build_state).lower():
                break
        except NoSuchElementException:
            continue
        except StaleElementReferenceException:
            continue
        except TimeoutException:
            break
        start_time += 1
        sleep(1) # take a breath and try again

def get_projectId_from_url(url):
    # url = 'http://domainename.com/toastergui/project/1656/whatever
    # or url = 'http://domainename.com/toastergui/project/1/
    # or url = 'http://domainename.com/toastergui/project/186
    assert '/toastergui/project/' in url, "URL is not valid"
    url_to_list = url.split('/toastergui/project/')
    return  int(url_to_list[1].split('/')[0])  # project_id