File tree Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Expand file tree Collapse file tree 2 files changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -74,12 +74,16 @@ export class SeleniumWebDriverHarnessEnvironment extends HarnessEnvironment<
74
74
/** The options for this environment. */
75
75
private _options : WebDriverHarnessEnvironmentOptions ;
76
76
77
+ /** Environment stabilization callback passed to the created test elements. */
78
+ private _stabilizeCallback : ( ) => Promise < void > ;
79
+
77
80
protected constructor (
78
81
rawRootElement : ( ) => webdriver . WebElement ,
79
82
options ?: WebDriverHarnessEnvironmentOptions ,
80
83
) {
81
84
super ( rawRootElement ) ;
82
85
this . _options = { ...defaultEnvironmentOptions , ...options } ;
86
+ this . _stabilizeCallback = ( ) => this . forceStabilize ( ) ;
83
87
}
84
88
85
89
/** Gets the ElementFinder corresponding to the given TestElement. */
@@ -123,7 +127,7 @@ export class SeleniumWebDriverHarnessEnvironment extends HarnessEnvironment<
123
127
124
128
/** Creates a `TestElement` from a raw element. */
125
129
protected createTestElement ( element : ( ) => webdriver . WebElement ) : TestElement {
126
- return new SeleniumWebDriverElement ( element , ( ) => this . forceStabilize ( ) ) ;
130
+ return new SeleniumWebDriverElement ( element , this . _stabilizeCallback ) ;
127
131
}
128
132
129
133
/** Creates a `HarnessLoader` rooted at the given raw element. */
Original file line number Diff line number Diff line change @@ -96,6 +96,9 @@ export class TestbedHarnessEnvironment extends HarnessEnvironment<Element> {
96
96
/** The options for this environment. */
97
97
private _options : TestbedHarnessEnvironmentOptions ;
98
98
99
+ /** Environment stabilization callback passed to the created test elements. */
100
+ private _stabilizeCallback : ( ) => Promise < void > ;
101
+
99
102
protected constructor (
100
103
rawRootElement : Element ,
101
104
private _fixture : ComponentFixture < unknown > ,
@@ -104,6 +107,7 @@ export class TestbedHarnessEnvironment extends HarnessEnvironment<Element> {
104
107
super ( rawRootElement ) ;
105
108
this . _options = { ...defaultEnvironmentOptions , ...options } ;
106
109
this . _taskState = TaskStateZoneInterceptor . setup ( ) ;
110
+ this . _stabilizeCallback = ( ) => this . forceStabilize ( ) ;
107
111
installAutoChangeDetectionStatusHandler ( _fixture ) ;
108
112
_fixture . componentRef . onDestroy ( ( ) => {
109
113
uninstallAutoChangeDetectionStatusHandler ( _fixture ) ;
@@ -198,7 +202,7 @@ export class TestbedHarnessEnvironment extends HarnessEnvironment<Element> {
198
202
199
203
/** Creates a `TestElement` from a raw element. */
200
204
protected createTestElement ( element : Element ) : TestElement {
201
- return new UnitTestElement ( element , ( ) => this . forceStabilize ( ) ) ;
205
+ return new UnitTestElement ( element , this . _stabilizeCallback ) ;
202
206
}
203
207
204
208
/** Creates a `HarnessLoader` rooted at the given raw element. */
You can’t perform that action at this time.
0 commit comments