Inside Bitcoin

751      compaction_count++;
752      Put("a", "begin");
753      Put("z", "end");
754      dbfull()->TEST_CompactMemTable();
755    }
756
757    // Step 2: clear level 1 if necessary.
758    dbfull()->TEST_CompactRange(1, nullptr, nullptr);
759    ASSERT_EQ(NumTableFilesAtLevel(0), 1);
760    ASSERT_EQ(NumTableFilesAtLevel(1), 0);
761    ASSERT_EQ(NumTableFilesAtLevel(2), 1);
762
763    // Step 3: many read misses must not schedule any compaction.
764    for (int i = 0; i < 1000; i++) {
765      ASSERT_EQ("NOT_FOUND", Get("missing"));
766    }
767    DelayMilliseconds(1000);
768    ASSERT_EQ(NumTableFilesAtLevel(0), 1);
769    ASSERT_EQ(NumTableFilesAtLevel(1), 0);
770    ASSERT_EQ(NumTableFilesAtLevel(2), 1);
771
772    // Step 4: a manual compaction still moves the L0 file down.
773    dbfull()->TEST_CompactRange(0, nullptr, nullptr);
774    ASSERT_EQ(NumTableFilesAtLevel(0), 0);
775  } while (ChangeOptions());
776}
777
778TEST(DBTest, IterEmpty) {
779  Iterator* iter = db_->NewIterator(ReadOptions());
780
781  iter->SeekToFirst();
782  ASSERT_EQ(IterStatus(iter), "(invalid)");
783
784  iter->SeekToLast();
785  ASSERT_EQ(IterStatus(iter), "(invalid)");
786
787  iter->Seek("foo");
788  ASSERT_EQ(IterStatus(iter), "(invalid)");
789
790  delete iter;
791}
792
793TEST(DBTest, IterSingle) {
794  ASSERT_OK(Put("a", "va"));
795  Iterator* iter = db_->NewIterator(ReadOptions());
796
797  iter->SeekToFirst();
798  ASSERT_EQ(IterStatus(iter), "a->va");
799  iter->Next();
800  ASSERT_EQ(IterStatus(iter), "(invalid)");
801  iter->SeekToFirst();
802  ASSERT_EQ(IterStatus(iter), "a->va");
803  iter->Prev();
804  ASSERT_EQ(IterStatus(iter), "(invalid)");
805
806  iter->SeekToLast();
807  ASSERT_EQ(IterStatus(iter), "a->va");
808  iter->Next();
809  ASSERT_EQ(IterStatus(iter), "(invalid)");
810  iter->SeekToLast();
811  ASSERT_EQ(IterStatus(iter), "a->va");
812  iter->Prev();
813  ASSERT_EQ(IterStatus(iter), "(invalid)");
814
815  iter->Seek("");
816  ASSERT_EQ(IterStatus(iter), "a->va");
817  iter->Next();
818  ASSERT_EQ(IterStatus(iter), "(invalid)");
819
820  iter->Seek("a");
821  ASSERT_EQ(IterStatus(iter), "a->va");
822  iter->Next();
823  ASSERT_EQ(IterStatus(iter), "(invalid)");
824
825  iter->Seek("b");
826  ASSERT_EQ(IterStatus(iter), "(invalid)");
827
828  delete iter;
829}
830
831TEST(DBTest, IterMulti) {
832  ASSERT_OK(Put("a", "va"));
833  ASSERT_OK(Put("b", "vb"));
834  ASSERT_OK(Put("c", "vc"));
835  Iterator* iter = db_->NewIterator(ReadOptions());
836
837  iter->SeekToFirst();
838  ASSERT_EQ(IterStatus(iter), "a->va");
839  iter->Next();
840  ASSERT_EQ(IterStatus(iter), "b->vb");
841  iter->Next();
842  ASSERT_EQ(IterStatus(iter), "c->vc");
843  iter->Next();
844  ASSERT_EQ(IterStatus(iter), "(invalid)");
845  iter->SeekToFirst();
846  ASSERT_EQ(IterStatus(iter), "a->va");
847  iter->Prev();
848  ASSERT_EQ(IterStatus(iter), "(invalid)");
849
850  iter->SeekToLast();
851  ASSERT_EQ(IterStatus(iter), "c->vc");
852  iter->Prev();
853  ASSERT_EQ(IterStatus(iter), "b->vb");
854  iter->Prev();
855  ASSERT_EQ(IterStatus(iter), "a->va");
856  iter->Prev();
857  ASSERT_EQ(IterStatus(iter), "(invalid)");
858  iter->SeekToLast();
859  ASSERT_EQ(IterStatus(iter), "c->vc");
860  iter->Next();
861  ASSERT_EQ(IterStatus(iter), "(invalid)");
862
863  iter->Seek("");
864  ASSERT_EQ(IterStatus(iter), "a->va");
865  iter->Seek("a");
866  ASSERT_EQ(IterStatus(iter), "a->va");
867  iter->Seek("ax");
868  ASSERT_EQ(IterStatus(iter), "b->vb");
869  iter->Seek("b");
870  ASSERT_EQ(IterStatus(iter), "b->vb");
871  iter->Seek("z");
872  ASSERT_EQ(IterStatus(iter), "(invalid)");
873
874  // Switch from reverse to forward
875  iter->SeekToLast();
876  iter->Prev();
877  iter->Prev();
878  iter->Next();
879  ASSERT_EQ(IterStatus(iter), "b->vb");
880
881  // Switch from forward to reverse
882  iter->SeekToFirst();
883  iter->Next();
884  iter->Next();
885  iter->Prev();
886  ASSERT_EQ(IterStatus(iter), "b->vb");
887
888  // Make sure iter stays at snapshot
889  ASSERT_OK(Put("a", "va2"));
890  ASSERT_OK(Put("a2", "va3"));
891  ASSERT_OK(Put("b", "vb2"));
892  ASSERT_OK(Put("c", "vc2"));
893  ASSERT_OK(Delete("b"));
894  iter->SeekToFirst();
895  ASSERT_EQ(IterStatus(iter), "a->va");
896  iter->Next();
897  ASSERT_EQ(IterStatus(iter), "b->vb");
898  iter->Next();
899  ASSERT_EQ(IterStatus(iter), "c->vc");
900  iter->Next();
901  ASSERT_EQ(IterStatus(iter), "(invalid)");
902  iter->SeekToLast();
903  ASSERT_EQ(IterStatus(iter), "c->vc");
904  iter->Prev();
905  ASSERT_EQ(IterStatus(iter), "b->vb");
906  iter->Prev();
907  ASSERT_EQ(IterStatus(iter), "a->va");
908  iter->Prev();
909  ASSERT_EQ(IterStatus(iter), "(invalid)");
910
911  delete iter;
912}
913
914TEST(DBTest, IterSmallAndLargeMix) {
915  ASSERT_OK(Put("a", "va"));
916  ASSERT_OK(Put("b", std::string(100000, 'b')));
917  ASSERT_OK(Put("c", "vc"));
918  ASSERT_OK(Put("d", std::string(100000, 'd')));
919  ASSERT_OK(Put("e", std::string(100000, 'e')));
920
921  Iterator* iter = db_->NewIterator(ReadOptions());
922
923  iter->SeekToFirst();
924  ASSERT_EQ(IterStatus(iter), "a->va");
925  iter->Next();
926  ASSERT_EQ(IterStatus(iter), "b->" + std::string(100000, 'b'));
927  iter->Next();
928  ASSERT_EQ(IterStatus(iter), "c->vc");
929  iter->Next();
930  ASSERT_EQ(IterStatus(iter), "d->" + std::string(100000, 'd'));
931  iter->Next();
932  ASSERT_EQ(IterStatus(iter), "e->" + std::string(100000, 'e'));
933  iter->Next();
934  ASSERT_EQ(IterStatus(iter), "(invalid)");
935
936  iter->SeekToLast();
937  ASSERT_EQ(IterStatus(iter), "e->" + std::string(100000, 'e'));
938  iter->Prev();
939  ASSERT_EQ(IterStatus(iter), "d->" + std::string(100000, 'd'));
940  iter->Prev();
941  ASSERT_EQ(IterStatus(iter), "c->vc");
942  iter->Prev();
943  ASSERT_EQ(IterStatus(iter), "b->" + std::string(100000, 'b'));
944  iter->Prev();
945  ASSERT_EQ(IterStatus(iter), "a->va");
946  iter->Prev();
947  ASSERT_EQ(IterStatus(iter), "(invalid)");
948
949  delete iter;
950}
951
952TEST(DBTest, IterMultiWithDelete) {
953  do {
954    ASSERT_OK(Put("a", "va"));
955    ASSERT_OK(Put("b", "vb"));
956    ASSERT_OK(Put("c", "vc"));
957    ASSERT_OK(Delete("b"));
958    ASSERT_EQ("NOT_FOUND", Get("b"));
959
960    Iterator* iter = db_->NewIterator(ReadOptions());
961    iter->Seek("c");
962    ASSERT_EQ(IterStatus(iter), "c->vc");
963    iter->Prev();
964    ASSERT_EQ(IterStatus(iter), "a->va");
965    delete iter;
966  } while (ChangeOptions());
967}
968
969TEST(DBTest, Recover) {
970  do {
971    ASSERT_OK(Put("foo", "v1"));
972    ASSERT_OK(Put("baz", "v5"));
973
974    Reopen();
975    ASSERT_EQ("v1", Get("foo"));
976
977    ASSERT_EQ("v1", Get("foo"));
978    ASSERT_EQ("v5", Get("baz"));
979    ASSERT_OK(Put("bar", "v2"));
980    ASSERT_OK(Put("foo", "v3"));
981
982    Reopen();
983    ASSERT_EQ("v3", Get("foo"));
984    ASSERT_OK(Put("foo", "v4"));
985    ASSERT_EQ("v4", Get("foo"));
986    ASSERT_EQ("v2", Get("bar"));
987    ASSERT_EQ("v5", Get("baz"));
988  } while (ChangeOptions());
989}
990
991TEST(DBTest, RecoveryWithEmptyLog) {
992  do {
993    ASSERT_OK(Put("foo", "v1"));
994    ASSERT_OK(Put("foo", "v2"));
995    Reopen();
996    Reopen();
997    ASSERT_OK(Put("foo", "v3"));
998    Reopen();
999    ASSERT_EQ("v3", Get("foo"));
1000  } while (ChangeOptions());