bghgary 5 سال پیش
والد
کامیت
054ef35133
1فایلهای تغییر یافته به همراه4 افزوده شده و 2 حذف شده
  1. 4 2
      tests/unit/babylon/src/Loading/babylon.sceneLoader.tests.ts

+ 4 - 2
tests/unit/babylon/src/Loading/babylon.sceneLoader.tests.ts

@@ -414,14 +414,14 @@ describe('Babylon Scene Loader', function() {
 
 
             const setRequestHeaderCalls = new Array<string>();
             const setRequestHeaderCalls = new Array<string>();
             const origSetRequestHeader = BABYLON.WebRequest.prototype.setRequestHeader;
             const origSetRequestHeader = BABYLON.WebRequest.prototype.setRequestHeader;
-            sinon.stub(BABYLON.WebRequest.prototype, "setRequestHeader").callsFake(function(...args) {
+            const setRequestHeaderStub = sinon.stub(BABYLON.WebRequest.prototype, "setRequestHeader").callsFake(function(...args) {
                 setRequestHeaderCalls.push(args.join(": "));
                 setRequestHeaderCalls.push(args.join(": "));
                 origSetRequestHeader.apply(this, args);
                 origSetRequestHeader.apply(this, args);
             });
             });
 
 
             // Simulate default CORS policy on some web servers that reject getResponseHeader calls with `Content-Range`.
             // Simulate default CORS policy on some web servers that reject getResponseHeader calls with `Content-Range`.
             const origGetResponseHeader = BABYLON.WebRequest.prototype.getResponseHeader;
             const origGetResponseHeader = BABYLON.WebRequest.prototype.getResponseHeader;
-            sinon.stub(BABYLON.WebRequest.prototype, "getResponseHeader").callsFake(function(...args) {
+            const getResponseHeaderStub = sinon.stub(BABYLON.WebRequest.prototype, "getResponseHeader").callsFake(function(...args) {
                 return (args[0] === "Content-Range") ? null : origGetResponseHeader.apply(this, args);
                 return (args[0] === "Content-Range") ? null : origGetResponseHeader.apply(this, args);
             });
             });
 
 
@@ -432,6 +432,8 @@ describe('Babylon Scene Loader', function() {
 
 
             promises.push(BABYLON.SceneLoader.AppendAsync("/Playground/scenes/", "LevelOfDetail.glb", scene).then(() => {
             promises.push(BABYLON.SceneLoader.AppendAsync("/Playground/scenes/", "LevelOfDetail.glb", scene).then(() => {
                 expect(setRequestHeaderCalls, "setRequestHeaderCalls").to.have.ordered.members(expectedSetRequestHeaderCalls);
                 expect(setRequestHeaderCalls, "setRequestHeaderCalls").to.have.ordered.members(expectedSetRequestHeaderCalls);
+                setRequestHeaderStub.restore();
+                getResponseHeaderStub.restore();
             }));
             }));
 
 
             return Promise.all(promises);
             return Promise.all(promises);