from PIL import CurImagePlugin, Image

from .helper import PillowTestCase

TEST_FILE = "Tests/images/deerstalker.cur"


class TestFileCur(PillowTestCase):
    def test_sanity(self):
        with Image.open(TEST_FILE) as im:
            self.assertEqual(im.size, (32, 32))
            self.assertIsInstance(im, CurImagePlugin.CurImageFile)
            # Check some pixel colors to ensure image is loaded properly
            self.assertEqual(im.getpixel((10, 1)), (0, 0, 0, 0))
            self.assertEqual(im.getpixel((11, 1)), (253, 254, 254, 1))
            self.assertEqual(im.getpixel((16, 16)), (84, 87, 86, 255))

    def test_invalid_file(self):
        invalid_file = "Tests/images/flower.jpg"

        self.assertRaises(SyntaxError, CurImagePlugin.CurImageFile, invalid_file)

        no_cursors_file = "Tests/images/no_cursors.cur"

        cur = CurImagePlugin.CurImageFile(TEST_FILE)
        cur.fp.close()
        with open(no_cursors_file, "rb") as cur.fp:
            self.assertRaises(TypeError, cur._open)