1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include <string> 6 7 #include "base/base64.h" 8 #include "base/file_util.h" 9 #include "base/files/file_path.h" 10 #include "base/files/scoped_temp_dir.h" 11 #include "chrome/test/chromedriver/chrome/status.h" 12 #include "chrome/test/chromedriver/util.h" 13 #include "testing/gtest/include/gtest/gtest.h" 14 15 TEST(UnzipSoleFile, Entry) { 16 base::ScopedTempDir temp_dir; 17 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 18 std::string data; 19 // A zip entry sent from a Java WebDriver client (v2.20) that contains a 20 // file with the contents "COW\n". 21 const char* kBase64ZipEntry = 22 "UEsDBBQACAAIAJpyXEAAAAAAAAAAAAAAAAAEAAAAdGVzdHP2D+" 23 "cCAFBLBwi/wAzGBgAAAAQAAAA="; 24 ASSERT_TRUE(base::Base64Decode(kBase64ZipEntry, &data)); 25 base::FilePath file; 26 Status status = UnzipSoleFile(temp_dir.path(), data, &file); 27 ASSERT_EQ(kOk, status.code()) << status.message(); 28 std::string contents; 29 ASSERT_TRUE(base::ReadFileToString(file, &contents)); 30 ASSERT_STREQ("COW\n", contents.c_str()); 31 } 32 33 TEST(UnzipSoleFile, Archive) { 34 base::ScopedTempDir temp_dir; 35 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 36 std::string data; 37 // A zip archive sent from a Python WebDriver client that contains a 38 // file with the contents "COW\n". 39 const char* kBase64ZipArchive = 40 "UEsDBBQAAAAAAMROi0K/wAzGBAAAAAQAAAADAAAAbW9vQ09XClBLAQIUAxQAAAAAAMROi0K/" 41 "wAzGBAAAAAQAAAADAAAAAAAAAAAAAACggQAAAABtb29QSwUGAAAAAAEAAQAxAAAAJQAAAAA" 42 "A"; 43 ASSERT_TRUE(base::Base64Decode(kBase64ZipArchive, &data)); 44 base::FilePath file; 45 Status status = UnzipSoleFile(temp_dir.path(), data, &file); 46 ASSERT_EQ(kOk, status.code()) << status.message(); 47 std::string contents; 48 ASSERT_TRUE(base::ReadFileToString(file, &contents)); 49 ASSERT_STREQ("COW\n", contents.c_str()); 50 } 51
