Stop infinite restarting

child_job_unittest.cc

 // Copyright (c) 2009-2010 The Chromium OS 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 "login_manager/child_job.h"
 
 #include <gtest/gtest.h>
 #include <unistd.h>
 
 #include <base/command_line.h>
@@ skipping 56 matching lines @@
 
   std::vector<std::string> data = job.ExtractArgvForTest();
   std::vector<std::string>::const_iterator it  = data.begin();
   for (int i = 1; i < my_argc; ++i) {
     EXPECT_EQ(my_argv[i], *it);
     ++it;
   }
   EXPECT_TRUE(data.end() == it) << *it;
 }
 
+TEST(SetUidExecJobTest, ShouldStopTest) {
+  const char *my_argv[] = { "zero",
+                            "one",
+                            "two" };
+  const int my_argc = 3;
+  CommandLine cl(my_argc, my_argv);
+
+  SetUidExecJob job(&cl, NULL, false);
+  job.RecordTime();
+  EXPECT_TRUE(job.ShouldStop());
+}
+
+TEST(SetUidExecJobTest, ShouldNotStopTest) {
+  const char *my_argv[] = { "zero",
+                            "one",
+                            "two" };
+  const int my_argc = 3;
+  CommandLine cl(my_argc, my_argv);
+
+  SetUidExecJob job(&cl, NULL, false);
+  job.RecordTime();
+  sleep(1);
+  EXPECT_FALSE(job.ShouldStop());
+}
+
 }  // namespace login_manager