nik/newdwh2021
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: コンフィグバケットのテストを追加

Browse Source
This commit is contained in:
shimoda.m@nds-tyo.co.jp 2022-08-02 16:19:32 +09:00
parent f2618831a2
commit ea6ebbc0a1
1 changed files with 25 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
ecs/crm-datafetch/tests/aws/test_s3.py
View File

@ -1,7 +1,5 @@
import pytest import pytest
from src.aws.s3 import ConfigBucket, S3Resource from src.aws.s3 import ConfigBucket, S3Resource
from src.system_var import environments
from src.system_var.constants import OBJECT_INFO_FILENAME, OBJECT_INFO_FOLDER
@pytest.fixture @pytest.fixture
@ -27,7 +25,7 @@ class TestS3Resource:
s3_resource = S3Resource(bucket_name) s3_resource = S3Resource(bucket_name)
s3_resource.put_object('hogehoge/test.txt', 'aaaaaaaaaaaaaaa') s3_resource.put_object('hogehoge/test.txt', 'aaaaaaaaaaaaaaa')
actual = s3_client.get_object(Bucket=bucket_name, Key='hogehoge/test.txt') actual = s3_client.get_object(Bucket=bucket_name, Key='hogehoge/test.txt')
assert actual['Body'].read() == b'aaaaaaaaaaaaaaa' assert actual['Body'].read().decode('utf-8') == 'aaaaaaaaaaaaaaa'
def test_copy(self, s3_test, s3_client, bucket_name): def test_copy(self, s3_test, s3_client, bucket_name):
for_copy_bucket = 'for-copy-bucket' for_copy_bucket = 'for-copy-bucket'
@ -47,7 +45,6 @@ class TestS3Resource:
class TestConfigBucket: class TestConfigBucket:
def test_get_object_info_file(self, s3_test, s3_client, bucket_name, monkeypatch): def test_get_object_info_file(self, s3_test, s3_client, bucket_name, monkeypatch):
# = bucket_name
monkeypatch.setattr('src.aws.s3.CRM_CONFIG_BUCKET', bucket_name) monkeypatch.setattr('src.aws.s3.CRM_CONFIG_BUCKET', bucket_name)
monkeypatch.setattr('src.aws.s3.OBJECT_INFO_FOLDER', 'crm') monkeypatch.setattr('src.aws.s3.OBJECT_INFO_FOLDER', 'crm')
monkeypatch.setattr('src.aws.s3.OBJECT_INFO_FILENAME', 'objects.json') monkeypatch.setattr('src.aws.s3.OBJECT_INFO_FILENAME', 'objects.json')
@ -57,3 +54,27 @@ class TestConfigBucket:
actual = config_bucket.get_object_info_file() actual = config_bucket.get_object_info_file()
print(actual) print(actual)
assert actual == 'aaaaaaaaaaaaaaa' assert actual == 'aaaaaaaaaaaaaaa'
def test_get_last_fetch_datetime_file(self, s3_test, s3_client, bucket_name, monkeypatch):
monkeypatch.setattr('src.aws.s3.CRM_CONFIG_BUCKET', bucket_name)
monkeypatch.setattr('src.aws.s3.LAST_FETCH_DATE_FOLDER', 'crm')
s3_client.put_object(Bucket=bucket_name, Key=f'crm/Object.json', Body=b'aaaaaaaaaaaaaaa')
config_bucket = ConfigBucket()
print('*' * 50, str(config_bucket), '*' * 50)
actual = config_bucket.get_last_fetch_datetime_file('Object.json')
print(actual)
assert actual == 'aaaaaaaaaaaaaaa'
def test_put_last_fetch_datetime_file(self, s3_test, s3_client, bucket_name, monkeypatch):
monkeypatch.setattr('src.aws.s3.CRM_CONFIG_BUCKET', bucket_name)
monkeypatch.setattr('src.aws.s3.LAST_FETCH_DATE_FOLDER', 'crm')
config_bucket = ConfigBucket()
print('*' * 50, str(config_bucket), '*' * 50)
config_bucket.put_last_fetch_datetime_file('Object.json', 'aaaaaaaaaaaaaaa')
actual = s3_client.get_object(Bucket=bucket_name, Key=f'crm/Object.json')
assert actual['Body'].read().decode('utf-8') == 'aaaaaaaaaaaaaaa'
def test_config_bucket_str(self, s3_test, s3_client, bucket_name, monkeypatch):
monkeypatch.setattr('src.aws.s3.CRM_CONFIG_BUCKET', bucket_name)
config_bucket = ConfigBucket()
assert str(config_bucket) == bucket_name