2
from mago.test_suite.gnome_screenshot import GnomeScreenshotTestSuite
4
class TakeScreenshot(GnomeScreenshotTestSuite):
5
def grab_desktop(self, delay=5, filename='baz.png'):
6
self.application.open()
7
self.application.take_whole_desktop_screenshot(delay)
9
self.application.save_to_file(filename)
11
file_path = os.getenv('HOME') + '/' + filename
13
if os.access(file_path, os.F_OK):
16
raise AssertionError('Screenshot ' + file_path + ' was not created')
18
def grab_window(self, delay=5, filename='baz.png'):
19
self.application.open()
21
self.application.take_current_window_screenshot(delay)
23
self.application.save_to_file(filename)
25
file_path = os.getenv('HOME') + '/' + filename
27
if os.access(file_path, os.F_OK):
30
raise AssertionError('Screenshot ' + file_path + ' was not created')
32
def grab_area(self, delay=5, filename='baz.png'):
33
self.application.open()
35
self.application.grab_selected_area(delay)
37
self.application.save_to_file(filename)
39
file_path = os.getenv('HOME') + '/' + filename
41
if os.access(file_path, os.F_OK):
44
raise AssertionError('Screenshot ' + file_path + ' was not created')
46
if __name__ == "__main__":
47
test = TakeScreenshot()