Guest User

Untitled

a guest
Apr 26th, 2018
73
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.30 KB | None | 0 0
  1. def test_upload_ok(self):
  2. file_names = list(validator.TABLE_COLUMN.keys())
  3. file_names.sort()
  4. files = {}
  5.  
  6. mock_data = {
  7. 'branch': [1, 'Tebet', 'Jl. Tebet Raya', 'Rafiano Ruby',
  8. '08131651104', [1], [1]],
  9. 'penyuluh': [1, 'Wicaksono', 'Wisnu', '08111710107'],
  10. 'territory': [1, 'DKI Jakarta', 'Jakarta Timur',
  11. 'Duren Sawit', 'Malaka Jaya'],
  12. 'ketua_arisan': [1, 'Bthari', 'Smart', 'Jl. Bunga Rampai', 1,
  13. "-6.198495", "106.837306", 3, "08/03/2018",
  14. False, False]
  15. }
  16.  
  17. for filename in file_names:
  18. file = StringIO()
  19. writer = csv.writer(file)
  20. writer.writerow(validator.TABLE_COLUMN[filename])
  21. writer.writerow(mock_data[filename])
  22. file.seek(0)
  23. file_b = BytesIO()
  24. file_b.write(file.read().encode())
  25. file_b.seek(0)
  26. files[filename] = (file_b, filename + ".csv")
  27.  
  28. with patch('resources.api.parse'):
  29. resp = self.app.post(
  30. '/api/v1/upload',
  31. data=files,
  32. content_type="multipart/form-data"
  33. )
  34.  
  35. self.assertEqual(
  36. resp.status_code, 200
  37. )
Add Comment
Please, Sign In to add comment