from unittest import TestCase

from parameterized import parameterized

from pyecsca.ec.curve import EllipticCurve
from pyecsca.ec.curves import get_curve
from pyecsca.ec.mod import Mod
from pyecsca.ec.model import MontgomeryModel
from pyecsca.ec.point import Point


class CurveTests(TestCase):
    def setUp(self):
        self.secp128r1 = get_curve("secp128r1", "projective")
        self.base = self.secp128r1.generator
        self.curve25519 = get_curve("curve25519", "xz")

    def test_init(self):
        with self.assertRaises(ValueError):
            EllipticCurve(MontgomeryModel(), self.secp128r1.curve.coordinate_model, 1,
                          parameters={})

        with self.assertRaises(ValueError):
            EllipticCurve(self.secp128r1.curve.model, self.secp128r1.curve.coordinate_model, 15,
                          parameters={"c": 0})

        with self.assertRaises(ValueError):
            EllipticCurve(self.secp128r1.curve.model, self.secp128r1.curve.coordinate_model, 15,
                          parameters={"a": Mod(1, 5), "b": Mod(2, 5)})

    def test_is_on_curve(self):
        pt = Point(self.secp128r1.curve.coordinate_model,
                   X=Mod(0x161ff7528b899b2d0c28607ca52c5b86, self.secp128r1.curve.prime),
                   Y=Mod(0xcf5ac8395bafeb13c02da292dded7a83, self.secp128r1.curve.prime),
                   Z=Mod(1, self.secp128r1.curve.prime))
        assert self.secp128r1.curve.is_on_curve(pt)
        assert self.secp128r1.curve.is_on_curve(pt.to_affine())
        other = Point(self.secp128r1.curve.coordinate_model,
                      X=Mod(0x161ff7528b899b2d0c28607ca52c5b86, self.secp128r1.curve.prime),
                      Y=Mod(0xaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa, self.secp128r1.curve.prime),
                      Z=Mod(1, self.secp128r1.curve.prime))
        assert not self.secp128r1.curve.is_on_curve(other)

    @parameterized.expand([
        ("secp128r1","projective"),
        ("secp256r1", "projective"),
        ("secp521r1", "projective"),
        ("curve25519", "xz"),
        ("ed25519", "projective"),
        ("ed448", "projective")
    ])
    def test_curve_utils(self, name, coords):
        group = get_curve(name, coords)
        try:
            assert group.curve.is_on_curve(group.generator)
        except NotImplementedError:
            pass

    def test_eq(self):
        self.assertEqual(self.secp128r1.curve, self.secp128r1.curve)
        self.assertNotEqual(self.secp128r1.curve, self.curve25519.curve)
        self.assertNotEqual(self.secp128r1.curve, None)