OXIESEC PANEL

Name	Size	Modified	Perms
📁 ..	-	05/26/2025 10:13:33 PM	rwxr-xr-x
📄 __init__.py	1.34 KB	04/20/2020 02:31:39 PM	rw-r--r--
📁 __pycache__	-	05/26/2025 10:13:33 PM	rwxr-xr-x
📄 test_netplan.py	1.61 KB	04/20/2020 02:31:39 PM	rw-r--r--
📄 test_view.py	2.97 KB	04/20/2020 02:31:39 PM	rw-r--r--

Editing: test_view.py

# Copyright 2020 Canonical, Ltd.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 3.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import urwid

from subiquitycore.tests import SubiTestCase
from subiquitycore.view import BaseView
from subiquitycore.ui.stretchy import Stretchy, StretchyOverlay
from subiquitycore.ui.utils import undisabled

class TestStretchy(Stretchy):
    was_opened = was_closed = False

def __init__(self):
        super().__init__("", [urwid.Text("")], 0, 0)

def opened(self):
        self.was_opened = True

def closed(self):
        self.was_closed = True

class TestBaseView(SubiTestCase):

def get_stretchy_chain(self, view):
        view = view._w
        r = []
        while isinstance(view, StretchyOverlay):
            r.append(view.stretchy)
            view = undisabled(view.bottom_w)
        return r

def test_basic(self):
        bv = BaseView(urwid.Text(""))
        s = TestStretchy()
        self.assertEqual(self.get_stretchy_chain(bv), [])
        bv.show_stretchy_overlay(s)
        self.assertTrue(s.was_opened)
        self.assertEqual(self.get_stretchy_chain(bv), [s])

bv.remove_overlay()
        self.assertTrue(s.was_closed)
        self.assertEqual(self.get_stretchy_chain(bv), [])

bv.show_stretchy_overlay(s)
        bv.remove_overlay(s)
        self.assertEqual(self.get_stretchy_chain(bv), [])

def make_view_with_overlays(self):
        bv = BaseView(urwid.Text(""))
        a = TestStretchy()
        b = TestStretchy()
        c = TestStretchy()
        bv.show_stretchy_overlay(a)
        bv.show_stretchy_overlay(b)
        bv.show_stretchy_overlay(c)
        return bv, a, b, c

def test_multiple(self):
        bv, a, b, c = self.make_view_with_overlays()
        self.assertEqual(self.get_stretchy_chain(bv), [c, b, a])

bv, a, b, c = self.make_view_with_overlays()
        bv.remove_overlay(a)
        self.assertTrue(a.was_closed)
        self.assertEqual(self.get_stretchy_chain(bv), [c, b])

bv, a, b, c = self.make_view_with_overlays()
        bv.remove_overlay(b)
        self.assertTrue(b.was_closed)
        self.assertEqual(self.get_stretchy_chain(bv), [c, a])

bv, a, b, c = self.make_view_with_overlays()
        bv.remove_overlay(c)
        self.assertTrue(c.was_closed)
        self.assertEqual(self.get_stretchy_chain(bv), [b, a])

bv, a, b, c = self.make_view_with_overlays()
        bv.remove_overlay()
        self.assertTrue(c.was_closed)
        self.assertEqual(self.get_stretchy_chain(bv), [b, a])