AU: Update Downloader to support our image formats.

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 "base/scoped_ptr.h"
 #include "update_engine/action.h"
 #include "update_engine/subprocess.h"
+#include "update_engine/utils.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 88 matching lines @@
     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_;
   DISALLOW_COPY_AND_ASSIGN(ScopedLoopbackDeviceReleaser);
 };
 
+class ScopedTempFile {
+ public:
+  ScopedTempFile() {
+    EXPECT_TRUE(utils::MakeTempFile("/tmp/update_engine_test_temp_file.XXXXXX",
+                                    &path_,
+                                    NULL));
+    unlinker_.reset(new ScopedPathUnlinker(path_));
+  }
+  const std::string& GetPath() { return path_; }
+ private:
+  std::string path_;
+  scoped_ptr<ScopedPathUnlinker> unlinker_;
+};
+
 // Useful actions for test
 
 class NoneType;
 
 template<typename T>
 class ObjectFeederAction;
 
 template<typename T>
 class ActionTraits<ObjectFeederAction<T> > {
  public:
@@ skipping 52 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__
+#endif  // CHROMEOS_PLATFORM_UPDATE_ENGINE_TEST_UTILS_H__