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)