/** * Copyright (C) 2018-present MongoDB, Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the Server Side Public License, version 1, * as published by MongoDB, Inc. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * Server Side Public License for more details. * * You should have received a copy of the Server Side Public License * along with this program. If not, see * . * * As a special exception, the copyright holders give permission to link the * code of portions of this program with the OpenSSL library under certain * conditions as described in each individual source file and distribute * linked combinations including the program with the OpenSSL library. You * must comply with the Server Side Public License in all respects for * all of the code used other than as permitted herein. If you modify file(s) * with this exception, you may extend this exception to your version of the * file(s), but you are not obligated to do so. If you do not wish to do so, * delete this exception statement from your version. If you delete this * exception statement from all source files in the program, then also delete * it in the license file. */ #include "mongo/base/data_view.h" #include #include "mongo/base/data_type_endian.h" #include "mongo/platform/endian.h" #include "mongo/unittest/unittest.h" namespace mongo { TEST(DataView, ConstDataView) { char buf[sizeof(uint32_t) * 3]; uint32_t native = 1234; uint32_t le = endian::nativeToLittle(native); uint32_t be = endian::nativeToBig(native); std::memcpy(buf, &native, sizeof(uint32_t)); std::memcpy(buf + sizeof(uint32_t), &le, sizeof(uint32_t)); std::memcpy(buf + sizeof(uint32_t) * 2, &be, sizeof(uint32_t)); ConstDataView cdv(buf); ASSERT_EQUALS(buf, cdv.view()); ASSERT_EQUALS(buf + 5, cdv.view(5)); ASSERT_EQUALS(native, cdv.read()); ASSERT_EQUALS(native, cdv.read>(sizeof(uint32_t))); ASSERT_EQUALS(native, cdv.read>(sizeof(uint32_t) * 2)); } TEST(DataView, DataView) { char buf[sizeof(uint32_t) * 3]; uint32_t native = 1234; DataView dv(buf); dv.write(native); dv.write(LittleEndian(native), sizeof(uint32_t)); dv.write(BigEndian(native), sizeof(uint32_t) * 2); ASSERT_EQUALS(buf, dv.view()); ASSERT_EQUALS(buf + 5, dv.view(5)); ASSERT_EQUALS(native, dv.read()); ASSERT_EQUALS(native, dv.read>(sizeof(uint32_t))); ASSERT_EQUALS(native, dv.read>(sizeof(uint32_t) * 2)); } } // namespace mongo