def test_credscache_add_new_sp_creds(self, _, mock_open_for_write, mock_read_file): test_sp = { "servicePrincipalId": "myapp", "servicePrincipalTenant": "mytenant", "accessToken": "Secret" } test_sp2 = { "servicePrincipalId": "myapp2", "servicePrincipalTenant": "mytenant2", "accessToken": "Secret2" } mock_open_for_write.return_value = FileHandleStub() mock_read_file.return_value = [self.token_entry1, test_sp] creds_cache = CredsCache() #action creds_cache.save_service_principal_cred( test_sp2['servicePrincipalId'], test_sp2['accessToken'], test_sp2['servicePrincipalTenant']) #assert token_entries = [ entry for _, entry in creds_cache.adal_token_cache.read_items() ] self.assertEqual(token_entries, [self.token_entry1]) self.assertEqual(creds_cache._service_principal_creds, [test_sp, test_sp2]) mock_open_for_write.assert_called_with(mock.ANY, 'w+')
def test_credscache_add_new_sp_creds(self, _, mock_open_for_write, mock_read_file): test_sp = { "servicePrincipalId": "myapp", "servicePrincipalTenant": "mytenant", "accessToken": "Secret" } test_sp2 = { "servicePrincipalId": "myapp2", "servicePrincipalTenant": "mytenant2", "accessToken": "Secret2" } mock_open_for_write.return_value = FileHandleStub() mock_read_file.return_value = [self.token_entry1, test_sp] creds_cache = CredsCache() #action creds_cache.save_service_principal_cred( test_sp2['servicePrincipalId'], test_sp2['accessToken'], test_sp2['servicePrincipalTenant']) #assert token_entries = [entry for _, entry in creds_cache.adal_token_cache.read_items()] self.assertEqual(token_entries, [self.token_entry1]) self.assertEqual(creds_cache._service_principal_creds, [test_sp, test_sp2]) mock_open_for_write.assert_called_with(mock.ANY, 'w+')
def test_credscache_add_preexisting_sp_creds(self, _, mock_open_for_write, mock_read_file): test_sp = { "servicePrincipalId": "myapp", "servicePrincipalTenant": "mytenant", "accessToken": "Secret" } mock_open_for_write.return_value = FileHandleStub() mock_read_file.return_value = [test_sp] creds_cache = CredsCache(async_persist=False) # action creds_cache.save_service_principal_cred(test_sp) # assert self.assertEqual(creds_cache._service_principal_creds, [test_sp])
def test_credscache_add_preexisting_sp_creds(self, _, mock_open_for_write, mock_read_file): test_sp = { "servicePrincipalId": "myapp", "servicePrincipalTenant": "mytenant", "accessToken": "Secret" } mock_open_for_write.return_value = FileHandleStub() mock_read_file.return_value = [test_sp] creds_cache = CredsCache() # action creds_cache.save_service_principal_cred(test_sp) # assert self.assertEqual(creds_cache._service_principal_creds, [test_sp])