import pytest
from lightphe import LightPHE
from lightphe.commons.logger import Logger

logger = Logger(module="tests/test_paillier.py")

cs = LightPHE(algorithm_name="Paillier", key_size=50)


def test_api():
    m1 = 17
    m2 = 21

    c1 = cs.encrypt(plaintext=m1)
    c2 = cs.encrypt(plaintext=m2)

    # homomorphic addition
    assert cs.decrypt(c1 + c2) == m1 + m2

    # homomorphic scalar multiplication
    assert cs.decrypt(c1 * m2) == m1 * m2
    assert cs.decrypt(c2 * m1) == m1 * m2
    assert cs.decrypt(m2 * c1) == m1 * m2
    assert cs.decrypt(m1 * c2) == m1 * m2

    c1_prime = cs.regenerate_ciphertext(c1)
    assert c1_prime.value != c1.value
    assert cs.decrypt(c1_prime) == cs.decrypt(c1)

    # unsupported homomorphic operations
    with pytest.raises(ValueError):
        _ = c1 * c2

    with pytest.raises(ValueError):
        _ = c1 ^ c2

    logger.info("✅ Paillier api test succeeded")


def test_float_operations():
    m1 = 1000
    m2 = -10

    c1 = cs.encrypt(plaintext=m1)
    c2 = cs.encrypt(plaintext=m2)

    assert cs.decrypt(c1 + c2) == m1 + m2
    assert cs.decrypt(c2 + c1) == m1 + m2

    k1 = 20
    assert cs.decrypt(c1 * k1) == m1 * k1
    assert cs.decrypt(k1 * c1) == m1 * k1

    k2 = 1.05
    assert cs.decrypt(c1 * k2) == m1 * k2
    assert cs.decrypt(k2 * c1) == m1 * k2

    k3 = -20
    assert cs.decrypt(c1 * k3) == (m1 * k3) % cs.cs.plaintext_modulo
    assert cs.decrypt(k3 * c1) == (m1 * k3) % cs.cs.plaintext_modulo