require_relative '../util'

def test_diffie_hellman(p, g)
  a, _A = generate_dh_keys(p, g)
  b, _B = generate_dh_keys(p, g)
  assert(modexp(_B, a, p) == modexp(_A, b, p))
end

test_diffie_hellman(37, 5)
test_diffie_hellman(0xffffffffffffffffc90fdaa22168c234c4c6628b80dc1cd129024e088a67cc74020bbea63b139b22514a08798e3404ddef9519b3cd3a431b302b0a6df25f14374fe1356d6d51c245e485b576625e7ec6f44c42e9a637ed6b0bff5cb6f406b7edee386bfb5a899fa5ae9f24117c4b1fe649286651ece45b3dc2007cb8a163bf0598da48361c55d39a69163fa8fd24cf5f83655d23dca3ad961c62f356208552bb9ed529077096966d670c354e4abc9804f1746c08ca237327ffffffffffffffff, 2)