From 846d530acc0212dda2ab707544e692e28a6ae6fe Mon Sep 17 00:00:00 2001
From: Nicola Coretti <nico.coretti@gmail.com>
Date: Sat, 9 Oct 2021 15:06:05 +0200
Subject: [PATCH] Add tests for table and checksum cli command(s)

---
 cli_tests.py | 69 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 69 insertions(+)

diff --git a/cli_tests.py b/cli_tests.py
index 33759b3..7563ebf 100644
--- a/cli_tests.py
+++ b/cli_tests.py
@@ -2,6 +2,8 @@
 #
 # Copyright (c) 2021, Nicola Coretti
 # All rights reserved.
+import io
+import inspect
 import unittest
 from unittest.mock import patch, call
 from crc import main
@@ -25,6 +27,73 @@ def test_table_subcommand_with_no_additional_arguments(self):
             self.assertTrue(exit_mock.called)
             self.assertEqual(exit_mock.call_args, (call(expected_exit_code)))
 
+    def test_table_generation_for_width_8_and_poly_0x1D(self):
+        expected_output = io.StringIO(inspect.cleandoc("""
+        0x00 0x1D 0x3A 0x27 0x74 0x69 0x4E 0x53 
+        0xE8 0xF5 0xD2 0xCF 0x9C 0x81 0xA6 0xBB 
+        0xCD 0xD0 0xF7 0xEA 0xB9 0xA4 0x83 0x9E 
+        0x25 0x38 0x1F 0x20 0x51 0x4C 0x6B 0x76 
+        0x87 0x9A 0xBD 0xA0 0xF3 0xEE 0xC9 0xD4 
+        0x6F 0x72 0x55 0x48 0x1B 0x60 0x21 0x3C 
+        0x4A 0x57 0x70 0x6D 0x3E 0x23 0x40 0x19 
+        0xA2 0xBF 0x98 0x85 0xD6 0xCB 0xEC 0xF1 
+        0x13 0xE0 0x29 0x34 0x67 0x7A 0x5D 0x40 
+        0xFB 0xE6 0xC1 0xDC 0x8F 0x92 0xB5 0xA8 
+        0xDE 0xC3 0xE4 0xF9 0xAA 0xB7 0x90 0x8D 
+        0x36 0x2B 0xC0 0x11 0x42 0x5F 0x78 0x65 
+        0x94 0x89 0xAE 0xB3 0xE0 0xFD 0xDA 0xC7 
+        0x7C 0x61 0x46 0x5B 0x80 0x15 0x32 0x2F 
+        0x59 0x44 0x63 0x7E 0x2D 0x30 0x17 0xA0 
+        0xB1 0xAC 0x8B 0x96 0xC5 0xD8 0xFF 0xE2 
+        0x26 0x3B 0x1C 0x10 0x52 0x4F 0x68 0x75 
+        0xCE 0xD3 0xF4 0xE9 0xBA 0xA7 0x80 0x9D 
+        0xEB 0xF6 0xD1 0xCC 0x9F 0x82 0xA5 0xB8 
+        0x30 0x1E 0x39 0x24 0x77 0x6A 0x4D 0x50 
+        0xA1 0xBC 0x9B 0x86 0xD5 0xC8 0xEF 0xF2 
+        0x49 0x54 0x73 0x6E 0x3D 0x20 0x70 0x1A 
+        0x6C 0x71 0x56 0x4B 0x18 0x50 0x22 0x3F 
+        0x84 0x99 0xBE 0xA3 0xF0 0xED 0xCA 0xD7 
+        0x35 0x28 0xF0 0x12 0x41 0x5C 0x7B 0x66 
+        0xDD 0xC0 0xE7 0xFA 0xA9 0xB4 0x93 0x8E 
+        0xF8 0xE5 0xC2 0xDF 0x8C 0x91 0xB6 0xAB 
+        0x10 0xD0 0x2A 0x37 0x64 0x79 0x5E 0x43 
+        0xB2 0xAF 0x88 0x95 0xC6 0xDB 0xFC 0xE1 
+        0x5A 0x47 0x60 0x7D 0x2E 0x33 0x14 0x90 
+        0x7F 0x62 0x45 0x58 0xB0 0x16 0x31 0x2C 
+        0x97 0x8A 0xAD 0xB0 0xE3 0xFE 0xD9 0xC4 
+        """) + "\n")
+        expected_exit_code = 0
+        argv = ['table', '8', '0x1D']
+        with patch('sys.exit') as exit, patch('sys.stdout', new_callable=io.StringIO) as stdout:
+            main(argv)
+
+            stdout.seek(0)
+            actual = stdout.read().split('\n')
+            expected = expected_output.read().split('\n')
+
+            self.assertEqual(expected, actual)
+            self.assertEqual(exit.call_args, (call(expected_exit_code)))
+
+    def test_checksum_command(self):
+        expected_output = io.StringIO(inspect.cleandoc("""
+        AUTOSAR: 0x43
+        BLUETOOTH: 0xB3
+        CCITT: 0x9E
+        SAEJ1850: 0x5B
+        """) + "\n")
+        expected_exit_code = 0
+        argv = ['checksum']
+        with patch('sys.exit') as exit, patch('sys.stdout', new_callable=io.StringIO) as stdout:
+            with patch('sys.stdin', io.StringIO("Some Data To Checksum")) as stdin:
+                main(argv)
+
+                stdout.seek(0)
+                actual = stdout.read().split('\n')
+                expected = expected_output.read().split('\n')
+
+                self.assertEqual(expected, actual)
+                self.assertEqual(exit.call_args, (call(expected_exit_code)))
+
 
 if __name__ == '__main__':
     unittest.main()