| Index: syzygy/refinery/types/type_repository_unittest.cc
|
| diff --git a/syzygy/refinery/types/type_repository_unittest.cc b/syzygy/refinery/types/type_repository_unittest.cc
|
| index 2dfb2534597ac2e3807e4332f61e3dd8bf4b9db1..b0d7ad254cd29e70337f4990f06ff52029127702 100644
|
| --- a/syzygy/refinery/types/type_repository_unittest.cc
|
| +++ b/syzygy/refinery/types/type_repository_unittest.cc
|
| @@ -16,6 +16,7 @@
|
|
|
| #include "base/memory/ref_counted.h"
|
| #include "gtest/gtest.h"
|
| +#include "syzygy/pe/pe_file.h"
|
| #include "syzygy/refinery/types/type.h"
|
|
|
| namespace refinery {
|
| @@ -84,6 +85,18 @@ TEST(TypeRepositoryTest, AddTypeWithId) {
|
| EXPECT_EQ(t2, repo->GetType(kId2));
|
| }
|
|
|
| +TEST(TypeRepositoryTest, GetSignature) {
|
| + pe::PEFile::Signature retrieved_sig;
|
| +
|
| + scoped_refptr<TypeRepository> repo = new TypeRepository();
|
| + ASSERT_FALSE(repo->GetModuleSignature(&retrieved_sig));
|
| +
|
| + pe::PEFile::Signature sig(L"Path", core::AbsoluteAddress(1U), 2, 3, 4);
|
| + repo = new TypeRepository(sig);
|
| + ASSERT_TRUE(repo->GetModuleSignature(&retrieved_sig));
|
| + ASSERT_EQ(sig, retrieved_sig);
|
| +}
|
| +
|
| TEST(TypeRepositoryTest, Iteration) {
|
| scoped_refptr<TypeRepository> repo = new TypeRepository();
|
|
|
|
|