def test_update_users(self): flexmock(AppDashboardHelper).should_receive('list_all_users')\ .and_return(['*****@*****.**', '*****@*****.**', '*****@*****.**']).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app1', 'app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app2']).once() self.setupUserInfoMocks() data1 = AppDashboardData() output = data1.update_users() self.assertEquals(len(output), 3) self.assertTrue(output[0].is_user_cloud_admin) self.assertFalse(output[1].is_user_cloud_admin) self.assertFalse(output[2].is_user_cloud_admin) self.assertTrue(output[0].can_upload_apps) self.assertTrue(output[1].can_upload_apps) self.assertFalse(output[2].can_upload_apps)
def test_update_users(self): flexmock(ndb).should_receive('put_multi').and_return() flexmock(AppDashboardHelper).should_receive('list_all_users')\ .and_return(['*****@*****.**', '*****@*****.**', '*****@*****.**', '*****@*****.**']).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin') \ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps') \ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app1', 'app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps') \ .with_args('*****@*****.**').and_return([]).once() self.setupUserInfoMocks() data1 = AppDashboardData() output = data1.update_users() self.assertEquals(len(output), 4) self.assertTrue(output[0].is_user_cloud_admin) self.assertFalse(output[1].is_user_cloud_admin) self.assertFalse(output[2].is_user_cloud_admin) self.assertFalse(output[3].is_user_cloud_admin) self.assertTrue(output[0].can_upload_apps) self.assertTrue(output[1].can_upload_apps) self.assertFalse(output[2].can_upload_apps) self.assertFalse(output[3].can_upload_apps) self.assertEqual( self.flatten_dash_layout(output[0].dash_layout_settings), self.user_info1_cloud_admin_dict) self.assertEqual( self.flatten_dash_layout(output[1].dash_layout_settings), self.user_info2_can_upload_apps_dict) self.assertEqual( self.flatten_dash_layout(output[2].dash_layout_settings), self.user_info3_cannot_upload_apps_dict) self.assertEqual( self.flatten_dash_layout(output[3].dash_layout_settings), self.user_info4_cannot_upload_and_owns_no_apps_dict)
def test_update_users(self): flexmock(ndb).should_receive('put_multi').and_return() flexmock(AppDashboardHelper).should_receive('list_all_users')\ .and_return(['*****@*****.**', '*****@*****.**', '*****@*****.**', '*****@*****.**']).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('is_user_cloud_admin') \ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(True).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps')\ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('can_upload_apps') \ .with_args('*****@*****.**').and_return(False).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app1', 'app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps')\ .with_args('*****@*****.**').and_return(['app2']).once() flexmock(AppDashboardHelper).should_receive('get_owned_apps') \ .with_args('*****@*****.**').and_return([]).once() self.setupUserInfoMocks() data1 = AppDashboardData() output = data1.update_users() self.assertEquals(len(output), 4) self.assertTrue(output[0].is_user_cloud_admin) self.assertFalse(output[1].is_user_cloud_admin) self.assertFalse(output[2].is_user_cloud_admin) self.assertFalse(output[3].is_user_cloud_admin) self.assertTrue(output[0].can_upload_apps) self.assertTrue(output[1].can_upload_apps) self.assertFalse(output[2].can_upload_apps) self.assertFalse(output[3].can_upload_apps) self.assertEqual(self.flatten_dash_layout(output[0].dash_layout_settings), self.user_info1_cloud_admin_dict) self.assertEqual(self.flatten_dash_layout(output[1].dash_layout_settings), self.user_info2_can_upload_apps_dict) self.assertEqual(self.flatten_dash_layout(output[2].dash_layout_settings), self.user_info3_cannot_upload_apps_dict) self.assertEqual(self.flatten_dash_layout(output[3].dash_layout_settings), self.user_info4_cannot_upload_and_owns_no_apps_dict)