class TestCustomAttributesDefinitions(TestCase):
    """Test for custom attribute definition columns."""
    def setUp(self):
        super(TestCustomAttributesDefinitions, self).setUp()
        self.generator = ObjectGenerator()

    def test_policy_definitions(self):
        """Test custom attribute definitions on Policy model."""
        self.generator.generate_custom_attribute("policy",
                                                 title="My Attribute")
        self.generator.generate_custom_attribute("policy",
                                                 title="Mandatory Attribute",
                                                 mandatory=True)
        definitions = get_object_column_definitions(all_models.Policy)
        mapping_names = get_mapping_names(all_models.Policy.__name__)
        display_names = {
            val["display_name"]
            for val in definitions.itervalues()
        }
        element_names = {
            "Code",
            "Title",
            "Description",
            "Notes",
            "Admin",
            "Reference URL",
            "Kind/Type",
            "Effective Date",
            "Last Deprecated Date",
            "State",
            "My Attribute",
            "Mandatory Attribute",
            "Review State",
            "Reviewers",
            "Delete",
            "Primary Contacts",
            "Secondary Contacts",
            "Recipients",
            "Send by default",
            "Comments",
            "Assessment Procedure",
            "Created Date",
            "Last Updated Date",
            "Last Updated By",
            "Gdrive Folder ID",
        }
        expected_names = element_names.union(mapping_names)
        self.assertEqual(expected_names, display_names)
        vals = {val["display_name"]: val for val in definitions.itervalues()}
        self.assertTrue(vals["Title"]["mandatory"])
        self.assertTrue(vals["Admin"]["mandatory"])
        self.assertTrue(vals["Title"]["unique"])
        self.assertTrue(vals["Mandatory Attribute"]["mandatory"])

    def test_program_definitions(self):
        """ test custom attribute headers for Program."""

        self.generator.generate_custom_attribute("program",
                                                 title="My Attribute")
        self.generator.generate_custom_attribute("program",
                                                 title="Mandatory Attribute",
                                                 mandatory=True)
        self.generator.generate_custom_attribute(
            "program",
            title="Choose",
            mandatory=True,
            attribute_type="Dropdown",
            multi_choice="hello,world,what's,up")
        definitions = get_object_column_definitions(all_models.Program)
        mapping_names = get_mapping_names(all_models.Program.__name__)
        display_names = {
            val["display_name"]
            for val in definitions.itervalues()
        }
        element_names = {
            "Title",
            "Description",
            "Notes",
            "Reference URL",
            "Code",
            "Effective Date",
            "Last Deprecated Date",
            "State",
            "My Attribute",
            "Mandatory Attribute",
            "Choose",
            "Review State",
            "Reviewers",
            "Delete",
            "Primary Contacts",
            "Secondary Contacts",
            "Program Managers",
            "Program Editors",
            "Program Readers",
            "Created Date",
            "Last Updated Date",
            "Last Updated By",
            "Gdrive Folder ID",
            "Send by default",
            "Recipients",
            "Comments",
        }
        expected_names = element_names.union(mapping_names)
        self.assertEqual(expected_names, display_names)
        vals = {val["display_name"]: val for val in definitions.itervalues()}
        self.assertTrue(vals["Title"]["mandatory"])
        self.assertTrue(vals["Title"]["unique"])
        self.assertTrue(vals["Mandatory Attribute"]["mandatory"])
        self.assertTrue(vals["Choose"]["mandatory"])
class TestCustomAttributesDefinitions(TestCase):

  def setUp(self):
    TestCase.setUp(self)
    self.generator = ObjectGenerator()

  def test_policy_definitions(self):
    self.generator.generate_custom_attribute("policy", title="My Attribute")
    self.generator.generate_custom_attribute(
        "policy", title="Mandatory Attribute", mandatory=True)
    definitions = get_object_column_definitions(models.Policy)
    mapping_names = get_mapping_names(models.Policy.__name__)
    display_names = {val["display_name"] for val in definitions.itervalues()}
    element_names = {
        "Code",
        "Title",
        "Description",
        "Notes",
        "Owner",
        "Primary Contact",
        "Secondary Contact",
        "Policy URL",
        "Reference URL",
        "Kind/Type",
        "Effective Date",
        "Stop Date",
        "State",
        "My Attribute",
        "Mandatory Attribute",
        "Delete",
    }
    expected_names = element_names.union(mapping_names)
    self.assertEqual(expected_names, display_names)
    vals = {val["display_name"]: val for val in definitions.itervalues()}
    self.assertTrue(vals["Title"]["mandatory"])
    self.assertTrue(vals["Owner"]["mandatory"])
    self.assertTrue(vals["Title"]["unique"])
    self.assertTrue(vals["Mandatory Attribute"]["mandatory"])

  def test_program_definitions(self):
    """ test custom attribute headers for Program """

    self.generator.generate_custom_attribute(
        "program",
        title="My Attribute")
    self.generator.generate_custom_attribute(
        "program",
        title="Mandatory Attribute",
        mandatory=True)
    self.generator.generate_custom_attribute(
        "program",
        title="Choose",
        mandatory=True,
        attribute_type="Dropdown",
        multi_choice="hello,world,what's,up"
    )
    definitions = get_object_column_definitions(models.Program)
    mapping_names = get_mapping_names(models.Program.__name__)
    display_names = {val["display_name"] for val in definitions.itervalues()}
    element_names = {
        "Title",
        "Description",
        "Notes",
        "Manager",
        "Reader",
        "Editor",
        "No Access",
        "Primary Contact",
        "Secondary Contact",
        "Program URL",
        "Reference URL",
        "Code",
        "Effective Date",
        "Stop Date",
        "State",
        "My Attribute",
        "Mandatory Attribute",
        "Choose",
        "Delete",
    }
    expected_names = element_names.union(mapping_names)
    self.assertEqual(expected_names, display_names)
    vals = {val["display_name"]: val for val in definitions.itervalues()}
    self.assertTrue(vals["Title"]["mandatory"])
    self.assertTrue(vals["Manager"]["mandatory"])
    self.assertTrue(vals["Title"]["unique"])
    self.assertTrue(vals["Mandatory Attribute"]["mandatory"])
    self.assertTrue(vals["Choose"]["mandatory"])
class TestCustomAttributesDefinitions(TestCase):

  def setUp(self):
    super(TestCustomAttributesDefinitions, self).setUp()
    self.generator = ObjectGenerator()

  def test_policy_definitions(self):
    self.generator.generate_custom_attribute("policy", title="My Attribute")
    self.generator.generate_custom_attribute(
        "policy", title="Mandatory Attribute", mandatory=True)
    definitions = get_object_column_definitions(models.Policy)
    mapping_names = get_mapping_names(models.Policy.__name__)
    display_names = {val["display_name"] for val in definitions.itervalues()}
    element_names = {
        "Code",
        "Title",
        "Description",
        "Notes",
        "Owner",
        "Primary Contact",
        "Secondary Contact",
        "Policy URL",
        "Reference URL",
        "Kind/Type",
        "Effective Date",
        "Stop Date",
        "State",
        "My Attribute",
        "Mandatory Attribute",
        "Review State",
        "Delete",
    }
    expected_names = element_names.union(mapping_names)
    self.assertEqual(expected_names, display_names)
    vals = {val["display_name"]: val for val in definitions.itervalues()}
    self.assertTrue(vals["Title"]["mandatory"])
    self.assertTrue(vals["Owner"]["mandatory"])
    self.assertTrue(vals["Title"]["unique"])
    self.assertTrue(vals["Mandatory Attribute"]["mandatory"])

  def test_program_definitions(self):
    """ test custom attribute headers for Program """

    self.generator.generate_custom_attribute(
        "program",
        title="My Attribute")
    self.generator.generate_custom_attribute(
        "program",
        title="Mandatory Attribute",
        mandatory=True)
    self.generator.generate_custom_attribute(
        "program",
        title="Choose",
        mandatory=True,
        attribute_type="Dropdown",
        multi_choice="hello,world,what's,up"
    )
    definitions = get_object_column_definitions(models.Program)
    mapping_names = get_mapping_names(models.Program.__name__)
    display_names = {val["display_name"] for val in definitions.itervalues()}
    element_names = {
        "Title",
        "Description",
        "Notes",
        "Manager",
        "Reader",
        "Editor",
        "Primary Contact",
        "Secondary Contact",
        "Program URL",
        "Reference URL",
        "Code",
        "Effective Date",
        "Stop Date",
        "State",
        "My Attribute",
        "Mandatory Attribute",
        "Choose",
        "Review State",
        "Delete",
    }
    expected_names = element_names.union(mapping_names)
    self.assertEqual(expected_names, display_names)
    vals = {val["display_name"]: val for val in definitions.itervalues()}
    self.assertTrue(vals["Title"]["mandatory"])
    self.assertTrue(vals["Manager"]["mandatory"])
    self.assertTrue(vals["Title"]["unique"])
    self.assertTrue(vals["Mandatory Attribute"]["mandatory"])
    self.assertTrue(vals["Choose"]["mandatory"])
Exemple #4
0
class TestCustomAttributesDefinitions(TestCase):
  """Test for custom attribute definition columns."""

  def setUp(self):
    super(TestCustomAttributesDefinitions, self).setUp()
    self.generator = ObjectGenerator()

  def test_policy_definitions(self):
    """Test custom attribute definitions on Policy model."""
    self.generator.generate_custom_attribute("policy", title="My Attribute")
    self.generator.generate_custom_attribute(
        "policy", title="Mandatory Attribute", mandatory=True)
    definitions = get_object_column_definitions(all_models.Policy)
    mapping_names = get_mapping_names(all_models.Policy.__name__)
    display_names = {val["display_name"] for val in definitions.itervalues()}
    element_names = {
        "Code",
        "Title",
        "Description",
        "Notes",
        "Admin",
        "Reference URL",
        "Kind/Type",
        "Effective Date",
        "Last Deprecated Date",
        "State",
        "My Attribute",
        "Mandatory Attribute",
        "Review State",
        "Reviewers",
        "Delete",
        "Primary Contacts",
        "Secondary Contacts",
        "Recipients",
        "Send by default",
        "Comments",
        "Assessment Procedure",
        "Created Date",
        "Last Updated Date",
        "Last Updated By",
        "Folder",
    }
    expected_names = element_names.union(mapping_names)
    self.assertEqual(expected_names, display_names)
    vals = {val["display_name"]: val for val in definitions.itervalues()}
    self.assertTrue(vals["Title"]["mandatory"])
    self.assertTrue(vals["Admin"]["mandatory"])
    self.assertTrue(vals["Title"]["unique"])
    self.assertTrue(vals["Mandatory Attribute"]["mandatory"])

  def test_program_definitions(self):
    """ test custom attribute headers for Program."""

    self.generator.generate_custom_attribute(
        "program",
        title="My Attribute")
    self.generator.generate_custom_attribute(
        "program",
        title="Mandatory Attribute",
        mandatory=True)
    self.generator.generate_custom_attribute(
        "program",
        title="Choose",
        mandatory=True,
        attribute_type="Dropdown",
        multi_choice="hello,world,what's,up"
    )
    definitions = get_object_column_definitions(all_models.Program)
    mapping_names = get_mapping_names(all_models.Program.__name__)
    display_names = {val["display_name"] for val in definitions.itervalues()}
    element_names = {
        "Title",
        "Description",
        "Notes",
        "Reference URL",
        "Code",
        "Effective Date",
        "Last Deprecated Date",
        "State",
        "My Attribute",
        "Mandatory Attribute",
        "Choose",
        "Review State",
        "Reviewers",
        "Delete",
        "Primary Contacts",
        "Secondary Contacts",
        "Program Managers",
        "Program Editors",
        "Program Readers",
        "Created Date",
        "Last Updated Date",
        "Last Updated By",
        "Folder",
        "Send by default",
        "Recipients",
        "Comments",
    }
    expected_names = element_names.union(mapping_names)
    self.assertEqual(expected_names, display_names)
    vals = {val["display_name"]: val for val in definitions.itervalues()}
    self.assertTrue(vals["Title"]["mandatory"])
    self.assertTrue(vals["Title"]["unique"])
    self.assertTrue(vals["Mandatory Attribute"]["mandatory"])
    self.assertTrue(vals["Choose"]["mandatory"])