Add unit test for testPrimaryKeyExtract