feat: add test for contract address as module_id

lidofinance · Sep 18, 2023 · c4aaba6 · c4aaba6
1 parent 990e637
commit c4aaba6
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 0 deletions.
diff --git a/src/http/sr-modules-operators-keys/sr-modules-operators-keys.e2e-spec.ts b/src/http/sr-modules-operators-keys/sr-modules-operators-keys.e2e-spec.ts
@@ -123,6 +123,12 @@ describe('SRModulesOperatorsKeysController (e2e)', () => {
         // Get all keys and operators without filters
         const resp = await request(app.getHttpServer()).get(`/v1/modules/${dvtModule.id}/operators/keys`);
 
+        const respByContractAddress = await request(app.getHttpServer()).get(
+          `/v1/modules/${dvtModule.stakingModuleAddress}/operators/keys`,
+        );
+
+        expect(resp.body).toEqual(respByContractAddress.body);
+
         expect(resp.status).toEqual(200);
         expect(resp.body.data.operators).toEqual(expect.arrayContaining([operatorOneDvt, operatorTwoDvt]));
         expect(resp.body.data.keys).toEqual(expect.arrayContaining(dvtModuleKeys));

diff --git a/src/http/sr-modules-operators/sr-modules-operators.e2e-spec.ts b/src/http/sr-modules-operators/sr-modules-operators.e2e-spec.ts
@@ -203,6 +203,12 @@ describe('SRModuleOperatorsController (e2e)', () => {
         // Get all operators without filters
         const resp = await request(app.getHttpServer()).get(`/v1/modules/${dvtModule.id}/operators`);
 
+        const respByContractAddress = await request(app.getHttpServer()).get(
+          `/v1/modules/${dvtModule.stakingModuleAddress}/operators`,
+        );
+
+        expect(resp.body).toEqual(respByContractAddress.body);
+
         expect(resp.status).toEqual(200);
         expect(resp.body.data.operators).toBeDefined();
         expect(resp.body.data.operators).toEqual(expect.arrayContaining([operatorOneDvt, operatorTwoDvt]));

diff --git a/src/http/sr-modules/sr-modules.e2e-spec.ts b/src/http/sr-modules/sr-modules.e2e-spec.ts
@@ -169,6 +169,17 @@ describe('SRModulesController (e2e)', () => {
         });
       });
 
+      it('should return module by contract address', async () => {
+        const resp = await request(app.getHttpServer()).get(`/v1/modules/${dvtModule.stakingModuleAddress}`);
+        expect(resp.status).toEqual(200);
+        expect(resp.body.data).toEqual(dvtModuleResp);
+        expect(resp.body.elBlockSnapshot).toEqual({
+          blockNumber: elMeta.number,
+          blockHash: elMeta.hash,
+          timestamp: elMeta.timestamp,
+        });
+      });
+
       it("should return 404 if module doesn't exist", async () => {
         const resp = await request(app.getHttpServer()).get(`/v1/modules/77`);
         expect(resp.status).toEqual(404);
@@ -189,6 +200,7 @@ describe('SRModulesController (e2e)', () => {
         });
       });
     });
+
     describe('too early response case', () => {
       beforeEach(async () => {
         await cleanDB();