11import * as assert from "assert" ;
2- import * as mockery from "mockery" ;
2+ // import * as mockery from "mockery";
33import { EndpointAuthorization } from "azure-pipelines-task-lib" ;
44import { ServiceConnectionAuthType , TokenServiceConnection , UsernamePasswordServiceConnection , ApiKeyServiceConnection , IAdditionalData } from "../serviceConnectionUtils" ;
5+ import * as mocker from "azure-pipelines-task-lib/lib-mocker" ;
56
67export function serviceConnectionUtilsTests ( ) {
78
89 const serviceConnectionsKey = "someProtocolServiceConnections" ;
910
1011 before ( ( ) => {
11- mockery . disable ( ) ; // needed to ensure that we can mock vsts-task-lib/task
12- mockery . enable ( {
12+ mocker . disable ( ) ; // needed to ensure that we can mock vsts-task-lib/task
13+ mocker . enable ( {
1314 useCleanCache : true ,
1415 warnOnUnregistered : true
15- } as mockery . MockeryEnableArgs ) ;
16+ } as mocker . MockOptions ) ;
1617 } ) ;
1718
1819 after ( ( ) => {
19- mockery . disable ( ) ;
20+ mocker . disable ( ) ;
2021 } ) ;
2122
2223 beforeEach ( ( ) => {
23- mockery . resetCache ( ) ;
24- mockery . registerAllowable ( "../serviceConnectionUtils" ) ;
24+ mocker . resetCache ( ) ;
25+ mocker . registerAllowable ( "../serviceConnectionUtils" ) ;
2526 } ) ;
2627
2728 afterEach ( ( ) => {
28- mockery . deregisterAll ( ) ;
29+ mocker . deregisterAll ( ) ;
2930 } ) ;
3031
3132 it ( "getPackagingServiceConnections null returns empty" , ( done : MochaDone ) => {
@@ -35,7 +36,7 @@ export function serviceConnectionUtilsTests() {
3536 } ,
3637 setResourcePath : ( path ) => { }
3738 } ;
38- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
39+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
3940
4041 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
4142 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) , [ ] ) ;
@@ -49,7 +50,7 @@ export function serviceConnectionUtilsTests() {
4950 } ,
5051 setResourcePath : ( path ) => { }
5152 } ;
52- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
53+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
5354
5455 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
5556 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) , [ ] ) ;
@@ -67,9 +68,10 @@ export function serviceConnectionUtilsTests() {
6768 scheme : "token"
6869 } ,
6970 setResourcePath : ( path ) => { } ,
70- getEndpointAuthorizationScheme : ( key , optional ) : string => "token"
71+ getEndpointAuthorizationScheme : ( key , optional ) : string => "token" ,
72+ setSecret : msg => null
7173 } ;
72- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
74+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
7375
7476 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
7577 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) , [ < TokenServiceConnection > {
@@ -96,7 +98,7 @@ export function serviceConnectionUtilsTests() {
9698 setResourcePath : ( path ) => { } ,
9799 getEndpointAuthorizationScheme : ( key , optional ) : string => "token"
98100 } ;
99- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
101+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
100102
101103 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
102104 assert . throws ( ( ) => serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) ) ;
@@ -114,9 +116,10 @@ export function serviceConnectionUtilsTests() {
114116 scheme : "usernamepassword"
115117 } ,
116118 setResourcePath : ( path ) => { } ,
117- getEndpointAuthorizationScheme : ( key , optional ) : string => "usernamepassword"
119+ getEndpointAuthorizationScheme : ( key , optional ) : string => "usernamepassword" ,
120+ setSecret : msg => null
118121 } ;
119- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
122+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
120123
121124 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
122125 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) , [ < UsernamePasswordServiceConnection > {
@@ -144,7 +147,7 @@ export function serviceConnectionUtilsTests() {
144147 setResourcePath : ( path ) => { } ,
145148 getEndpointAuthorizationScheme : ( key , optional ) : string => "usernamepassword"
146149 } ;
147- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
150+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
148151
149152 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
150153 assert . throws ( ( ) => serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) ) ;
@@ -163,7 +166,7 @@ export function serviceConnectionUtilsTests() {
163166 setResourcePath : ( path ) => { } ,
164167 getEndpointAuthorizationScheme : ( key , optional ) : string => "usernamepassword"
165168 } ;
166- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
169+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
167170
168171 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
169172 assert . throws ( ( ) => serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) ) ;
@@ -181,9 +184,10 @@ export function serviceConnectionUtilsTests() {
181184 scheme : "none"
182185 } ,
183186 setResourcePath : ( path ) => { } ,
184- getEndpointAuthorizationScheme : ( key , optional ) : string => "none"
187+ getEndpointAuthorizationScheme : ( key , optional ) : string => "none" ,
188+ setSecret : msg => null
185189 } ;
186- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
190+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
187191
188192 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
189193 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) , [ < ApiKeyServiceConnection > {
@@ -210,7 +214,7 @@ export function serviceConnectionUtilsTests() {
210214 setResourcePath : ( path ) => { } ,
211215 getEndpointAuthorizationScheme : ( key , optional ) : string => "none"
212216 } ;
213- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
217+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
214218
215219 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
216220 assert . throws ( ( ) => serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey ) ) ;
@@ -235,9 +239,10 @@ export function serviceConnectionUtilsTests() {
235239 }
236240 return values [ key ] ;
237241 } ,
238- setResourcePath : ( path ) => { }
242+ setResourcePath : ( path ) => { } ,
243+ setSecret : msg => null
239244 } ;
240- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
245+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
241246
242247 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
243248 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey , [ "key1" , "key2" ] ) , [ < TokenServiceConnection > {
@@ -270,9 +275,10 @@ export function serviceConnectionUtilsTests() {
270275 "key2" : "value2"
271276 }
272277 return values [ key ] ;
273- }
278+ } ,
279+ setSecret : msg => null
274280 } ;
275- mockery . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
281+ mocker . registerMock ( 'azure-pipelines-task-lib/task' , mockTask ) ;
276282
277283 let serviceConnectionUtilsWithMocks = require ( "../serviceConnectionUtils" ) ;
278284 assert . deepEqual ( serviceConnectionUtilsWithMocks . getPackagingServiceConnections ( serviceConnectionsKey , [ "key1" , "key2" ] ) , [ < TokenServiceConnection > {
0 commit comments