Пример #1
0
    def test_returns_empty_list_if_no_such_tag_exists(self):
        tag = Tag('cat')
        table = Mock()
        table.get_item.side_effect = DynamoDBKeyNotFoundError('no such tag')
        tag._stored_table = table

        eq_(tag.get_filenames(), [])
Пример #2
0
    def test_appended_filenames_are_unique(self):
        table = Mock()
        item = MagicMock()
        item.__getitem__.return_value = '[ "Sewing_cat.gif" ]'
        table.get_item.return_value = item

        tag = Tag('cat')
        tag._stored_table = table
        tag.add_file('Sewing_cat.gif')
        item.__setitem__.assert_called_with('filenames', '["Sewing_cat.gif"]')
Пример #3
0
    def test_get_filenames(self):
        item = MagicMock()
        item.__getitem__.return_value = '["BADCAFE", "DEADBEEF"]'
        table = Mock()
        table.get_item.return_value = item
        tag = Tag('cat')
        tag._stored_table = table

        filenames = tag.get_filenames()
        eq_(filenames, ['BADCAFE', 'DEADBEEF'])
        table.get_item.assert_called_with('cat')
Пример #4
0
    def test_sends_to_dynamo(self):
        item = MagicMock()
        table = Mock()
        table.get_item.side_effect = DynamoDBKeyNotFoundError('no such tag')
        table.new_item.return_value = item
        tag = Tag('cat')
        tag._stored_table = table

        tag.add_file('Sewing_cat.gif')
        table.new_item.assert_called_with(hash_key='cat', attrs ={})
        item.__setitem__.assert_called_with('filenames', '["Sewing_cat.gif"]')
        item.put.assert_called_with()
Пример #5
0
    def test_updates_existing_tag(self):
        table = Mock()
        item = MagicMock()
        item.__getitem__.return_value = '[ "Sewing_cat.gif" ]'
        table.get_item.return_value = item

        tag = Tag('cat')
        tag._stored_table = table
        tag.add_file('other_cat.gif')

        eq_(table.new_item.call_count, 0, "shouldn't've made a new entry")
        item.__setitem__.assert_called_with('filenames',
                '["Sewing_cat.gif", "other_cat.gif"]')
        item.put.assert_called_with()