@@ -240,6 +240,33 @@ describe('al_aws_collector tests', function() {
240240 } ) ;
241241 } ) ;
242242
243+ it ( 'checkin forced update success' , function ( done ) {
244+ alserviceStub . post . restore ( ) ;
245+ alserviceStub . post = sinon . stub ( m_alCollector . AlServiceC . prototype , 'post' ) . callsFake (
246+ function fakeFn ( path , extraOptions ) {
247+ return new Promise ( function ( resolve , reject ) {
248+ return resolve ( { force_update : true } ) ;
249+ } ) ;
250+ } ) ;
251+ let fakeSelfUpdate = sinon . stub ( AlAwsCollector . prototype , 'selfUpdate' ) . callsFake (
252+ ( callback ) => { callback ( ) ; } ) ;
253+ let fakeSelfConfigUpdate = sinon . stub ( AlAwsCollector . prototype , 'selfConfigUpdate' ) . callsFake (
254+ ( callback ) => { callback ( ) ; } ) ;
255+ AlAwsCollector . load ( ) . then ( function ( creds ) {
256+ var collector = new AlAwsCollector (
257+ mockContext , 'cwe' , AlAwsCollector . IngestTypes . SECMSGS , '1.0.0' , creds , undefined , [ ] , [ ] ) ;
258+ collector . checkin ( function ( error ) {
259+ assert . equal ( error , undefined ) ;
260+ sinon . assert . calledWith ( alserviceStub . post , colMock . CHECKIN_URL , colMock . CHECKIN_AZCOLLECT_QUERY ) ;
261+ sinon . assert . called ( fakeSelfConfigUpdate ) ;
262+ sinon . assert . called ( fakeSelfUpdate ) ;
263+ fakeSelfUpdate . restore ( ) ;
264+ fakeSelfConfigUpdate . restore ( ) ;
265+ done ( ) ;
266+ } ) ;
267+ } ) ;
268+ } ) ;
269+
243270 it ( 'checkin with custom check success' , function ( done ) {
244271 AlAwsCollector . load ( ) . then ( function ( creds ) {
245272 var spyHealthCheck = sinon . spy ( function ( callback ) {
0 commit comments