def test_get_source_url__returns_none_if_no_such_image(self): table = Mock() table.get_item.side_effect = DynamoDBKeyNotFoundError('no such image') image = Image("TH15I5N0TEVENHEX") image._stored_table = table source_url = image.get_source_url() eq_(source_url, None)
def test_get_source_url(self): table = Mock() item = MagicMock() item.__getitem__.return_value = 'booya.com/woot.gif' table.get_item.return_value = item image = Image('BADD00D') image._stored_table = table source_url = image.get_source_url() eq_(source_url, 'booya.com/woot.gif')