import unittest

from common import get_context


class TestCase(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        cls.ctx = get_context()

    def test_viewport(self):
        self.ctx.viewport = (1, 2, 3, 4)
        self.assertTupleEqual(self.ctx.viewport, (1, 2, 3, 4))
        self.assertTupleEqual(self.ctx.screen.viewport, (1, 2, 3, 4))
        fbo = self.ctx.simple_framebuffer((10, 10))
        fbo.use()
        self.assertTupleEqual(self.ctx.viewport, (0, 0, 10, 10))
        self.assertTupleEqual(fbo.viewport, (0, 0, 10, 10))
        self.ctx.viewport = (2, 3, 7, 8)
        self.assertTupleEqual(fbo.viewport, (2, 3, 7, 8))
        self.assertTupleEqual(self.ctx.viewport, (2, 3, 7, 8))
        self.assertTupleEqual(self.ctx.screen.viewport, (1, 2, 3, 4))
        self.ctx.screen.viewport = (3, 3, 3, 3)
        self.assertTupleEqual(fbo.viewport, (2, 3, 7, 8))
        self.assertTupleEqual(self.ctx.viewport, (2, 3, 7, 8))
        self.assertTupleEqual(self.ctx.screen.viewport, (3, 3, 3, 3))
        self.ctx.screen.use()
        self.assertTupleEqual(self.ctx.viewport, (3, 3, 3, 3))
        fbo.viewport = (0, 0, 7, 7)
        self.assertTupleEqual(self.ctx.viewport, (3, 3, 3, 3))
        self.assertTupleEqual(self.ctx.screen.viewport, (3, 3, 3, 3))
        self.ctx.viewport = (3, 5, 7, 9)
        self.assertTupleEqual(fbo.viewport, (0, 0, 7, 7))
        self.assertTupleEqual(self.ctx.viewport, (3, 5, 7, 9))
        self.assertTupleEqual(self.ctx.screen.viewport, (3, 5, 7, 9))
        fbo.use()
        self.assertTupleEqual(self.ctx.viewport, (0, 0, 7, 7))
        self.ctx.screen.use()


if __name__ == '__main__':
    unittest.main()

@@ -183,16 +183,46 @@
 (self):%0A
+        self.ctx.screen.use()%0A
         

use screen first