"""Test helpers for swf-feed-sender."""

from unittest.mock import patch

from feed_sender.flows import helpers


@patch('feed_sender.flows.helpers.boto3')
def test_set_status(_):
    """Test set status task."""
    mock_feed_name = 'Apple Music Samis Smart'
    mock_date = '2015-08-03'
    helpers.set_status(mock_feed_name, mock_date)


@patch('feed_sender.flows.helpers.boto3')
def test_get_status(mock_boto):
    """Test get status."""
    mock_boto.client.return_value.get_item.return_value = {'Item': {'status': {'S': 'Sent'}}}
    actual_status = helpers.get_status('feed_name', '2015-04-02')
    assert actual_status == 'Sent'


@patch('feed_sender.flows.helpers.boto3')
def test_delete_status(mock_boto):
    """Test delete feed status."""
    mock_boto.client.return_value.delete_item.return_value = None
    helpers.delete_status('test_feed_name', '2016-01-01')
    mock_boto.client.return_value.delete_item.assert_any_call(
        Key={'date': {'S': '2016-01-01'}, 'feed_name': {'S': 'test_feed_name'}},
        TableName='fake-table-name'
    )