Python FileFormatAwareContainer.get Examples

Example #1

File: format_aware_container_test.py Project: empire/schallenge-python-2014

def test_container():
    format_aware = Mock()
    format_aware.supported_format.return_value = 'abc'

    container = FileFormatAwareContainer()
    assert not container.is_format_supported('abc')
    container.register(format_aware)
    format_aware.supported_format.assert_called_once_with()
    assert container.is_format_supported('abc')
    assert container.get('abc') == format_aware

Example #2

File: format_aware_container_test.py Project: Sajjad-HTLO/schallenge-python-2014

def test_container():
    format_aware = Mock()
    format_aware.supported_format.return_value = 'abc'

    container = FileFormatAwareContainer()
    assert not container.is_format_supported('abc')
    container.register(format_aware)
    format_aware.supported_format.assert_called_once_with()
    assert container.is_format_supported('abc')
    assert container.get('abc') == format_aware