Add a comment and some tests including the NULL byte.
[oota-llvm.git] / unittests / Support / MD5Test.cpp
1 //===- llvm/unittest/Support/MD5Test.cpp - MD5 tests ----------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 //
10 // This file implements unit tests for the MD5 functions.
11 //
12 //===----------------------------------------------------------------------===//
13
14 #include "llvm/ADT/ArrayRef.h"
15 #include "llvm/ADT/SmallString.h"
16 #include "llvm/Support/MD5.h"
17 #include "gtest/gtest.h"
18
19 using namespace llvm;
20
21 namespace {
22 /// \brief Tests an arbitrary set of bytes passed as \p Input.
23 void TestMD5Sum(ArrayRef<unsigned char> Input, StringRef Final) {
24   MD5 Hash;
25   Hash.update(Input);
26   MD5::MD5Result MD5Res;
27   Hash.final(MD5Res);
28   SmallString<32> Res;
29   MD5::stringifyResult(MD5Res, Res);
30   EXPECT_EQ(Res, Final);
31 }
32
33 TEST(MD5Test, MD5) {
34   TestMD5Sum(ArrayRef<unsigned char>((const unsigned char *)"", (size_t) 0),
35              "d41d8cd98f00b204e9800998ecf8427e");
36   TestMD5Sum(ArrayRef<unsigned char>((const unsigned char *)"a", (size_t) 1),
37              "0cc175b9c0f1b6a831c399e269772661");
38   TestMD5Sum(ArrayRef<unsigned char>(
39                  (const unsigned char *)"abcdefghijklmnopqrstuvwxyz",
40                  (size_t) 26),
41              "c3fcd3d76192e4007dfb496cca67e13b");
42   TestMD5Sum(ArrayRef<unsigned char>((const unsigned char *)"\0", (size_t) 1),
43              "93b885adfe0da089cdf634904fd59f71");
44   TestMD5Sum(ArrayRef<unsigned char>((const unsigned char *)"a\0", (size_t) 2),
45              "4144e195f46de78a3623da7364d04f11");
46   TestMD5Sum(ArrayRef<unsigned char>(
47                  (const unsigned char *)"abcdefghijklmnopqrstuvwxyz\0",
48                  (size_t) 27),
49              "81948d1f1554f58cd1a56ebb01f808cb");
50 }
51 }