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(); |