55 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
			
		
		
	
	
			55 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			JavaScript
		
	
	
| /*
 | |
| Copyright 2019 The Matrix.org Foundation C.I.C.
 | |
| 
 | |
| Licensed under the Apache License, Version 2.0 (the "License");
 | |
| you may not use this file except in compliance with the License.
 | |
| You may obtain a copy of the License at
 | |
| 
 | |
|     http://www.apache.org/licenses/LICENSE-2.0
 | |
| 
 | |
| Unless required by applicable law or agreed to in writing, software
 | |
| distributed under the License is distributed on an "AS IS" BASIS,
 | |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | |
| See the License for the specific language governing permissions and
 | |
| limitations under the License.
 | |
| */
 | |
| 
 | |
| import ScalarAuthClient from '../src/ScalarAuthClient';
 | |
| import { MatrixClientPeg } from '../src/MatrixClientPeg';
 | |
| import { stubClient } from './test-utils';
 | |
| 
 | |
| describe('ScalarAuthClient', function() {
 | |
|     beforeEach(function() {
 | |
|         window.localStorage.getItem = jest.fn((arg) => {
 | |
|             if (arg === "mx_scalar_token") return "brokentoken";
 | |
|         });
 | |
|         stubClient();
 | |
|     });
 | |
| 
 | |
|     it('should request a new token if the old one fails', async function() {
 | |
|         const sac = new ScalarAuthClient();
 | |
| 
 | |
|         sac.getAccountName = jest.fn((arg) => {
 | |
|             switch (arg) {
 | |
|                 case "brokentoken":
 | |
|                     return Promise.reject({
 | |
|                         message: "Invalid token",
 | |
|                     });
 | |
|                 case "wokentoken":
 | |
|                     return Promise.resolve(MatrixClientPeg.get().getUserId());
 | |
|             }
 | |
|         });
 | |
| 
 | |
|         MatrixClientPeg.get().getOpenIdToken = jest.fn().mockResolvedValue('this is your openid token');
 | |
| 
 | |
|         sac.exchangeForScalarToken = jest.fn((arg) => {
 | |
|             if (arg === "this is your openid token") return Promise.resolve("wokentoken");
 | |
|         });
 | |
| 
 | |
|         await sac.connect();
 | |
| 
 | |
|         expect(sac.exchangeForScalarToken).toBeCalledWith('this is your openid token');
 | |
|         expect(sac.scalarToken).toEqual('wokentoken');
 | |
|     });
 | |
| });
 |