Test call filtering for collator proxy type

Author: rockbmb

packages/kusama/src/__snapshots__/assetHubKusama.proxy.e2e.test.ts.snap

@@ -848,6 +848,25 @@ exports[`Kusama AssetHub Proxy > proxy call filtering test for CancelProxy > eve
 ]
 `;
 
+exports[`Kusama AssetHub Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`Kusama AssetHub Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/kusama/src/__snapshots__/coretimeKusama.proxy.e2e.test.ts.snap

@@ -562,6 +562,25 @@ exports[`Kusama Coretime Proxy > proxy call filtering test for CancelProxy > eve
 ]
 `;
 
+exports[`Kusama Coretime Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`Kusama Coretime Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/kusama/src/__snapshots__/peopleKusama.proxy.e2e.test.ts.snap

@@ -514,6 +514,25 @@ exports[`People Kusama Proxy > proxy call filtering test for CancelProxy > event
 ]
 `;
 
+exports[`People Kusama Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`People Kusama Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/polkadot/src/__snapshots__/assetHubPolkadot.proxy.e2e.test.ts.snap

@@ -848,6 +848,25 @@ exports[`Polkadot AssetHub Proxy > proxy call filtering test for CancelProxy > e
 ]
 `;
 
+exports[`Polkadot AssetHub Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`Polkadot AssetHub Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/polkadot/src/__snapshots__/collectivesPolkadot.proxy.e2e.test.ts.snap

@@ -562,6 +562,25 @@ exports[`Polkadot Collectives Proxy > proxy call filtering test for CancelProxy
 ]
 `;
 
+exports[`Polkadot Collectives Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`Polkadot Collectives Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/polkadot/src/__snapshots__/coretimePolkadot.proxy.e2e.test.ts.snap

@@ -554,6 +554,25 @@ exports[`Polkadot Coretime Proxy > proxy call filtering test for CancelProxy > e
 ]
 `;
 
+exports[`Polkadot Coretime Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`Polkadot Coretime Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/polkadot/src/__snapshots__/peoplePolkadot.proxy.e2e.test.ts.snap

@@ -514,6 +514,25 @@ exports[`People Polkadot Proxy > proxy call filtering test for CancelProxy > eve
 ]
 `;
 
+exports[`People Polkadot Proxy > proxy call filtering test for Collator > events for proxy action: proxy type Collator, pallet collatorSelection, call registerAsCandidate 1`] = `
+[
+  {
+    "data": {
+      "result": {
+        "Err": {
+          "Module": {
+            "error": "0x08000000",
+            "index": 21,
+          },
+        },
+      },
+    },
+    "method": "ProxyExecuted",
+    "section": "proxy",
+  },
+]
+`;
+
 exports[`People Polkadot Proxy > proxy call filtering test for NonTransfer > events for proxy action: proxy type NonTransfer, pallet system, call remark 1`] = `
 [
   {

packages/shared/src/proxy.ts

@@ -71,6 +71,7 @@ interface ProxyActionBuilder {
   buildBalancesAction(): ProxyAction[]
   buildBountyAction(): ProxyAction[]
   buildCancelProxyAction(): ProxyAction[]
+  buildCollatorSelectionAction(): ProxyAction[]
   buildCrowdloanAction(): ProxyAction[]
   buildFastUnstakeAction(): ProxyAction[]
   buildGovernanceAction(): ProxyAction[]
@@ -189,6 +190,19 @@ class ProxyActionBuilderImpl<
     return cancelProxyCalls
   }
 
+  buildCollatorSelectionAction(): ProxyAction[] {
+    const collatorSelectionCalls: ProxyAction[] = []
+    if (this.client.api.tx.collatorSelection) {
+      collatorSelectionCalls.push({
+        pallet: 'collatorSelection',
+        extrinsic: 'registerAsCandidate',
+        call: this.client.api.tx.collatorSelection.registerAsCandidate(),
+      })
+    }
+
+    return collatorSelectionCalls
+  }
+
   buildCrowdloanAction(): ProxyAction[] {
     const crowdloanCalls: ProxyAction[] = []
     if (this.client.api.tx.crowdloan) {
@@ -521,6 +535,7 @@ function buildProxyAction<
       ...proxyActionBuilder.buildUniquesOwnerAction(),
       ...proxyActionBuilder.buildUtilityAction(),
     ])
+    .with('Collator', () => [...proxyActionBuilder.buildCollatorSelectionAction()])
 
     .otherwise(() => [])
 
@@ -645,6 +660,7 @@ async function proxyCallFilteringTestRunner<
     'Assets',
     'AssetOwner',
     'AssetManager',
+    'Collator',
   ]
 
   for (const [proxyType, proxyTypeIx] of Object.entries(proxyTypes)) {