Exemplo n.º 1
0
    def test_extract_all_field_names(self):
        specifier = DataNodeSpecifier(data_node_specifier_test_driver[0])
        key_field_names = specifier.extract_field_names(primary_key_only=None)

        expected_field_names = [
            'risk_description', 'likelihood', 'impact', 'mitigation'
        ]
        self.assertEqual(expected_field_names, key_field_names)
Exemplo n.º 2
0
    def test_extract_primary_key_field_names(self):
        specifier = DataNodeSpecifier(data_node_specifier_test_driver[0])
        key_field_names = specifier.extract_field_names(primary_key_only=True)

        expected_field_names = ['risk_description']
        self.assertEqual(expected_field_names, key_field_names)