Coverage for /Users/ajo/work/jumpstarter/jumpstarter/packages/jumpstarter-driver-uboot/jumpstarter_driver_uboot/driver.py: 88%

1from dataclasses import dataclass 

2 

3from jumpstarter.driver import Driver, export 

4 

5 

6@dataclass(kw_only=True) 

7class UbootConsole(Driver): 

8 prompt: str = "=>" 

9 

10 @classmethod 

11 def client(cls) -> str: 

12 return "jumpstarter_driver_uboot.client.UbootConsoleClient" 

13 

14 def __post_init__(self): 

15 if hasattr(super(), "__post_init__"): 

16 super().__post_init__() 

17 

18 for child in ("power", "serial"): 

19 if child not in self.children: 

20 raise ValueError("UbootConsole: {} driver not configured as a child".format(child)) 

21 

22 @export 

23 def get_prompt(self) -> str: 

24 return self.prompt