#include "../kit/sha256.h"
#include "../kit/array_ref.h"

#define KIT_TEST_FILE sha256_64
#include "../kit/kit_test.h"

#include <string.h>

TEST("sha256") {
  uint8_t text1[] = "abc";
  uint8_t text2[] =
      "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq";

  uint8_t text3[1000000];
  memset(text3, 'a', sizeof text3);

  uint8_t hash1[] = {
    0xba, 0x78, 0x16, 0xbf, 0x8f, 0x01, 0xcf, 0xea, 0x41, 0x41, 0x40,
    0xde, 0x5d, 0xae, 0x22, 0x23, 0xb0, 0x03, 0x61, 0xa3, 0x96, 0x17,
    0x7a, 0x9c, 0xb4, 0x10, 0xff, 0x61, 0xf2, 0x00, 0x15, 0xad
  };
  uint8_t hash2[] = {
    0x24, 0x8d, 0x6a, 0x61, 0xd2, 0x06, 0x38, 0xb8, 0xe5, 0xc0, 0x26,
    0x93, 0x0c, 0x3e, 0x60, 0x39, 0xa3, 0x3c, 0xe4, 0x59, 0x64, 0xff,
    0x21, 0x67, 0xf6, 0xec, 0xed, 0xd4, 0x19, 0xdb, 0x06, 0xc1
  };
  uint8_t hash3[] = {
    0xcd, 0xc7, 0x6e, 0x5c, 0x99, 0x14, 0xfb, 0x92, 0x81, 0xa1, 0xc7,
    0xe2, 0x84, 0xd7, 0x3e, 0x67, 0xf1, 0x80, 0x9a, 0x48, 0xa4, 0x97,
    0x20, 0x0e, 0x04, 0x6d, 0x39, 0xcc, 0xc7, 0x11, 0x2c, 0xd0
  };

  REQUIRE(ar_equal_bytes(1, SHA256_BLOCK_SIZE,
                         sha256((sizeof text1) - 1, text1).v, 1,
                         SHA256_BLOCK_SIZE, hash1));
  REQUIRE(ar_equal_bytes(1, SHA256_BLOCK_SIZE,
                         sha256((sizeof text2) - 1, text2).v, 1,
                         SHA256_BLOCK_SIZE, hash2));
  REQUIRE(ar_equal_bytes(1, SHA256_BLOCK_SIZE,
                         sha256(sizeof text3, text3).v, 1,
                         SHA256_BLOCK_SIZE, hash3));
}

#undef KIT_TEST_FILE