Add periodic check to see if context has been reset

rlz/lib/rlz_lib_test.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 //
 // A test application for the RLZ library.
 //
 // These tests should not be executed on the build server:
 // - They assert for the failed cases.
 // - They modify machine state (registry).
 //
 // These tests require write access to HKLM and HKCU.
 //
 // The "GGLA" brand is used to test the normal code flow of the code, and the
 // "TEST" brand is used to test the supplementary brand code code flow.
 
 #include "base/posix/eintr_wrapper.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+#include "rlz/lib/financial_ping.h"
 #include "rlz/lib/rlz_lib.h"
 #include "rlz/lib/rlz_value_store.h"
 #include "rlz/test/rlz_test_helpers.h"
 
 #if defined(OS_WIN)
 #include <Windows.h>
 #include "rlz/win/lib/machine_deal.h"
 #endif
 
 #if defined(RLZ_NETWORK_IMPLEMENTATION_CHROME_NET)
@@ skipping 377 matching lines @@
   // calling it, trying to record a stateful again should result in it being
   // recorded again.
   EXPECT_TRUE(rlz_lib::ClearAllProductEvents(rlz_lib::TOOLBAR_NOTIFIER));
   EXPECT_TRUE(rlz_lib::RecordProductEvent(rlz_lib::TOOLBAR_NOTIFIER,
       rlz_lib::IE_HOME_PAGE, rlz_lib::INSTALL));
   EXPECT_TRUE(rlz_lib::GetProductEventsAsCgi(rlz_lib::TOOLBAR_NOTIFIER,
                                              value, 50));
   EXPECT_STREQ("events=W1I", value);
 }
 
+class URLRequestRAII {
+ public:
+  URLRequestRAII(net::URLRequestContextGetter* context) {
+    rlz_lib::SetURLRequestContext(context);
+  }
+  ~URLRequestRAII() {
+    rlz_lib::SetURLRequestContext(NULL);
+  }
+};
+
 TEST_F(RlzLibTest, SendFinancialPing) {
   // We don't really check a value or result in this test. All this does is
   // attempt to ping the financial server, which you can verify in Fiddler.
   // TODO: Make this a measurable test.
 
 #if defined(RLZ_NETWORK_IMPLEMENTATION_CHROME_NET)
 #if defined(OS_MACOSX)
   base::mac::ScopedNSAutoreleasePool pool;
 #endif
 
   base::Thread::Options options;
   options.message_loop_type = base::MessageLoop::TYPE_IO;
 
   base::Thread io_thread("rlz_unittest_io_thread");
   ASSERT_TRUE(io_thread.StartWithOptions(options));
 
   scoped_refptr<net::TestURLRequestContextGetter> context =
       new net::TestURLRequestContextGetter(
           io_thread.message_loop()->message_loop_proxy());
   rlz_lib::SetURLRequestContext(context.get());
 
-  class URLRequestRAII {
-    public:
-     URLRequestRAII(net::URLRequestContextGetter* context) {
-       rlz_lib::SetURLRequestContext(context);
-     }
-     ~URLRequestRAII() {
-       rlz_lib::SetURLRequestContext(NULL);
-     }
-  };
-
   URLRequestRAII set_context(context.get());
 #endif
 
   MachineDealCodeHelper::Clear();
 #if defined(OS_WIN)
   EXPECT_TRUE(rlz_lib::MachineDealCode::Set("dcc_value"));
 #endif
 
   EXPECT_TRUE(rlz_lib::SetAccessPointRlz(rlz_lib::IETB_SEARCH_BOX,
       "TbRlzValue"));
 
   EXPECT_TRUE(rlz_lib::ClearAllProductEvents(rlz_lib::TOOLBAR_NOTIFIER));
   EXPECT_TRUE(rlz_lib::RecordProductEvent(rlz_lib::TOOLBAR_NOTIFIER,
       rlz_lib::IE_DEFAULT_SEARCH, rlz_lib::SET_TO_GOOGLE));
   EXPECT_TRUE(rlz_lib::RecordProductEvent(rlz_lib::TOOLBAR_NOTIFIER,
       rlz_lib::IE_HOME_PAGE, rlz_lib::INSTALL));
 
   rlz_lib::AccessPoint points[] =
     {rlz_lib::IETB_SEARCH_BOX, rlz_lib::NO_ACCESS_POINT,
      rlz_lib::NO_ACCESS_POINT};
 
   std::string request;
   rlz_lib::SendFinancialPing(rlz_lib::TOOLBAR_NOTIFIER, points,
       "swg", "GGLA", "SwgProductId1234", "en-UK", false,
       /*skip_time_check=*/true);
 }
 
+#if defined(RLZ_NETWORK_IMPLEMENTATION_CHROME_NET)
+
+void ResetContext() {
+  rlz_lib::SetURLRequestContext(NULL);
+}
+
+TEST_F(RlzLibTest, SendFinancialPingDuringShutdown) {
+  // rlz_lib::SendFinancialPing fails when this is set.
+  if (!rlz_lib::SupplementaryBranding::GetBrand().empty())
+    return;
+
+#if defined(OS_MACOSX)
+  base::mac::ScopedNSAutoreleasePool pool;
+#endif
+
+  base::Thread::Options options;
+  options.message_loop_type = base::MessageLoop::TYPE_IO;
+
+  base::Thread io_thread("rlz_unittest_io_thread");
+  ASSERT_TRUE(io_thread.StartWithOptions(options));
+
+  scoped_refptr<net::TestURLRequestContextGetter> context =
+      new net::TestURLRequestContextGetter(
+          io_thread.message_loop()->message_loop_proxy());
+  rlz_lib::SetURLRequestContext(context.get());
+
+  URLRequestRAII set_context(context.get());
+
+  rlz_lib::AccessPoint points[] =
+    {rlz_lib::IETB_SEARCH_BOX, rlz_lib::NO_ACCESS_POINT,
+     rlz_lib::NO_ACCESS_POINT};
+  rlz_lib::test::ResetSendFinancialPingInterrupted();
+  EXPECT_FALSE(rlz_lib::test::WasSendFinancialPingInterrupted());
+
+  base::MessageLoop loop;
+  loop.PostTask(FROM_HERE, base::Bind(&ResetContext));
+  std::string request;
+  EXPECT_FALSE(rlz_lib::SendFinancialPing(rlz_lib::TOOLBAR_NOTIFIER, points,
+      "swg", "GGLA", "SwgProductId1234", "en-UK", false,
+      /*skip_time_check=*/true));
+
+  EXPECT_TRUE(rlz_lib::test::WasSendFinancialPingInterrupted());
+  rlz_lib::test::ResetSendFinancialPingInterrupted();
+}
+#endif
+
 TEST_F(RlzLibTest, ClearProductState) {
   MachineDealCodeHelper::Clear();
 
   EXPECT_TRUE(rlz_lib::SetAccessPointRlz(rlz_lib::IETB_SEARCH_BOX,
       "TbRlzValue"));
   EXPECT_TRUE(rlz_lib::SetAccessPointRlz(rlz_lib::GD_DESKBAND,
       "GdbRlzValue"));
 
   rlz_lib::AccessPoint points[] =
       { rlz_lib::IETB_SEARCH_BOX, rlz_lib::NO_ACCESS_POINT };
@@ skipping 396 matching lines @@
   int mkdir_result = mkdir(rlz_lib::testing::RlzStoreFilenameStr().c_str(),
                            0500);
   ASSERT_EQ(0, mkdir_result);
 
   rlz_lib::SupplementaryBranding branding("TEST");
   EXPECT_FALSE(rlz_lib::RecordProductEvent(rlz_lib::TOOLBAR_NOTIFIER,
       rlz_lib::IE_DEFAULT_SEARCH, rlz_lib::INSTALL));
 }
 
 #endif