libs/jpegrecoverymap/tests/jpegdecoder_test.cpp

/*
 * Copyright 2022 The Android Open Source Project
 *
 * 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.
 */

#include <jpegrecoverymap/jpegdecoder.h>
#include <gtest/gtest.h>
#include <utils/Log.h>

#include <fcntl.h>

namespace android::recoverymap {

#define YUV_IMAGE "/sdcard/Documents/minnie-320x240-yuv.jpg"
#define YUV_IMAGE_SIZE 20193
#define GREY_IMAGE "/sdcard/Documents/minnie-320x240-y.jpg"
#define GREY_IMAGE_SIZE 20193

class JpegDecoderTest : public testing::Test {
public:
    struct Image {
        std::unique_ptr<uint8_t[]> buffer;
        size_t size;
    };
    JpegDecoderTest();
    ~JpegDecoderTest();
protected:
    virtual void SetUp();
    virtual void TearDown();

    Image mYuvImage, mGreyImage;
};

JpegDecoderTest::JpegDecoderTest() {}

JpegDecoderTest::~JpegDecoderTest() {}

static size_t getFileSize(int fd) {
    struct stat st;
    if (fstat(fd, &st) < 0) {
        ALOGW("%s : fstat failed", __func__);
        return 0;
    }
    return st.st_size; // bytes
}

static bool loadFile(const char filename[], JpegDecoderTest::Image* result) {
    int fd = open(filename, O_CLOEXEC);
    if (fd < 0) {
        return false;
    }
    int length = getFileSize(fd);
    if (length == 0) {
        close(fd);
        return false;
    }
    result->buffer.reset(new uint8_t[length]);
    if (read(fd, result->buffer.get(), length) != static_cast<ssize_t>(length)) {
        close(fd);
        return false;
    }
    close(fd);
    return true;
}

void JpegDecoderTest::SetUp() {
    if (!loadFile(YUV_IMAGE, &mYuvImage)) {
        FAIL() << "Load file " << YUV_IMAGE << " failed";
    }
    mYuvImage.size = YUV_IMAGE_SIZE;
    if (!loadFile(GREY_IMAGE, &mGreyImage)) {
        FAIL() << "Load file " << GREY_IMAGE << " failed";
    }
    mGreyImage.size = GREY_IMAGE_SIZE;
}

void JpegDecoderTest::TearDown() {}

TEST_F(JpegDecoderTest, decodeYuvImage) {
    JpegDecoder decoder;
    EXPECT_TRUE(decoder.decompressImage(mYuvImage.buffer.get(), mYuvImage.size));
    ASSERT_GT(decoder.getDecompressedImageSize(), static_cast<uint32_t>(0));
}

TEST_F(JpegDecoderTest, decodeGreyImage) {
    JpegDecoder decoder;
    EXPECT_TRUE(decoder.decompressImage(mGreyImage.buffer.get(), mGreyImage.size));
    ASSERT_GT(decoder.getDecompressedImageSize(), static_cast<uint32_t>(0));
}

}