DxePrmModuleDiscoveryLibUnitTest.c

Go to the documentation of this file.

 
#include <stdio.h>
#include <string.h>
#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
 
#include <Library/BaseLib.h>
#include <Library/BaseMemoryLib.h>
#include <Library/DebugLib.h>
#include <Library/MemoryAllocationLib.h>
#include <Library/PrmModuleDiscoveryLib.h>
#include <Library/UefiBootServicesTableLib.h>
#include <Library/UnitTestLib.h>
 
#include "../PrmModuleDiscovery.h"
 
#define UNIT_TEST_NAME     "PRM Module Discovery Library Unit Test"
#define UNIT_TEST_VERSION  "0.1"
 
 
 
UNIT_TEST_STATUS
EFIAPI
PrmModuleImageContextListEntryShouldDeallocate (
  IN UNIT_TEST_CONTEXT  Context
  )
{
  PRM_MODULE_IMAGE_CONTEXT_LIST_ENTRY  *ListEntry;
 
  ListEntry = CreateNewPrmModuleImageContextListEntry ();
 
  UT_ASSERT_NOT_NULL (ListEntry);
  if (ListEntry != NULL) {
    FreePool (ListEntry);
  }
 
  return UNIT_TEST_PASSED;
}
 
UNIT_TEST_STATUS
EFIAPI
PrmModuleImageContextListEntrySignatureShouldBeValid (
  IN UNIT_TEST_CONTEXT  Context
  )
{
  PRM_MODULE_IMAGE_CONTEXT_LIST_ENTRY  *ListEntry;
 
  ListEntry = CreateNewPrmModuleImageContextListEntry ();
 
  UT_ASSERT_TRUE (ListEntry->Signature == PRM_MODULE_IMAGE_CONTEXT_LIST_ENTRY_SIGNATURE);
 
  if (ListEntry != NULL) {
    FreePool (ListEntry);
  }
 
  return UNIT_TEST_PASSED;
}
 
UNIT_TEST_STATUS
EFIAPI
PrmModuleImageContextListEntryImageContextShouldBeZeroed (
  IN UNIT_TEST_CONTEXT  Context
  )
{
  PRM_MODULE_IMAGE_CONTEXT_LIST_ENTRY  *ListEntry;
  PRM_MODULE_IMAGE_CONTEXT             ImageContext;
 
  ListEntry = CreateNewPrmModuleImageContextListEntry ();
 
  ZeroMem (&ImageContext, sizeof (ImageContext));
  UT_ASSERT_MEM_EQUAL (&ListEntry->Context, &ImageContext, sizeof (ImageContext));
 
  if (ListEntry != NULL) {
    FreePool (ListEntry);
  }
 
  return UNIT_TEST_PASSED;
}
 
 
int
main (
  )
{
  EFI_STATUS                  Status;
  UNIT_TEST_FRAMEWORK_HANDLE  Framework;
  UNIT_TEST_SUITE_HANDLE      CreateNewPrmModuleImageContextListEntryTests;
 
  Framework = NULL;
 
  DEBUG ((DEBUG_INFO, "%a v%a\n", UNIT_TEST_NAME, UNIT_TEST_VERSION));
 
  //
  // Start setting up the test framework for running the tests.
  //
  Status = InitUnitTestFramework (&Framework, UNIT_TEST_NAME, gEfiCallerBaseName, UNIT_TEST_VERSION);
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_ERROR, "Failed in InitUnitTestFramework. Status = %r\n", Status));
    goto EXIT;
  }
 
  Status =  CreateUnitTestSuite (
              &CreateNewPrmModuleImageContextListEntryTests,
              Framework,
              "Create New PRM Module Image Context List Entry Tests",
              "PrmModuleDiscoveryLib.CreateNewPrmModuleImageContextListEntry",
              NULL,
              NULL
              );
  if (EFI_ERROR (Status)) {
    DEBUG ((DEBUG_ERROR, "Failed in CreateUnitTestSuite for PrmModuleDiscoveryLib.CreateNewPrmModuleImageContextListEntry\n"));
    Status = EFI_OUT_OF_RESOURCES;
    goto EXIT;
  }
 
  AddTestCase (
    CreateNewPrmModuleImageContextListEntryTests,
    "",
    "PrmModuleDiscoveryLib.CreateNewPrmModuleImageContextListEntry.ListEntryShouldDeallocate",
    PrmModuleImageContextListEntryShouldDeallocate,
    NULL,
    NULL,
    NULL
    );
 
  AddTestCase (
    CreateNewPrmModuleImageContextListEntryTests,
    "",
    "PrmModuleDiscoveryLib.CreateNewPrmModuleImageContextListEntry.ListEntrySignatureShouldBeValid",
    PrmModuleImageContextListEntrySignatureShouldBeValid,
    NULL,
    NULL,
    NULL
    );
 
  AddTestCase (
    CreateNewPrmModuleImageContextListEntryTests,
    "",
    "PrmModuleDiscoveryLib.CreateNewPrmModuleImageContextListEntry.ListEntryImageContextShouldBeZeroed",
    PrmModuleImageContextListEntryImageContextShouldBeZeroed,
    NULL,
    NULL,
    NULL
    );
 
  //
  // Execute the tests.
  //
  Status = RunAllTestSuites (Framework);
 
EXIT:
  if (Framework) {
    FreeUnitTestFramework (Framework);
  }
 
  return Status;
}