haiku.git

/*
 * Copyright 2023, Andrew Lindesay <apl@lindesay.co.nz>.
 * All rights reserved. Distributed under the terms of the MIT License.
 */


#include "DataIOUtilsTest.h"

#include <String.h>

#include <cppunit/TestCaller.h>
#include <cppunit/TestSuite.h>

#include <string.h>

#include "DataIOUtils.h"


DataIOUtilsTest::DataIOUtilsTest()
{
}


DataIOUtilsTest::~DataIOUtilsTest()
{
}


void
DataIOUtilsTest::TestReadBase64JwtClaims_1()
{
	const char* jwtToken = "eyJpc3MiOiJkZXYuaGRzIiwic3ViIjoiZXJpazY0QGhkcyIs"
		"ImV4cCI6MTY5MzE5MTMzMiwiaWF0IjoxNjkzMTkxMDMyfQ";
	BMemoryIO memoryIo(jwtToken, strlen(jwtToken));
	Base64DecodingDataIO base64DecodingIo(&memoryIo, '-', '_');
	char actualOutputBuffer[71];
	size_t actualReadBytes;

	bzero(actualOutputBuffer, 71);

// ----------------------
	status_t result = base64DecodingIo.ReadExactly(actualOutputBuffer, 70, &actualReadBytes);
// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, result);
	CPPUNIT_ASSERT_EQUAL(70, actualReadBytes);
	actualOutputBuffer[actualReadBytes] = 0;

	CPPUNIT_ASSERT_EQUAL(0x7b, (uint8) actualOutputBuffer[0]);

	CPPUNIT_ASSERT_EQUAL(
		BString("{\"iss\":\"dev.hds\",\"sub\":\"erik64@hds\",\"exp\":1693191332,\"iat\""
			":1693191032}"),
		BString(actualOutputBuffer));
}


void
DataIOUtilsTest::TestReadBase64JwtClaims_2()
{
	const char* jwtToken = "eyJpc3MiOiJkZXYuaGRzIiwic3ViIjoidG93ZWxkb3dudGVhQ"
		"GhkcyIsImV4cCI6MTY5MzczODgyNiwiaWF0IjoxNjkzNzM4NTI2fQ";
	BMemoryIO memoryIo(jwtToken, strlen(jwtToken));
	Base64DecodingDataIO base64DecodingIo(&memoryIo, '-', '_');
	char actualOutputBuffer[77];
	size_t actualReadBytes;

	bzero(actualOutputBuffer, 77);

// ----------------------
	status_t result = base64DecodingIo.ReadExactly(actualOutputBuffer, 76, &actualReadBytes);
// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, result);
	CPPUNIT_ASSERT_EQUAL(76, actualReadBytes);
	actualOutputBuffer[actualReadBytes] = 0;

	CPPUNIT_ASSERT_EQUAL(0x7b, (uint8) actualOutputBuffer[0]);

	CPPUNIT_ASSERT_EQUAL(
		BString("{\"iss\":\"dev.hds\",\"sub\":\"toweldowntea@hds\",\"exp\":1693738826,\"iat\""
			":1693738526}"),
		BString(actualOutputBuffer));
}


void
DataIOUtilsTest::TestReadBase64Corrupt()
{
	const char* jwtToken = "QW5k$mV3";
		// note that '$' is not a valid base64 character
	BMemoryIO memoryIo(jwtToken, strlen(jwtToken));
	Base64DecodingDataIO base64DecodingIo(&memoryIo, '-', '_');
	char actualOutputBuffer[7];
	size_t actualReadBytes;

	bzero(actualOutputBuffer, 7);

// ----------------------
	status_t result = base64DecodingIo.ReadExactly(actualOutputBuffer, 6, &actualReadBytes);
// ----------------------

	CPPUNIT_ASSERT(B_OK != result);
}


/*static*/ void
DataIOUtilsTest::AddTests(BTestSuite& parent)
{
	CppUnit::TestSuite& suite = *new CppUnit::TestSuite("DataIOUtilsTest");

	suite.addTest(
		new CppUnit::TestCaller<DataIOUtilsTest>(
			"DataIOUtilsTest::TestReadBase64JwtClaims_1",
			&DataIOUtilsTest::TestReadBase64JwtClaims_1));

	suite.addTest(
		new CppUnit::TestCaller<DataIOUtilsTest>(
			"DataIOUtilsTest::TestReadBase64JwtClaims_2",
			&DataIOUtilsTest::TestReadBase64JwtClaims_2));

	suite.addTest(
		new CppUnit::TestCaller<DataIOUtilsTest>(
			"DataIOUtilsTest::TestReadBase64Corrupt",
			&DataIOUtilsTest::TestReadBase64Corrupt));

	parent.addTest("DataIOUtilsTest", &suite);
}