Restores a stub, which breaks other test files

Signed-off-by: Konstantina Blazhukova <konstantina.blajukova@gmail.com>
hiero-ledger · Jan 24, 2024 · d8d64b4 · d8d64b4
1 parent abe8a97
commit d8d64b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/test/unit/controller/StateController.spec.ts b/test/unit/controller/StateController.spec.ts
@@ -90,13 +90,14 @@ describe('StateController', () => {
   describe('should handle other events correctly', () => {
     let stateController: StateController;
     let getStartConfigurationStub: sinon.SinonStub;
+    let getStub1;
 
     before(() => {
       const loggerServiceStub = sinon.createStubInstance(LoggerService);
       const cliServiceStub = sinon.createStubInstance(CLIService);
 
       // Stub the first call to ServiceLocator.Current.get
-      const getStub1 = sinon.stub(ServiceLocator.Current, 'get');
+      getStub1 = sinon.stub(ServiceLocator.Current, 'get');
       getStub1.onFirstCall().returns(loggerServiceStub);
       getStub1.onSecondCall().returns(cliServiceStub);
       getStub1.onThirdCall().returns(loggerServiceStub);
@@ -111,6 +112,7 @@ describe('StateController', () => {
     after(() => {
       // Restore the original ServiceLocator after each test
       getStartConfigurationStub.restore();
+      getStub1.restore();
     });
 
     it('should handle other events correctly', async () => {

diff --git a/test/unit/states/AccountCreationState.spec.ts b/test/unit/states/AccountCreationState.spec.ts
@@ -25,7 +25,6 @@ import { LoggerService } from '../../../src/services/LoggerService';
 import { CLIService } from '../../../src/services/CLIService'
 import { ClientService } from '../../../src/services/ClientService';
 import { ServiceLocator } from '../../../src/services/ServiceLocator';
-import { StateController } from '../../../src/controller/StateController';
 import { PrivateKey } from '@hashgraph/sdk';
 
 describe('AccountCreationState', () => {