def test_export_diff(self):
CellFactory.create_batch(10, radio=Radio.gsm)
self.session.commit()
with mock_s3() as mock_key:
cell_export_diff(_bucket='localhost.bucket')
pat = r'MLS-diff-cell-export-\d+-\d+-\d+T\d+0000\.csv\.gz'
self.assertRegex(mock_key.key, pat)
method = mock_key.set_contents_from_filename
self.assertRegex(method.call_args[0][0], pat)
def test_export_diff(self):
CellShardFactory.create_batch(10, radio=Radio.gsm)
self.session.commit()
with mock_s3() as mock_key:
cell_export_diff(_bucket='localhost.bucket')
pat = r'MLS-diff-cell-export-\d+-\d+-\d+T\d+0000\.csv\.gz'
self.assertRegex(mock_key.key, pat)
method = mock_key.set_contents_from_filename
self.assertRegex(method.call_args[0][0], pat)
def test_export_diff(self, celery, session):
CellShardFactory.create_batch(10, radio=Radio.gsm)
session.commit()
pattern = re.compile(
r'MLS-diff-cell-export-\d+-\d+-\d+T\d+0000\.csv\.gz')
with mock_s3() as mock_key:
cell_export_diff(_bucket='localhost.bucket')
assert pattern.search(mock_key.key)
method = mock_key.set_contents_from_filename
assert pattern.search(method.call_args[0][0])
def test_export_diff(self, celery, session):
CellShardFactory.create_batch(10, radio=Radio.gsm)
session.commit()
pattern = re.compile(r"MLS-diff-cell-export-\d+-\d+-\d+T\d+0000\.csv\.gz")
mock_conn = mock.MagicMock()
mock_bucket = mock.MagicMock(name="bucket")
mock_obj = mock.MagicMock()
mock_conn.return_value.Bucket.return_value = mock_bucket
mock_bucket.Object.return_value = mock_obj
with mock.patch.object(boto3, "resource", mock_conn):
cell_export_diff(_bucket="bucket")
s3_key = mock_bucket.Object.call_args[0][0]
assert pattern.search(s3_key)
tmp_file = mock_obj.upload_file.call_args[0][0]
assert pattern.search(tmp_file)
def test_export_diff(self, celery, session):
CellShardFactory.create_batch(10, radio=Radio.gsm)
session.commit()
pattern = re.compile(
r'MLS-diff-cell-export-\d+-\d+-\d+T\d+0000\.csv\.gz')
mock_conn = mock.MagicMock()
mock_bucket = mock.MagicMock(name='bucket')
mock_obj = mock.MagicMock()
mock_conn.return_value.Bucket.return_value = mock_bucket
mock_bucket.Object.return_value = mock_obj
with mock.patch.object(boto3, 'resource', mock_conn):
cell_export_diff(_bucket='bucket')
s3_key = mock_bucket.Object.call_args[0][0]
assert pattern.search(s3_key)
tmp_file = mock_obj.upload_file.call_args[0][0]
assert pattern.search(tmp_file)