/*

 * Copyright 2016 Intel Corporation

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

// primary header

#include "bitWriter.h"

// library headers

#include "common/unittest.h"

namespace YamiParser {

static const uint16_t writeArrary[16][2] = {

    { 1, 1 }, { 2, 2 }, { 3, 2 }, { 4, 3 },

    { 5, 3 }, { 6, 3 }, { 7, 3 }, { 8, 4 },

    { 9, 4 }, { 10, 4 }, { 11, 4 }, { 12, 4 },

    { 13, 4 }, { 14, 4 }, { 15, 4 }, { 16, 5 }

};

class BitWriterTest

    : public ::testing::Test {

};

#define BITWriter_TEST(name) \

    TEST_F(BitWriterTest, name)

BITWriter_TEST(Writer_Simple)

{

    BitWriter Writer;

    EXPECT_EQ(0u, Writer.getCodedBitsCount());

    EXPECT_EQ(NULL, Writer.getBitWriterData());

    uint32_t i, bitsSum;

    for (i = 0, bitsSum = 0; i < 16; i++) {

        EXPECT_TRUE(Writer.writeBits(writeArrary[i][0], writeArrary[i][1]));

        bitsSum += writeArrary[i][1];

    }

    EXPECT_EQ(bitsSum, Writer.getCodedBitsCount());

}

} // namespace YamiParser