def _check(node_data, signal_d, hash_value, data): """Check the upload was ok.""" mdid, node_id = node_data # download it buf = NoCloseStringIO() self.patch(self.main.fs, 'get_partial_for_writing', lambda s, n: buf) yield self.aq.download(request.ROOT, node_id, hash_value, mdid) yield self.wait_for_nirvana() self.assertEqual(buf.getvalue(), data) # it's ok, we're happy signal_d.callback(None)
def test_upload(self): """Test we can upload stuff.""" buf = NoCloseStringIO() self.patch(self.main.fs, 'get_partial_for_writing', lambda s, n: buf) hash_value, _, data, d = self._mk_file_w_content() mdid, node_id = yield d yield self.aq.download(request.ROOT, node_id, hash_value, mdid) yield self.wait_for_nirvana() self.assertEqual(buf.getvalue(), data) self.assertEvent(('AQ_UPLOAD_FINISHED', {'share_id': request.ROOT, 'node_id': node_id, 'hash': hash_value, 'new_generation': 2L}))
def test_upload(self): """Test we can upload stuff.""" buf = NoCloseStringIO() self.patch(self.main.fs, 'get_partial_for_writing', lambda s, n: buf) hash_value, _, data, d = self._mk_file_w_content() mdid, node_id = yield d yield self.aq.download(request.ROOT, node_id, hash_value, mdid) yield self.wait_for_nirvana() self.assertEqual(buf.getvalue(), data) self.assertEvent(('AQ_UPLOAD_FINISHED', { 'share_id': request.ROOT, 'node_id': node_id, 'hash': hash_value, 'new_generation': 2L }))