exth_crypto

defmodule ExthCrypto.Test do
  @moduledoc """
  A variety of helper functions to make the tests consistent
  in their usage of keys, etc.
  """

  @public_keys %{
    key_a: <<4, 54, 241, 224, 126, 85, 135, 69, 213, 129, 115, 3, 41, 161, 217, 87, 215,
             159, 64, 17, 167, 128, 113, 172, 232, 46, 34, 145, 136, 72, 160, 207, 161,
             171, 255, 26, 163, 160, 158, 227, 196, 92, 62, 119, 84, 156, 99, 224, 155,
             120, 250, 153, 134, 180, 218, 177, 186, 200, 199, 106, 97, 103, 50, 215, 114>>,
    key_b: <<4, 152, 113, 235, 8, 21, 103, 130, 50, 103, 89, 42, 186, 200, 236, 158, 159,
             221, 253, 236, 231, 144, 26, 21, 242, 51, 181, 63, 48, 77, 120, 96, 104, 108,
             33, 96, 27, 161, 167, 245, 102, 128, 226, 45, 10, 192, 62, 204, 208, 142, 73,
             100, 105, 81, 76, 37, 174, 29, 94, 85, 243, 145, 193, 149, 111>>,
    key_c: <<4, 146, 201, 161, 205, 19, 177, 147, 33, 107, 190, 144, 81, 145, 173, 83, 20,
             105, 150, 114, 196, 249, 143, 167, 152, 63, 225, 96, 184, 86, 203, 38, 134,
             241, 40, 152, 74, 34, 68, 233, 204, 91, 240, 208, 254, 62, 169, 53, 201, 248,
             156, 236, 34, 203, 156, 75, 18, 121, 162, 104, 3, 164, 156, 46, 186>>,
    key_d: <<4, 205, 208, 163, 120, 32, 181, 60, 4, 225, 235, 149, 145, 233, 130, 114,
             254, 250, 181, 166, 76, 124, 6, 241, 169, 49, 36, 251, 61, 166, 127, 63, 10,
             171, 21, 243, 26, 250, 157, 62, 224, 19, 31, 127, 12, 29, 32, 146, 191, 9,
             168, 124, 96, 120, 236, 233, 81, 8, 23, 61, 48, 4, 9, 213, 31>>
  }

  @private_keys %{
    key_a: <<94, 217, 126, 139, 193, 247, 132, 35, 174, 8, 191, 12, 133, 229, 115, 237,
             78, 81, 160, 114, 73, 118, 207, 206, 98, 114, 27, 62, 25, 29, 219, 206>>,
    key_b: <<226, 137, 30, 163, 26, 230, 61, 203, 158, 81, 58, 4, 197, 149, 169, 80, 34,
             11, 157, 221, 132, 75, 78, 202, 254, 8, 94, 254, 229, 98, 104, 5>>,
    key_c: <<178, 68, 134, 194, 0, 187, 118, 35, 33, 220, 4, 3, 50, 96, 97, 91, 96, 14,
             71, 239, 7, 102, 33, 187, 194, 221, 152, 36, 95, 22, 121, 48>>,
    key_d: <<157, 39, 212, 230, 12, 119, 100, 92, 134, 157, 151, 31, 151, 234, 63, 46,
             213, 201, 34, 95, 208, 141, 55, 74, 2, 59, 121, 174, 7, 243, 230, 46>>
  }

  @symmetric_keys %{
    key_a: <<1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21,
             22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32>>,
    key_b: <<11, 22, 33, 44, 55, 66, 77, 88, 99, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21,
             22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32>>,
  }

  @doc """
  Returns a generic elliptic curve public key based on a given curve.
  """
  def public_key(key \\ :key_a) do
    @public_keys[key]
  end

  @doc """
  Returns a generic elliptic curve public key based on a given curve,
  which is paired with the corresponding public key.
  """
  def private_key(key \\ :key_a) do
    @private_keys[key]
  end

  @doc """
  Returns a generic elliptic curve key pair.
  """
  def key_pair(key \\ :key_a) do
    {@public_keys[key], @private_keys[key]}
  end

  def symmetric_key(key \\ :key_a) do
    @symmetric_keys[key]
  end

  @doc """
  Returns an initialization vector of the given size that is specifically
  not random at all (just for testing).
  """
  def init_vector(base \\ 1, block_size \\ 16) do
    for i <- base..(base + block_size - 1), do: <<i>>, into: <<>>
  end
end