courgette/ensemble_unittest.cc

   1 // Copyright (c) 2011 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 "courgette/base_test_unittest.h"
   6 #include "courgette/courgette.h"
   7 #include "courgette/streams.h"
   8
   9 class EnsembleTest : public BaseTest {
  10  public:
  11
  12   void TestEnsemble(std::string src_bytes, std::string tgt_bytes) const;
  13
  14   void PeEnsemble() const;
  15   void Pe64Ensemble() const;
  16   void Elf32Ensemble() const;
  17 };
  18
  19 void EnsembleTest::TestEnsemble(std::string src_bytes,
  20                                 std::string tgt_bytes) const {
  21
  22   courgette::SourceStream source;
  23   courgette::SourceStream target;
  24
  25   source.Init(src_bytes);
  26   target.Init(tgt_bytes);
  27
  28   courgette::SinkStream patch_sink;
  29
  30   courgette::Status status;
  31
  32   status = courgette::GenerateEnsemblePatch(&source, &target, &patch_sink);
  33   EXPECT_EQ(courgette::C_OK, status);
  34
  35   courgette::SourceStream patch_source;
  36   patch_source.Init(patch_sink.Buffer(), patch_sink.Length());
  37
  38   courgette::SinkStream patch_result;
  39
  40   status = courgette::ApplyEnsemblePatch(&source, &patch_source, &patch_result);
  41   EXPECT_EQ(courgette::C_OK, status);
  42
  43   EXPECT_EQ(target.OriginalLength(), patch_result.Length());
  44   EXPECT_FALSE(memcmp(target.Buffer(),
  45                       patch_result.Buffer(),
  46                       target.OriginalLength()));
  47 }
  48
  49 void EnsembleTest::Elf32Ensemble() const {
  50   std::list<std::string> src_ensemble;
  51   std::list<std::string> tgt_ensemble;
  52
  53   src_ensemble.push_back("elf-32-1");
  54
  55   tgt_ensemble.push_back("elf-32-2");
  56
  57   std::string src_bytes = FilesContents(src_ensemble);
  58   std::string tgt_bytes = FilesContents(tgt_ensemble);
  59
  60   src_bytes = "aaabbbccc" + src_bytes + "dddeeefff";
  61   tgt_bytes = "aaagggccc" + tgt_bytes + "dddeeefff";
  62
  63   TestEnsemble(src_bytes, tgt_bytes);
  64 }
  65
  66 void EnsembleTest::PeEnsemble() const {
  67   std::list<std::string> src_ensemble;
  68   std::list<std::string> tgt_ensemble;
  69
  70   src_ensemble.push_back("en-US.dll");
  71   src_ensemble.push_back("setup1.exe");
  72
  73   tgt_ensemble.push_back("en-US.dll");
  74   tgt_ensemble.push_back("setup2.exe");
  75
  76   std::string src_bytes = FilesContents(src_ensemble);
  77   std::string tgt_bytes = FilesContents(tgt_ensemble);
  78
  79   src_bytes = "aaabbbccc" + src_bytes + "dddeeefff";
  80   tgt_bytes = "aaagggccc" + tgt_bytes + "dddeeefff";
  81
  82   TestEnsemble(src_bytes, tgt_bytes);
  83 }
  84
  85 void EnsembleTest::Pe64Ensemble() const {
  86   std::list<std::string> src_ensemble;
  87   std::list<std::string> tgt_ensemble;
  88
  89   src_ensemble.push_back("en-US-64.dll");
  90   src_ensemble.push_back("chrome64_1.exe");
  91   src_ensemble.push_back("pe-64.exe");
  92
  93   tgt_ensemble.push_back("en-US-64.dll");
  94   tgt_ensemble.push_back("chrome64_2.exe");
  95   tgt_ensemble.push_back("pe-64.exe");
  96
  97   std::string src_bytes = FilesContents(src_ensemble);
  98   std::string tgt_bytes = FilesContents(tgt_ensemble);
  99
 100   src_bytes = "aaabbbccc" + src_bytes + "dddeeefff";
 101   tgt_bytes = "aaagggccc" + tgt_bytes + "dddeeefff";
 102
 103   TestEnsemble(src_bytes, tgt_bytes);
 104 }
 105
 106 // Ensemble tests still take too long on Windows so disabling for now
 107 // TODO(dgarrett) http://code.google.com/p/chromium/issues/detail?id=101614
 108
 109 TEST_F(EnsembleTest, DISABLED_PE) {
 110   PeEnsemble();
 111 }
 112
 113 TEST_F(EnsembleTest, DISABLED_PE64) {
 114   Pe64Ensemble();
 115 }
 116
 117 TEST_F(EnsembleTest, DISABLED_Elf32) {
 118   Elf32Ensemble();
 119 }