From 41fd77c3664c944095ccc6f3ab831bd1c1033f04 Mon Sep 17 00:00:00 2001
From: RanKKI <youloveonlymeh@gmail.com>
Date: Sun, 26 Jun 2022 18:29:48 +0800
Subject: [PATCH] test: raw parser

---
 auto_bangumi/test.sh                  |  3 +++
 auto_bangumi/tests/test_raw_parser.py | 21 +++++++++++++++++++++
 2 files changed, 24 insertions(+)
 create mode 100755 auto_bangumi/test.sh
 create mode 100644 auto_bangumi/tests/test_raw_parser.py

diff --git a/auto_bangumi/test.sh b/auto_bangumi/test.sh
new file mode 100755
index 00000000..e12d8809
--- /dev/null
+++ b/auto_bangumi/test.sh
@@ -0,0 +1,3 @@
+#!/bin/bash
+
+exec python -m unittest discover tests
\ No newline at end of file
diff --git a/auto_bangumi/tests/test_raw_parser.py b/auto_bangumi/tests/test_raw_parser.py
new file mode 100644
index 00000000..320dcf6c
--- /dev/null
+++ b/auto_bangumi/tests/test_raw_parser.py
@@ -0,0 +1,21 @@
+import unittest
+
+from parser.analyser import RawParser
+
+
+class TestRawParser(unittest.TestCase):
+
+    def test_raw_parser(self):
+        parser = RawParser()
+        content = "【幻樱字幕组】【4月新番】【古见同学有交流障碍症 第二季 Komi-san wa, Komyushou Desu. S02】【22】【GB_MP4】【1920X1080】"
+        info = parser.analyse(content)
+
+        self.assertEqual(info.title, "Komi-san wa, Komyushou Desu.")
+        self.assertEqual(info.dpi, "1920X1080")
+        self.assertEqual(info.ep_info.number, 22)
+        self.assertEqual(info.season_info.number, 2)
+
+    def test_pre_process(self):
+        content = "【幻樱字幕组】【4月新番】"
+        expected_content = "[幻樱字幕组][4月新番]"
+        self.assertEqual(RawParser.pre_process(content), expected_content)