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)
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)