Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def test_upload_ok(self):
- file_names = list(validator.TABLE_COLUMN.keys())
- file_names.sort()
- files = {}
- mock_data = {
- 'branch': [1, 'Tebet', 'Jl. Tebet Raya', 'Rafiano Ruby',
- '08131651104', [1], [1]],
- 'penyuluh': [1, 'Wicaksono', 'Wisnu', '08111710107'],
- 'territory': [1, 'DKI Jakarta', 'Jakarta Timur',
- 'Duren Sawit', 'Malaka Jaya'],
- 'ketua_arisan': [1, 'Bthari', 'Smart', 'Jl. Bunga Rampai', 1,
- "-6.198495", "106.837306", 3, "08/03/2018",
- False, False]
- }
- for filename in file_names:
- file = StringIO()
- writer = csv.writer(file)
- writer.writerow(validator.TABLE_COLUMN[filename])
- writer.writerow(mock_data[filename])
- file.seek(0)
- file_b = BytesIO()
- file_b.write(file.read().encode())
- file_b.seek(0)
- files[filename] = (file_b, filename + ".csv")
- with patch('resources.api.parse'):
- resp = self.app.post(
- '/api/v1/upload',
- data=files,
- content_type="multipart/form-data"
- )
- self.assertEqual(
- resp.status_code, 200
- )
Add Comment
Please, Sign In to add comment