fix: partitionsOf needs to return an array of Partitions

Author: Victor Wiebe

src/contract_wrappers/tokens/__tests__/security_token_wrapper.test.ts

@@ -148,7 +148,7 @@ describe('SecurityTokenWrapper', () => {
 
   describe('partitionsOf', () => {
     test('should call to partitionsOf', async () => {
-      const expectedResult = stringArrayToBytes32Array(['LOCKED', 'UNLOCKED']);
+      const expectedResult = stringArrayToBytes32Array(['UNDEFINED', 'UNLOCKED']);
       const mockedParams = {
         partition: 'UNLOCKED',
         operator: '0x5555555555555555555555555555555555555555',

src/contract_wrappers/tokens/security_token_wrapper.ts

@@ -418,11 +418,9 @@ interface GetSecurityTokenLogsAsyncParams extends GetLogs {
   (params: GetModuleAddedLogsAsyncParams): Promise<LogWithDecodedArgs<SecurityTokenModuleAddedEventArgs>[]>;
   (params: GetModuleUpgradedLogsAsyncParams): Promise<LogWithDecodedArgs<SecurityTokenModuleUpgradedEventArgs>[]>;
   (params: GetUpdateTokenDetailsLogsAsyncParams): Promise<
-      LogWithDecodedArgs<SecurityTokenUpdateTokenDetailsEventArgs>[]
-      >;
-  (params: GetUpdateTokenNameLogsAsyncParams): Promise<
-      LogWithDecodedArgs<SecurityTokenUpdateTokenNameEventArgs>[]
-      >;
+    LogWithDecodedArgs<SecurityTokenUpdateTokenDetailsEventArgs>[]
+  >;
+  (params: GetUpdateTokenNameLogsAsyncParams): Promise<LogWithDecodedArgs<SecurityTokenUpdateTokenNameEventArgs>[]>;
   (params: GetGranularityChangedLogsAsyncParams): Promise<
     LogWithDecodedArgs<SecurityTokenGranularityChangedEventArgs>[]
   >;
@@ -912,9 +910,11 @@ export default class SecurityTokenWrapper extends ERC20TokenWrapper {
     );
   };
 
-  public partitionsOf = async (params: PartitionsOfParams): Promise<string[]> => {
+  public partitionsOf = async (params: PartitionsOfParams): Promise<Partition[]> => {
     const partitions = await (await this.contract).partitionsOf.callAsync(params.tokenHolder);
-    return bytes32ArrayToStringArray(partitions);
+    return bytes32ArrayToStringArray(partitions).map(element => {
+      return element === Partition.Unlocked ? Partition.Unlocked : Partition.Undefined;
+    });
   };
 
   /**