minor refactor

latticexyz · Feb 10, 2022 · b0a7659 · b0a7659
1 parent ea1fcb1
commit b0a7659
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/src/Persona.sol b/src/Persona.sol
@@ -83,10 +83,6 @@ contract Persona is ERC721 {
         owner = newContractOwner;
     }
 
-    function getOwner(uint256 personaId) public view returns (address) {
-        return ownerOf[personaId];
-    }
-
     /*///////////////////////////////////////////////////////////////
                                 VIEW
     //////////////////////////////////////////////////////////////*/
@@ -126,7 +122,7 @@ contract Persona is ERC721 {
             return getAuthorization(personaId, user, consumer).isAuthorized;
         } else if (getPermission(personaId, user) == PersonaPermission.FUNCTION_SPECIFIC) {
             // TODO: Fix this hacky solution
-            if(fnSignature == bytes4(0)) {
+            if (fnSignature == bytes4(0)) {
                 return true;
             }
             bytes4[] memory fns = getAuthorization(personaId, user, consumer).authorizedFns;

diff --git a/src/test/Persona.t.sol b/src/test/Persona.t.sol
@@ -290,15 +290,15 @@ contract PersonaTest is BaseTest {
         vm.stopPrank();
     }
 
-    /// Transfer \\\ 
+    /// Transfer \\\
     function testTransferOwnership() public {
         uint256 id = _mintTo(personaOwner);
 
         vm.startPrank(personaOwner);
         persona.transferFrom(address(personaOwner), address(alice), id);
         vm.stopPrank();
 
-        assertTrue(persona.getOwner(id) == address(alice));
+        assertTrue(persona.ownerOf(id) == address(alice));
     }
 
     function testTransferRemovesAuthorization() public {