From 4f884cd2359182924e2e8bb7eb88801bc2414db2 Mon Sep 17 00:00:00 2001
From: Vyazovetskov Vladimir <vyazovetckov_vs@magnit.ru>
Date: Sat, 28 May 2022 14:49:56 +0300
Subject: [PATCH] Change signature to more AWS-way.

---
 s3_objects_tracker/tracker.py | 8 ++++----
 tests/conftest.py             | 2 +-
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/s3_objects_tracker/tracker.py b/s3_objects_tracker/tracker.py
index 32e77e1..935ed8d 100644
--- a/s3_objects_tracker/tracker.py
+++ b/s3_objects_tracker/tracker.py
@@ -17,14 +17,14 @@ class ObjectWithIDProtocol(Protocol[IDT]):
 class S3ObjectsTracker(Generic[IDT]):
     def __init__(
         self,
-        bucket: str,
+        bucket_name: str,
         endpoint_url: str,
         aws_access_key_id: str,
         aws_secret_access_key: str,
         filename: str = "data.json",
         max_published_objects: int = 50,
     ):
-        self.bucket = bucket
+        self.bucket_name = bucket_name
         self.endpoint_url = endpoint_url
         self.aws_access_key_id = aws_access_key_id
         self.aws_secret_access_key = aws_secret_access_key
@@ -37,7 +37,7 @@ def __init__(
 
     async def _fetch_from_s3(self) -> None:
         try:
-            response = await self._s3_client.get_object(Bucket=self.bucket, Key=self.filename)
+            response = await self._s3_client.get_object(Bucket=self.bucket_name, Key=self.filename)
         except self._s3_client.exceptions.NoSuchKey:
             self._published_ids = []
             return
@@ -49,7 +49,7 @@ async def _upload_to_s3(self) -> None:
         self._published_ids = self._published_ids[-self.max_published_objects :]
         json_str = json.dumps(self._published_ids) + "\n"
         await self._s3_client.put_object(
-            Bucket=self.bucket,
+            Bucket=self.bucket_name,
             Key=self.filename,
             Body=json_str.encode("utf-8"),
             ContentType="application/json",
diff --git a/tests/conftest.py b/tests/conftest.py
index 5d20d48..abd77af 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -94,7 +94,7 @@ def empty_object_storage_mock():
 @pytest.fixture(scope="session")
 def s3_credentials():
     return {
-        "bucket": "test-bucket",
+        "bucket_name": "test-bucket",
         "endpoint_url": "http://localhost:9000",
         "aws_access_key_id": "test-access-key-id",
         "aws_secret_access_key": "test-secret-access-key",