Index: chrome/installer/util/product_unittest.cc |
diff --git a/chrome/installer/util/product_unittest.cc b/chrome/installer/util/product_unittest.cc |
index 6adaf01e6ae33bcbabacd579d0777d5f7ea8a589..6763bc67f9ba5ed2bb67bc3c1b21e65a1260604f 100644 |
--- a/chrome/installer/util/product_unittest.cc |
+++ b/chrome/installer/util/product_unittest.cc |
@@ -61,8 +61,7 @@ TEST(ProductTest, ProductInstallBasic) { |
// There should be no installed version in the registry. |
machine_state.Initialize(); |
- EXPECT_TRUE(machine_state.GetProductState( |
- system_level, distribution->GetType()) == NULL); |
+ EXPECT_EQ(nullptr, machine_state.GetProductState(system_level)); |
// Let's pretend chrome is installed. |
RegKey version_key(root, distribution->GetVersionKey().c_str(), |
@@ -78,9 +77,9 @@ TEST(ProductTest, ProductInstallBasic) { |
// We started out with a non-msi product. |
machine_state.Initialize(); |
const installer::ProductState* chrome_state = |
- machine_state.GetProductState(system_level, distribution->GetType()); |
- EXPECT_TRUE(chrome_state != NULL); |
- if (chrome_state != NULL) { |
+ machine_state.GetProductState(system_level); |
+ EXPECT_NE(nullptr, chrome_state); |
+ if (chrome_state) { |
EXPECT_EQ(chrome_state->version(), current_version); |
EXPECT_FALSE(chrome_state->is_msi()); |
} |
@@ -94,10 +93,9 @@ TEST(ProductTest, ProductInstallBasic) { |
// Set the MSI marker, refresh, and verify that we now see the MSI marker. |
EXPECT_TRUE(product->SetMsiMarker(system_level, true)); |
machine_state.Initialize(); |
- chrome_state = |
- machine_state.GetProductState(system_level, distribution->GetType()); |
- EXPECT_TRUE(chrome_state != NULL); |
- if (chrome_state != NULL) |
+ chrome_state = machine_state.GetProductState(system_level); |
+ EXPECT_NE(nullptr, chrome_state); |
+ if (chrome_state) |
EXPECT_TRUE(chrome_state->is_msi()); |
} |
} |