haiku.git

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


#include "JwtTokenHelperTest.h"

#include <String.h>

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

#include <string.h>

#include "JwtTokenHelper.h"


JwtTokenHelperTest::JwtTokenHelperTest()
{
}


JwtTokenHelperTest::~JwtTokenHelperTest()
{
}


void
JwtTokenHelperTest::TestParseTokenClaims()
{
	const char* jwtToken = "eyJhbGciOiJIUzI1NiJ9.eyJpc3MiOiJkZXYuaGRzIiwic3ViIj"
		"oiZXJpazY0QGhkcyIsImV4cCI6MTY5MzkwNzM1NywiaWF0IjoxNjkzOTA3MDU3fQ.DJOz0"
		"TmgN0Ya8De-oV0mBwWb-8FYavLbaFUFhCLqr-s";
	BMessage actualMessage;

// ----------------------
	status_t result = JwtTokenHelper::ParseClaims(BString(jwtToken), actualMessage);
// ----------------------

	CPPUNIT_ASSERT_EQUAL(B_OK, result);

	_AssertStringValue(actualMessage, "iss", "dev.hds");
	_AssertStringValue(actualMessage, "sub", "erik64@hds");
	_AssertDoubleValue(actualMessage, "exp", 1693907357);
	_AssertDoubleValue(actualMessage, "iat", 1693907057);
}


void
JwtTokenHelperTest::TestCorrupt()
{
	const char* jwtToken = "application/json";
	BMessage actualMessage;

// ----------------------
	status_t result = JwtTokenHelper::ParseClaims(BString(jwtToken), actualMessage);
// ----------------------

	CPPUNIT_ASSERT(B_OK != result);
}


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

	suite.addTest(
		new CppUnit::TestCaller<JwtTokenHelperTest>(
			"JwtTokenHelperTest::TestParseTokenClaims",
			&JwtTokenHelperTest::TestParseTokenClaims));

	suite.addTest(
		new CppUnit::TestCaller<JwtTokenHelperTest>(
			"JwtTokenHelperTest::TestCorrupt",
			&JwtTokenHelperTest::TestCorrupt));

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


void
JwtTokenHelperTest::_AssertStringValue(const BMessage& message, const char* key,
	const char* expectedValue) const
{
	BString value;
	status_t result = message.FindString(key, &value);
	CPPUNIT_ASSERT_EQUAL(B_OK, result);
	CPPUNIT_ASSERT_EQUAL(BString(expectedValue), value);
}


void
JwtTokenHelperTest::_AssertDoubleValue(const BMessage& message, const char* key,
	int64 expectedValue) const
{
	double value;
	status_t result = message.FindDouble(key, &value);
	CPPUNIT_ASSERT_EQUAL(B_OK, result);
	CPPUNIT_ASSERT_EQUAL((double) expectedValue, value);
}