Add mutator test working.

1 files changed, 31 insertions, 2 deletions

diff --git a/xdelta3/testing/regtest.cc b/xdelta3/testing/regtest.cc
index 12499ba..487a407 100644
--- a/xdelta3/testing/regtest.cc
+++ b/xdelta3/testing/regtest.cc
@@ -268,9 +268,7 @@ void TestModifyMutator() {
   for (size_t i = 0; i < SIZEOF_ARRAY(test_cases); i++) {
     ChangeList cl1;
     cl1.push_back(Change(Change::MODIFY, test_cases[i].size, test_cases[i].addr));
-    spec0.Print();
     spec0.ModifyTo(ChangeListMutator(cl1), &spec1);
-    spec1.Print();
     
     size_t diff = CmpDifferentBytes(spec0, spec1);
     CHECK_LE(diff, test_cases[i].size);
@@ -280,12 +278,43 @@ void TestModifyMutator() {
   }
 }
 
+void TestAddMutator() {
+  MTRandom rand;
+  FileSpec spec0(&rand);
+  FileSpec spec1(&rand);
+
+  spec0.GenerateFixedSize(Constants::BLOCK_SIZE * 2);
+
+  struct {
+    size_t size;
+    size_t addr;
+  } test_cases[] = {
+    { 1, 0 },
+    { 1, 1 },
+    { 1, Constants::BLOCK_SIZE - 1 },
+    { 1, Constants::BLOCK_SIZE },
+    { 1, Constants::BLOCK_SIZE + 1},
+    { 1, 2 * Constants::BLOCK_SIZE },
+  };
+
+  for (size_t i = 0; i < SIZEOF_ARRAY(test_cases); i++) {
+    ChangeList cl1;
+    cl1.push_back(Change(Change::ADD, test_cases[i].size, test_cases[i].addr));
+    spec0.Print();
+    spec0.ModifyTo(ChangeListMutator(cl1), &spec1);
+    spec1.Print();
+    
+    InMemoryEncodeDecode(spec0, spec1);
+  }
+}
+
 int main(int argc, char **argv) {
 #define TEST(x) cerr << #x << "..." << endl; x()
   TEST(TestRandomNumbers);
   TEST(TestRandomFile);
   TEST(TestFirstByte);
   TEST(TestModifyMutator);
+  TEST(TestAddMutator);
   return 0;
 }