athena: Do not show canceled upload items in listUploadQueue (#22627)

* do not show canceled upload items in listUploadQueue

* return item._asdict()

* athena: updated test for listUploadQueue
pull/22640/head
Devin Leamy 4 years ago committed by GitHub
parent b983f8aa9b
commit 12165c8865
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 2
      selfdrive/athena/athenad.py
  2. 4
      selfdrive/athena/tests/test_athenad.py

@ -253,7 +253,7 @@ def uploadFileToUrl(fn, url, headers):
@dispatcher.add_method @dispatcher.add_method
def listUploadQueue(): def listUploadQueue():
items = list(upload_queue.queue) + list(cur_upload_items.values()) items = list(upload_queue.queue) + list(cur_upload_items.values())
return [i._asdict() for i in items if i is not None] return [i._asdict() for i in items if i.id not in cancelled_uploads and i is not None]
@dispatcher.add_method @dispatcher.add_method

@ -245,6 +245,10 @@ class TestAthenadMethods(unittest.TestCase):
self.assertDictEqual(items[0], item._asdict()) self.assertDictEqual(items[0], item._asdict())
self.assertFalse(items[0]['current']) self.assertFalse(items[0]['current'])
athenad.cancelled_uploads.add(item.id)
items = dispatcher["listUploadQueue"]()
self.assertEqual(len(items), 0)
@mock.patch('selfdrive.athena.athenad.create_connection') @mock.patch('selfdrive.athena.athenad.create_connection')
def test_startLocalProxy(self, mock_create_connection): def test_startLocalProxy(self, mock_create_connection):
end_event = threading.Event() end_event = threading.Event()

Loading…
Cancel
Save