| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "courgette/base_test_unittest.h" | 5 #include "courgette/base_test_unittest.h" |
| 6 | 6 |
| 7 #include <string> | 7 #include <string> |
| 8 | 8 |
| 9 #include "base/basictypes.h" | 9 #include "base/basictypes.h" |
| 10 #include "courgette/courgette.h" | 10 #include "courgette/courgette.h" |
| 11 #include "courgette/streams.h" | 11 #include "courgette/streams.h" |
| 12 #include "courgette/third_party/bsdiff.h" |
| 12 | 13 |
| 13 class VersioningTest : public BaseTest { | 14 class VersioningTest : public BaseTest { |
| 14 public: | 15 public: |
| 15 void TestApplyingOldPatch(const char* src_file, | 16 void TestApplyingOldBsDiffPatch(const char* src_file, |
| 16 const char* patch_file, | 17 const char* patch_file, |
| 17 const char* expected_file) const; | 18 const char* expected_file) const; |
| 18 }; | 19 }; |
| 19 | 20 |
| 20 void VersioningTest::TestApplyingOldPatch(const char* src_file, | 21 void VersioningTest::TestApplyingOldBsDiffPatch( |
| 21 const char* patch_file, | 22 const char* src_file, |
| 22 const char* expected_file) const { | 23 const char* patch_file, |
| 24 const char* expected_file) const { |
| 23 std::string old_buffer = FileContents(src_file); | 25 std::string old_buffer = FileContents(src_file); |
| 24 std::string new_buffer = FileContents(patch_file); | 26 std::string new_buffer = FileContents(patch_file); |
| 25 std::string expected_buffer = FileContents(expected_file); | 27 std::string expected_buffer = FileContents(expected_file); |
| 26 | 28 |
| 27 courgette::SourceStream old_stream; | 29 courgette::SourceStream old_stream; |
| 28 courgette::SourceStream patch_stream; | 30 courgette::SourceStream patch_stream; |
| 29 old_stream.Init(old_buffer); | 31 old_stream.Init(old_buffer); |
| 30 patch_stream.Init(new_buffer); | 32 patch_stream.Init(new_buffer); |
| 31 | 33 |
| 32 courgette::SinkStream generated_stream; | 34 courgette::SinkStream generated_stream; |
| 33 | 35 |
| 34 courgette::Status status = | 36 courgette::BSDiffStatus status = courgette::ApplyBinaryPatch( |
| 35 courgette::ApplyEnsemblePatch(&old_stream, | 37 &old_stream, &patch_stream, &generated_stream); |
| 36 &patch_stream, | |
| 37 &generated_stream); | |
| 38 | 38 |
| 39 EXPECT_EQ(status, courgette::C_OK); | 39 EXPECT_EQ(status, courgette::OK); |
| 40 | 40 |
| 41 size_t expected_length = expected_buffer.size(); | 41 size_t expected_length = expected_buffer.size(); |
| 42 size_t generated_length = generated_stream.Length(); | 42 size_t generated_length = generated_stream.Length(); |
| 43 | 43 |
| 44 EXPECT_EQ(generated_length, expected_length); | 44 EXPECT_EQ(generated_length, expected_length); |
| 45 EXPECT_EQ(0, memcmp(generated_stream.Buffer(), | 45 EXPECT_EQ(0, memcmp(generated_stream.Buffer(), |
| 46 expected_buffer.c_str(), | 46 expected_buffer.c_str(), |
| 47 expected_length)); | 47 expected_length)); |
| 48 } | 48 } |
| 49 | 49 |
| 50 | 50 TEST_F(VersioningTest, BsDiff) { |
| 51 TEST_F(VersioningTest, All) { | 51 TestApplyingOldBsDiffPatch("setup1.exe", "setup1-setup2.v1.bsdiff", |
| 52 TestApplyingOldPatch("setup1.exe", "setup1-setup2.v1.patch", "setup2.exe"); | 52 "setup2.exe"); |
| 53 TestApplyingOldPatch("chrome64_1.exe", "chrome64-1-2.v1.patch", | 53 TestApplyingOldBsDiffPatch("chrome64_1.exe", "chrome64-1-2.v1.bsdiff", |
| 54 "chrome64_2.exe"); | 54 "chrome64_2.exe"); |
| 55 | 55 |
| 56 // We also need a way to test that newly generated patches are appropriately | 56 // We also need a way to test that newly generated patches are appropriately |
| 57 // applicable by older clients... not sure of the best way to do that. | 57 // applicable by older clients... not sure of the best way to do that. |
| 58 } | 58 } |
| OLD | NEW |