From 1847a70a47b12ff5c7e7db9361f69d6d1e79a919 Mon Sep 17 00:00:00 2001
From: Maykon Pacheco <38124066+maykonpacheco@users.noreply.github.com>
Date: Sat, 11 Jun 2022 14:08:24 -0400
Subject: [PATCH] test for the strip_bz2_extension method (#24826)

---
 selfdrive/athena/tests/test_athenad.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/selfdrive/athena/tests/test_athenad.py b/selfdrive/athena/tests/test_athenad.py
index b6457ca01d..a0b308c216 100755
--- 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')