def test_snapshots(self, mocked_upload): mocked_upload.return_value = 'http://s3.com/your-file.csv' SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=False) snapshot_queries() self.assertEqual(mocked_upload.call_count, 3)
def test_snapshots(self, mocked_upload): mocked_upload.return_value = 'http://s3.com/your-file.csv' SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=False) snapshot_queries() self.assertEqual(mocked_upload.call_count, 3)
def test_snapshots(self, mocked_s3): conn = Mock() conn.upload = Mock() conn.upload.return_value = type('obj', (object,), {'url': 'http://s3.com/your-file.csv'}) mocked_s3.return_value = conn SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=False) snapshot_queries() self.assertEqual(conn.upload.call_count, 3)
def test_snapshots(self, mocked_s3): conn = Mock() conn.upload = Mock() conn.upload.return_value = type('obj', (object,), {'url': 'http://s3.com/your-file.csv'}) mocked_s3.return_value = conn SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=False) snapshot_queries() self.assertEqual(conn.upload.call_count, 3)
def test_snapshots(self, mocked_s3): s3_client = Mock() s3_client.generate_presigned_url = Mock() s3_client.generate_presigned_url.return_value = 'http://s3.com/your-file.csv' mocked_s3.return_value = s3_client SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=True) SimpleQueryFactory(snapshot=False) snapshot_queries() self.assertEqual(s3_client.generate_presigned_url.call_count, 3)