mirror of
https://github.com/public-apis/public-apis
synced 2024-11-28 08:51:41 +01:00
63 lines
2.2 KiB
Python
63 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from email import message
|
|
import unittest
|
|
|
|
from validate.format import error_message
|
|
from validate.format import get_categories_content
|
|
|
|
|
|
class TestValidadeFormat(unittest.TestCase):
|
|
|
|
def test_error_message_return_and_return_type(self):
|
|
line_num_unity = 1
|
|
line_num_ten = 10
|
|
line_num_hundred = 100
|
|
line_num_thousand = 1000
|
|
|
|
msg = 'This is a unit test'
|
|
|
|
err_msg_unity = error_message(line_num_unity, msg)
|
|
err_msg_ten = error_message(line_num_ten, msg)
|
|
err_msg_hundred = error_message(line_num_hundred, msg)
|
|
err_msg_thousand = error_message(line_num_thousand, msg)
|
|
|
|
self.assertIsInstance(err_msg_unity, str)
|
|
self.assertIsInstance(err_msg_ten, str)
|
|
self.assertIsInstance(err_msg_hundred, str)
|
|
self.assertIsInstance(err_msg_thousand, str)
|
|
|
|
self.assertEqual(err_msg_unity, '(L002) This is a unit test')
|
|
self.assertEqual(err_msg_ten, '(L011) This is a unit test')
|
|
self.assertEqual(err_msg_hundred, '(L101) This is a unit test')
|
|
self.assertEqual(err_msg_thousand, '(L1001) This is a unit test')
|
|
|
|
def test_if_get_categories_content_return_correct_data_of_categories(self):
|
|
fake_contents = [
|
|
'### A',
|
|
'API | Description | Auth | HTTPS | CORS |',
|
|
'|---|---|---|---|---|',
|
|
'| [AA](https://www.ex.com) | Desc | `apiKey` | Yes | Yes |',
|
|
'| [AB](https://www.ex.com) | Desc | `apiKey` | Yes | Yes |',
|
|
'',
|
|
'### B',
|
|
'API | Description | Auth | HTTPS | CORS |',
|
|
'|---|---|---|---|---|',
|
|
'| [BA](https://www.ex.com) | Desc | `apiKey` | Yes | Yes |',
|
|
'| [BB](https://www.ex.com) | Desc | `apiKey` | Yes | Yes |'
|
|
]
|
|
|
|
result = get_categories_content(fake_contents)
|
|
self.assertIsInstance(result, tuple)
|
|
|
|
categories, category_line_num = result
|
|
self.assertIsInstance(categories, dict)
|
|
self.assertIsInstance(category_line_num, dict)
|
|
|
|
expected_result = ({'A': ['AA', 'AB'], 'B': ['BA', 'BB']}, {'A': 0, 'B': 6})
|
|
|
|
for res, ex_res in zip(result, expected_result):
|
|
|
|
with self.subTest():
|
|
self.assertEqual(res, ex_res)
|