@@ -18,6 +18,12 @@ class _FakeExportResponse:
|
||||
'Content-Type': 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
|
||||
}
|
||||
|
||||
def iter_content(self, chunk_size=None):
|
||||
yield self.content
|
||||
|
||||
def close(self):
|
||||
return None
|
||||
|
||||
|
||||
class AdminDedupeTotalDataTest(unittest.TestCase):
|
||||
def setUp(self):
|
||||
@@ -138,12 +144,14 @@ class AdminDedupeTotalDataTest(unittest.TestCase):
|
||||
response = admin_api.export_dedupe_total_data.__wrapped__()
|
||||
|
||||
self.assertEqual(response.status_code, 200)
|
||||
self.assertEqual(response.get_data(), b'xlsx')
|
||||
self.assertEqual(session.kwargs['params'], {
|
||||
'operatorId': 7,
|
||||
'username': 'operator',
|
||||
'groupId': 3,
|
||||
})
|
||||
self.assertEqual(session.kwargs['headers'], {'X-Internal-Token': 'token'})
|
||||
self.assertTrue(session.kwargs['stream'])
|
||||
|
||||
def test_import_requires_group(self):
|
||||
with self.app.test_request_context(
|
||||
|
||||
Reference in New Issue
Block a user