Code refractor

This commit is contained in:
Anurag Mishra 2019-02-06 12:34:10 +05:30
parent 515d262881
commit 86f37dc0ea
2 changed files with 38 additions and 51 deletions

View file

@ -10,40 +10,19 @@ import unittest
class TestUserPermission(unittest.TestCase):
def test_apply_to_all(self):
user = get_user()
created = add_user_permissions({
"user": user.name,
"doctype":"User",
"docname":user.name ,
"apply_to_all_doctypes":1})
param = get_params(user, apply = 1)
created = add_user_permissions(param)
self.assertEquals(created, 1)
def test_for_applicables_on_update_from_apply_to_all(self):
user = get_user()
create = add_user_permissions({
"user": user.name,
"doctype":"User",
"docname":user.name ,
"apply_to_all_doctypes":0,
"applicable_doctypes":["Chat Room","Chat Message"]})
param = get_params(user, applicable = ["Chat Room", "Chat Message"])
create = add_user_permissions(param)
frappe.db.commit()
removed_apply_to_all = frappe.db.exists("User Permission", {
"user": user.name,
"allow": "User",
"for_value": user.name,
"apply_to_all_doctypes": 1})
created_applicable_first = frappe.db.exists("User Permission", {
"user": user.name,
"allow": "User",
"for_value": user.name,
"apply_to_all_doctypes": 0,
"applicable_for": "Chat Room"})
created_applicable_second = frappe.db.exists("User Permission", {
"user": user.name,
"allow": "User",
"for_value": user.name,
"apply_to_all_doctypes": 0,
"applicable_for": "Chat Message"})
removed_apply_to_all = frappe.db.exists("User Permission", get_exists_param(user))
created_applicable_first = frappe.db.exists("User Permission", get_exists_param(user, applicable = "Chat Room"))
created_applicable_second = frappe.db.exists("User Permission", get_exists_param(user, applicable = "Chat Message"))
self.assertIsNone(removed_apply_to_all)
self.assertIsNotNone(created_applicable_first)
@ -52,27 +31,11 @@ class TestUserPermission(unittest.TestCase):
def test_for_apply_to_all_on_update_from_applicables(self):
user = get_user()
created = add_user_permissions({
"user": user.name,
"doctype":"User",
"docname":user.name ,
"apply_to_all_doctypes":1})
created_apply_to_all = frappe.db.exists("User Permission", {
"user": user.name,
"allow": "User",
"for_value": user.name,
"apply_to_all_doctypes": 1})
removed_applicable_first = frappe.db.exists("User Permission", {
"user": user.name,
"allow": "User",
"for_value": user.name,
"apply_to_all_doctypes": 0,
"applicable_for": "Chat Room"})
removed_applicable_second = frappe.db.exists("User Permission", {
"user": user.name, "allow": "User",
"for_value": user.name,
"apply_to_all_doctypes": 0,
"applicable_for": "Chat Message"})
param = get_params(user, apply = 1)
created = add_user_permissions(param)
created_apply_to_all = frappe.db.exists("User Permission", get_exists_param(user))
removed_applicable_first = frappe.db.exists("User Permission", get_exists_param(user, applicable = "Chat Room"))
removed_applicable_second = frappe.db.exists("User Permission", get_exists_param(user, applicable = "Chat Message"))
self.assertIsNotNone(created_apply_to_all)
@ -90,4 +53,28 @@ def get_user():
user.add_roles("System Manager")
return user
def get_params(user, apply = None , applicable= None):
param = {
"user": user.name,
"doctype":"User",
"docname":user.name
}
if apply:
param.update({"apply_to_all_doctypes": 1})
param.update({"applicable_doctypes": []})
if applicable:
param.update({"apply_to_all_doctypes": 0})
param.update({"applicable_doctypes": applicable})
return param
def get_exists_param(user, applicable = None):
param = {
"user": user.name,
"allow": "User",
"for_value": user.name,
}
if applicable:
param.update({"applicable_for": applicable})
else:
param.update({"apply_to_all_doctypes": 1})
return param

View file

@ -3,7 +3,7 @@ frappe.listview_settings['User Permission'] = {
onload: function(list_view) {
var me = this;
list_view.page.add_inner_button( __("Create/Update User Permissions"), function() {
var dialog =new frappe.ui.Dialog({
let dialog =new frappe.ui.Dialog({
title : __('Add User Permissions'),
fields: [
{
@ -208,7 +208,7 @@ frappe.listview_settings['User Permission'] = {
dialog.set_primary_action("Update");
dialog.set_title("Update User Permissions");
dialog.set_df_property("applicable_doctypes", "options", options);
if(dialog.fields_dict.applicable_doctypes.get_checked_options().length == options.length){
if(dialog.fields_dict.applicable_doctypes.get_checked_options().length == options.length) {
dialog.set_df_property("applicable_doctypes", "hidden", 1);
dialog.set_value("apply_to_all_doctypes", "checked", 1);
} else {