Properly honor base::SharedMemory semantics for name="" to mean...

base/stats_table_unittest.cc

 // Copyright (c) 2006-2008 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.
 
 #include "base/multiprocess_test.h"
 #include "base/platform_thread.h"
 #include "base/simple_thread.h"
+#include "base/shared_memory.h"
 #include "base/stats_table.h"
 #include "base/stats_counters.h"
 #include "base/string_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/multiprocess_func_list.h"
 
 #if defined(OS_WIN)
 #include <process.h>
 #include <windows.h>
 #endif
 
 namespace base {
 
 class StatsTableTest : public MultiProcessTest {
+ public:
+  void DeleteShmem(std::string name) {
+    base::SharedMemory mem;
+    mem.Delete(UTF8ToWide(name));
+  }
 };
 
 // Open a StatsTable and verify that we can write to each of the
 // locations in the table.
 TEST_F(StatsTableTest, VerifySlots) {
   const std::string kTableName = "VerifySlotsStatTable";
   const int kMaxThreads = 1;
   const int kMaxCounter = 5;
+  DeleteShmem(kTableName);
   StatsTable table(kTableName, kMaxThreads, kMaxCounter);
 
   // Register a single thread.
   std::string thread_name = "mainThread";
   int slot_id = table.RegisterThread(thread_name);
   EXPECT_TRUE(slot_id);
 
   // Fill up the table with counters.
   std::string counter_base_name = "counter";
   for (int index=0; index < kMaxCounter; index++) {
     std::string counter_name = counter_base_name;
     StringAppendF(&counter_name, "counter.ctr%d", index);
     int counter_id = table.FindCounter(counter_name);
     EXPECT_GT(counter_id, 0);
   }
 
   // Try to allocate an additional thread.  Verify it fails.
   slot_id = table.RegisterThread("too many threads");
   EXPECT_EQ(slot_id, 0);
 
   // Try to allocate an additional counter.  Verify it fails.
   int counter_id = table.FindCounter(counter_base_name);
   EXPECT_EQ(counter_id, 0);
+
+  DeleteShmem(kTableName);
 }
 
 // CounterZero will continually be set to 0.
 const std::string kCounterZero = "CounterZero";
 // Counter1313 will continually be set to 1313.
 const std::string kCounter1313 = "Counter1313";
 // CounterIncrement will be incremented each time.
 const std::string kCounterIncrement = "CounterIncrement";
 // CounterDecrement will be decremented each time.
 const std::string kCounterDecrement = "CounterDecrement";
@@ skipping 34 matching lines @@
     PlatformThread::Sleep(index % 10);   // short wait
   }
 }
 
 // Create a few threads and have them poke on their counters.
 TEST_F(StatsTableTest, MultipleThreads) {
   // Create a stats table.
   const std::string kTableName = "MultipleThreadStatTable";
   const int kMaxThreads = 20;
   const int kMaxCounter = 5;
+  DeleteShmem(kTableName);
   StatsTable table(kTableName, kMaxThreads, kMaxCounter);
   StatsTable::set_current(&table);
 
   EXPECT_EQ(0, table.CountThreadsRegistered());
 
   // Spin up a set of threads to go bang on the various counters.
   // After we join the threads, we'll make sure the counters
   // contain the values we expected.
   StatsTableThread* threads[kMaxThreads];
 
@@ skipping 25 matching lines @@
   name = "c:" + kCounterIncrement;
   EXPECT_EQ(kMaxThreads * kThreadLoops,
       table.GetCounterValue(name));
   name = "c:" + kCounterDecrement;
   EXPECT_EQ(-kMaxThreads * kThreadLoops,
       table.GetCounterValue(name));
   name = "c:" + kCounterMixed;
   EXPECT_EQ((kMaxThreads % 2) * kThreadLoops,
       table.GetCounterValue(name));
   EXPECT_EQ(0, table.CountThreadsRegistered());
+
+  DeleteShmem(kTableName);
 }
 
-const std::string kTableName = "MultipleProcessStatTable";
+const std::string kMPTableName = "MultipleProcessStatTable";
 
 MULTIPROCESS_TEST_MAIN(StatsTableMultipleProcessMain) {
   // Each process will open the shared memory and set counters
   // concurrently in a loop.  We'll use some pauses to
   // mixup the scheduling.
 
-  StatsTable table(kTableName, 0, 0);
+  StatsTable table(kMPTableName, 0, 0);
   StatsTable::set_current(&table);
   StatsCounter zero_counter(kCounterZero);
   StatsCounter lucky13_counter(kCounter1313);
   StatsCounter increment_counter(kCounterIncrement);
   StatsCounter decrement_counter(kCounterDecrement);
   for (int index = 0; index < kThreadLoops; index++) {
     zero_counter.Set(0);
     lucky13_counter.Set(1313);
     increment_counter.Increment();
     decrement_counter.Decrement();
     PlatformThread::Sleep(index % 10);   // short wait
   }
   return 0;
 }
 
 // Create a few processes and have them poke on their counters.
 TEST_F(StatsTableTest, MultipleProcesses) {
   // Create a stats table.
-  const std::string kTableName = "MultipleProcessStatTable";
   const int kMaxProcs = 20;
   const int kMaxCounter = 5;
-  StatsTable table(kTableName, kMaxProcs, kMaxCounter);
+  DeleteShmem(kMPTableName);
+  StatsTable table(kMPTableName, kMaxProcs, kMaxCounter);
   StatsTable::set_current(&table);
-
   EXPECT_EQ(0, table.CountThreadsRegistered());
 
   // Spin up a set of processes to go bang on the various counters.
   // After we join the processes, we'll make sure the counters
   // contain the values we expected.
   ProcessHandle procs[kMaxProcs];
 
   // Spawn the processes.
   for (int16 index = 0; index < kMaxProcs; index++) {
     procs[index] = this->SpawnChild(L"StatsTableMultipleProcessMain");
@@ skipping 17 matching lines @@
   name = "c:" + kCounter1313;
   EXPECT_EQ(1313 * kMaxProcs,
       table.GetCounterValue(name));
   name = "c:" + kCounterIncrement;
   EXPECT_EQ(kMaxProcs * kThreadLoops,
       table.GetCounterValue(name));
   name = "c:" + kCounterDecrement;
   EXPECT_EQ(-kMaxProcs * kThreadLoops,
       table.GetCounterValue(name));
   EXPECT_EQ(0, table.CountThreadsRegistered());
+
+  DeleteShmem(kMPTableName);
 }
 
 class MockStatsCounter : public StatsCounter {
  public:
   MockStatsCounter(const std::string& name)
       : StatsCounter(name) {}
   int* Pointer() { return GetPtr(); }
 };
 
 // Test some basic StatsCounter operations
 TEST_F(StatsTableTest, StatsCounter) {
   // Create a stats table.
   const std::string kTableName = "StatTable";
   const int kMaxThreads = 20;
   const int kMaxCounter = 5;
+  DeleteShmem(kTableName);
   StatsTable table(kTableName, kMaxThreads, kMaxCounter);
   StatsTable::set_current(&table);
 
   MockStatsCounter foo("foo");
 
   // Test initial state.
   EXPECT_TRUE(foo.Enabled());
   EXPECT_NE(foo.Pointer(), static_cast<int*>(0));
   EXPECT_EQ(0, table.GetCounterValue("c:foo"));
   EXPECT_EQ(0, *(foo.Pointer()));
@@ skipping 16 matching lines @@
   foo.Set(0);
   EXPECT_EQ(0, table.GetCounterValue("c:foo"));
 
   // Test Decrement.
   foo.Decrement(1);
   EXPECT_EQ(-1, table.GetCounterValue("c:foo"));
   foo.Decrement(0);
   EXPECT_EQ(-1, table.GetCounterValue("c:foo"));
   foo.Decrement(-1);
   EXPECT_EQ(0, table.GetCounterValue("c:foo"));
+
+  DeleteShmem(kTableName);
 }
 
 class MockStatsCounterTimer : public StatsCounterTimer {
  public:
   MockStatsCounterTimer(const std::string& name)
       : StatsCounterTimer(name) {}
 
   TimeTicks start_time() { return start_time_; }
   TimeTicks stop_time() { return stop_time_; }
 };
@@ skipping 57 matching lines @@
   EXPECT_EQ(2, table.GetCounterValue("c:baz"));
   EXPECT_LE(1000, table.GetCounterValue("t:baz"));
 }
 
 // Test some basic StatsScope operations
 TEST_F(StatsTableTest, StatsScope) {
   // Create a stats table.
   const std::string kTableName = "StatTable";
   const int kMaxThreads = 20;
   const int kMaxCounter = 5;
+  DeleteShmem(kTableName);
   StatsTable table(kTableName, kMaxThreads, kMaxCounter);
   StatsTable::set_current(&table);
 
   StatsCounterTimer foo("foo");
   StatsRate bar("bar");
 
   // Test initial state.
   EXPECT_EQ(0, table.GetCounterValue("t:foo"));
   EXPECT_EQ(0, table.GetCounterValue("t:bar"));
   EXPECT_EQ(0, table.GetCounterValue("c:bar"));
@@ skipping 10 matching lines @@
 
   // Try a second scope.
   {
     StatsScope<StatsCounterTimer> timer(foo);
     StatsScope<StatsRate> timer2(bar);
     PlatformThread::Sleep(500);
   }
   EXPECT_LE(1000, table.GetCounterValue("t:foo"));
   EXPECT_LE(1000, table.GetCounterValue("t:bar"));
   EXPECT_EQ(2, table.GetCounterValue("c:bar"));
+
+  DeleteShmem(kTableName);
 }
 
 }  // namespace base