diff options
| author | eshemchik <eshemchik@yandex-team.ru> | 2022-02-10 16:50:53 +0300 |
|---|---|---|
| committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:50:53 +0300 |
| commit | 9905811fdc643c722d6464b8cf6661f931510f2f (patch) | |
| tree | 5d5cb817648f650d76cf1076100726fd9b8448e8 /library/cpp/archive/directory_models_archive_reader_ut.cpp | |
| parent | face8a85ff3525043b92e692a9891cd1c19519c4 (diff) | |
| download | ydb-9905811fdc643c722d6464b8cf6661f931510f2f.tar.gz | |
Restoring authorship annotation for <eshemchik@yandex-team.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/archive/directory_models_archive_reader_ut.cpp')
| -rw-r--r-- | library/cpp/archive/directory_models_archive_reader_ut.cpp | 108 |
1 files changed, 54 insertions, 54 deletions
diff --git a/library/cpp/archive/directory_models_archive_reader_ut.cpp b/library/cpp/archive/directory_models_archive_reader_ut.cpp index 92b710a495..09994de9b4 100644 --- a/library/cpp/archive/directory_models_archive_reader_ut.cpp +++ b/library/cpp/archive/directory_models_archive_reader_ut.cpp @@ -1,55 +1,55 @@ -#include "directory_models_archive_reader.h" - +#include "directory_models_archive_reader.h" + #include <library/cpp/testing/unittest/registar.h> - -#include <util/folder/tempdir.h> -#include <util/string/cast.h> -#include <util/stream/file.h> -#include <util/system/tempfile.h> -#include <util/memory/blob.h> - -class TDirectoryModelsArchiveReaderTest: public TTestBase { - UNIT_TEST_SUITE(TDirectoryModelsArchiveReaderTest) - UNIT_TEST(TestRead); - UNIT_TEST_SUITE_END(); - -private: - void TestRead(); -}; - -UNIT_TEST_SUITE_REGISTRATION(TDirectoryModelsArchiveReaderTest); - -const TString MAIN_DIR = "./dir"; -const TString SUBDIR = "/subdir"; -const TString SAMPLE_FILE1 = "/sample1"; -const TString SAMPLE_FILE2 = "/sample2"; -const TString TEST_TEXT = "Test Text."; - -void TDirectoryModelsArchiveReaderTest::TestRead() { - TTempDir mainDir(MAIN_DIR); - TTempDir subDir(MAIN_DIR + SUBDIR); - TTempFileHandle file1(MAIN_DIR + SAMPLE_FILE1); - TTempFileHandle file2(MAIN_DIR + SUBDIR + SAMPLE_FILE2); - - file1.Write(TEST_TEXT.data(), TEST_TEXT.size()); - file1.FlushData(); - - TDirectoryModelsArchiveReader reader(MAIN_DIR, false); - - UNIT_ASSERT_EQUAL(reader.Count(), 2); - - UNIT_ASSERT(reader.Has(SAMPLE_FILE1)); - UNIT_ASSERT(reader.Has(SUBDIR + SAMPLE_FILE2)); - - UNIT_ASSERT_EQUAL(reader.KeyByIndex(0), SAMPLE_FILE1); - UNIT_ASSERT_EQUAL(reader.KeyByIndex(1), SUBDIR + SAMPLE_FILE2); - - TBlob blob = reader.BlobByKey(SAMPLE_FILE1); - Cout << "'" << TString(blob.AsCharPtr(), blob.Size()) << "' - '" << TEST_TEXT << "'" << Endl; - UNIT_ASSERT_VALUES_EQUAL(TString(blob.AsCharPtr(), blob.Size()), TString(TEST_TEXT)); - - TAutoPtr<IInputStream> is = reader.ObjectByKey(SAMPLE_FILE1); - const TString data = is->ReadAll(); - Cout << "'" << data << "' - '" << TEST_TEXT << "'" << Endl; - UNIT_ASSERT_VALUES_EQUAL(data, TEST_TEXT); -} + +#include <util/folder/tempdir.h> +#include <util/string/cast.h> +#include <util/stream/file.h> +#include <util/system/tempfile.h> +#include <util/memory/blob.h> + +class TDirectoryModelsArchiveReaderTest: public TTestBase { + UNIT_TEST_SUITE(TDirectoryModelsArchiveReaderTest) + UNIT_TEST(TestRead); + UNIT_TEST_SUITE_END(); + +private: + void TestRead(); +}; + +UNIT_TEST_SUITE_REGISTRATION(TDirectoryModelsArchiveReaderTest); + +const TString MAIN_DIR = "./dir"; +const TString SUBDIR = "/subdir"; +const TString SAMPLE_FILE1 = "/sample1"; +const TString SAMPLE_FILE2 = "/sample2"; +const TString TEST_TEXT = "Test Text."; + +void TDirectoryModelsArchiveReaderTest::TestRead() { + TTempDir mainDir(MAIN_DIR); + TTempDir subDir(MAIN_DIR + SUBDIR); + TTempFileHandle file1(MAIN_DIR + SAMPLE_FILE1); + TTempFileHandle file2(MAIN_DIR + SUBDIR + SAMPLE_FILE2); + + file1.Write(TEST_TEXT.data(), TEST_TEXT.size()); + file1.FlushData(); + + TDirectoryModelsArchiveReader reader(MAIN_DIR, false); + + UNIT_ASSERT_EQUAL(reader.Count(), 2); + + UNIT_ASSERT(reader.Has(SAMPLE_FILE1)); + UNIT_ASSERT(reader.Has(SUBDIR + SAMPLE_FILE2)); + + UNIT_ASSERT_EQUAL(reader.KeyByIndex(0), SAMPLE_FILE1); + UNIT_ASSERT_EQUAL(reader.KeyByIndex(1), SUBDIR + SAMPLE_FILE2); + + TBlob blob = reader.BlobByKey(SAMPLE_FILE1); + Cout << "'" << TString(blob.AsCharPtr(), blob.Size()) << "' - '" << TEST_TEXT << "'" << Endl; + UNIT_ASSERT_VALUES_EQUAL(TString(blob.AsCharPtr(), blob.Size()), TString(TEST_TEXT)); + + TAutoPtr<IInputStream> is = reader.ObjectByKey(SAMPLE_FILE1); + const TString data = is->ReadAll(); + Cout << "'" << data << "' - '" << TEST_TEXT << "'" << Endl; + UNIT_ASSERT_VALUES_EQUAL(data, TEST_TEXT); +} |