def test_EntityViewSchema__ignore_annotation_column_names():
syn = synapseclient.client.Synapse(debug=True, skip_checks=True)
scopeIds = ['123']
entity_view = EntityViewSchema("someName",
scopes=scopeIds,
parent="syn123",
ignoredAnnotationColumnNames={'long1'},
addDefaultViewColumns=False,
addAnnotationColumns=True)
mocked_annotation_result1 = [
Column(name='long1', columnType='INTEGER'),
Column(name='long2', columnType='INTEGER')
]
with patch.object(syn, '_get_annotation_entity_view_columns', return_value=mocked_annotation_result1) as mocked_get_annotations,\
patch.object(syn, 'getColumns') as mocked_get_columns,\
patch.object(SchemaBase, "_before_synapse_store"):
entity_view._before_synapse_store(syn)
mocked_get_columns.assert_called_once_with([])
mocked_get_annotations.assert_called_once_with(scopeIds, 'file')
assert_equals([Column(name='long2', columnType='INTEGER')],
entity_view.columns_to_store)
def test_EntityViewSchema__before_synapse_store(syn):
syn = Synapse(debug=True, skip_checks=True)
with patch.object(syn, '_get_default_view_columns') as mocked_get_default,\
patch.object(syn, '_get_annotation_view_columns') as mocked_get_annotations,\
patch.object(SchemaBase, "_before_synapse_store"):
submission_view = EntityViewSchema(scopes=['syn123'], parent="idk")
submission_view._before_synapse_store(syn)
mocked_get_default.assert_called_once_with("entityview",
view_type_mask=1)
mocked_get_annotations.assert_called_once_with(['syn123'],
"entityview",
view_type_mask=1)
def test_EntityViewSchema__ignore_annotation_column_names():
syn = synapseclient.client.Synapse(debug=True, skip_checks=True)
scopeIds = ['123']
entity_view = EntityViewSchema("someName", scopes=scopeIds, parent="syn123", ignoredAnnotationColumnNames={'long1'},
addDefaultViewColumns=False, addAnnotationColumns=True)
mocked_annotation_result1 = [Column(name='long1', columnType='INTEGER'), Column(name='long2',
columnType='INTEGER')]
with patch.object(syn, '_get_annotation_entity_view_columns', return_value=mocked_annotation_result1)\
as mocked_get_annotations,\
patch.object(syn, 'getColumns') as mocked_get_columns,\
patch.object(SchemaBase, "_before_synapse_store"):
entity_view._before_synapse_store(syn)
mocked_get_columns.assert_called_once_with([])
mocked_get_annotations.assert_called_once_with(scopeIds, EntityViewType.FILE.value)
assert_equals([Column(name='long2', columnType='INTEGER')], entity_view.columns_to_store)