Skip to content

Commit

Permalink
chore: add granule_to_sns test
Browse files Browse the repository at this point in the history
  • Loading branch information
mckadesorensen committed May 3, 2024
1 parent 3e2572a commit 609f733
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 1 deletion.
15 changes: 14 additions & 1 deletion tests/test_granule_to_sns.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
import boto3
import pytest

from granule_to_sns import generate_message
from granule_to_sns import generate_message, granule_to_sns


@pytest.fixture
Expand Down Expand Up @@ -91,5 +92,17 @@ def get_message():
}


@pytest.fixture
def sns_client(mocker):
sns_client = boto3.client("sns", region_name="us-west-2")
mocker.patch("boto3.client", return_value=sns_client)
return sns_client


def test_generate_message(get_event, get_message):
assert generate_message(get_event["input"]["granules"][0]) == get_message


def test_granule_to_sns(sns_client, get_event, mocker):
mocker.patch.object(sns_client, "publish", return_value={})
assert granule_to_sns(get_event, None) == get_event
1 change: 1 addition & 0 deletions tox.ini
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,7 @@ minversion = 4.4.6
[testenv]
deps =
pytest>=7
pytest-mock>=3.14.0
-r src/requirements.txt
commands =
pytest {tty:--color=yes} {posargs}

0 comments on commit 609f733

Please sign in to comment.