masq/i3pystatus
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Changed Config.test

Browse Source
This commit is contained in:
enkore 2013-03-11 00:28:24 +01:00
parent a5b274fd23
commit c340b2cf75
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
i3pystatus/core/config.py
View File

@ -2,6 +2,7 @@
import os.path import os.path
import runpy import runpy
import sys import sys
import contextlib
from .render import render_json from .render import render_json
@ -46,8 +47,8 @@ class Config:
runpy.run_path(self.config_file, run_name="i3pystatus._config") runpy.run_path(self.config_file, run_name="i3pystatus._config")
def test(self): def test(self):
@contextlib.contextmanager
def setup(): def setup():
"""This is a wrapped method so no one ever tries to use it outside of this"""
import i3pystatus import i3pystatus
class TestStatus(i3pystatus.Status): class TestStatus(i3pystatus.Status):
def run(self): def run(self):
@ -60,8 +61,10 @@ class Config:
print(render_json(output)) print(render_json(output))
i3pystatus.Status = TestStatus i3pystatus.Status = TestStatus
setup() yield
i3pystatus.Status = i3pystatus.Status.__bases__[0]
with setup():
print("Using configuration file {file}".format(file=self.config_file)) print("Using configuration file {file}".format(file=self.config_file))
print("Output, would be displayed right to left in i3bar") print("Output, would be displayed right to left in i3bar")
self.run() self.run()
sys.exit(0) # Exit program, kill any state left behind by TestStatus