test for the strip_bz2_extension method (#24826)

3 years ago · 1847a70a47
parent e375087720
commit 1847a70a47
1 changed files with 7 additions and 0 deletions
--- a/selfdrive/athena/tests/test_athenad.py
+++ b/selfdrive/athena/tests/test_athenad.py
@ -120,6 +120,13 @@ class TestAthenadMethods(unittest.TestCase):
    self.assertTrue(resp, 'list empty!')
    self.assertCountEqual(resp, expected)

+  def test_strip_bz2_extension(self):
+    fn = os.path.join(athenad.ROOT, 'qlog.bz2')
+    Path(fn).touch()
+    if fn.endswith('.bz2'):
+      self.assertEqual(athenad.strip_bz2_extension(fn), fn[:-4]) 
+
+
  @with_http_server
  def test_do_upload(self, host):
    fn = os.path.join(athenad.ROOT, 'qlog.bz2')