AU: Class to perform delta updates.

src/platform/update_engine/test_utils.h

 // Copyright (c) 2009 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.
 
 #ifndef CHROMEOS_PLATFORM_UPDATE_ENGINE_TEST_UTILS_H__
 #define CHROMEOS_PLATFORM_UPDATE_ENGINE_TEST_UTILS_H__
 
 #include <set>
 #include <string>
 #include <vector>
 #include <gtest/gtest.h>
 #include "update_engine/action.h"
+#include "update_engine/subprocess.h"
 
 // These are some handy functions for unittests.
 
 namespace chromeos_update_engine {
 
 // Writes the data passed to path. The file at path will be overwritten if it
 // exists. Returns true on success, false otherwise.
 bool WriteFileVector(const std::string& path, const std::vector<char>& data);
 bool WriteFileString(const std::string& path, const std::string& data);
 
@@ skipping 75 matching lines @@
 void CreateExtImageAtPath(const std::string& path,
                           std::vector<std::string>* out_paths);
 
 // Verifies that for each path in paths, it exists in the filesystem under
 // parent. Also, verifies that no additional paths are present under parent.
 // Also tests properties of various files created by CreateExtImageAtPath().
 // Intentionally copies expected_paths.
 void VerifyAllPaths(const std::string& parent,
                     std::set<std::string> expected_paths);
 
+class ScopedLoopbackDeviceReleaser {
+ public:
+  explicit ScopedLoopbackDeviceReleaser(const std::string& dev) : dev_(dev) {}
+  ~ScopedLoopbackDeviceReleaser() {
+    std::vector<std::string> args;
+    args.push_back("/sbin/losetup");
+    args.push_back("-d");
+    args.push_back(dev_);
+    int return_code = 0;
+    EXPECT_TRUE(Subprocess::SynchronousExec(args, &return_code));
+    EXPECT_EQ(0, return_code);
+  }
+ private:
+  const std::string dev_;

[Daniel Erat, 2010/04/23 17:47:25]

add DISALLOW_COPY_AND_ASSIGN (especially for RAII-ish classes)

[adlr, 2010/04/23 20:50:17]

Done.

+};
+
 // Useful actions for test
 
 class NoneType;
 
 template<typename T>
 class ObjectFeederAction;
 
 template<typename T>
 class ActionTraits<ObjectFeederAction<T> > {
  public:
@@ skipping 53 matching lines @@
   static std::string StaticType() { return "ObjectCollectorAction"; }
   std::string Type() const { return StaticType(); }
   const T& object() const { return object_; }
  private:
   T object_;
 };
 
 }  // namespace chromeos_update_engine
 
 #endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_TEST_UTILS_H__