sangge/waka-readme
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add test_decode_readme, add test_generate_new_readme (#50)

Browse Source 
* add test_generate_new_readme

* add test_decode_readme
This commit is contained in:
Siddharth Chandra 2020-11-27 13:21:49 +05:30 committed by GitHub
parent 25201232d8
commit 4ae860d6e3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
tests/test_main.py
View File

@ -2,12 +2,14 @@
Tests for the main.py
'''
import unittest
import datetime
import base64
import os
try:
# For travis build which uses
# python -m unittest discover
from main import make_graph
from main import make_graph, generate_new_readme, decode_readme
except Exception as e:
print("Error: missing 'main.py'\nTrying ablsolute import...")
@ -62,6 +64,31 @@ class TestMain(unittest.TestCase):
for j, graph in enumerate(graphs):
test(percents[j], blocks[i], graph)
def test_generate_new_readme(self):
'''Tests generate_new_readme method from main.py'''
dummy_readme = '''My Readme Start
<!--START_SECTION:waka-->
<!--END_SECTION:waka-->
My Readme End'''
dummy_stats = '''```text
Python 24 hrs 15 mins █████████████████████████ 100.00 %
```'''
expected_generated_readme = '''My Readme Start
<!--START_SECTION:waka-->\n```text
Python 24 hrs 15 mins █████████████████████████ 100.00 %
```\n<!--END_SECTION:waka-->
My Readme End
'''
expected_generated_readme = expected_generated_readme.strip()
actual_generated_readme = generate_new_readme(dummy_stats, dummy_readme)
self.assertEqual(actual_generated_readme, expected_generated_readme)
def test_decode_readme(self):
'''Tests decode_readme method from main.py'''
dummy_data = base64.b64encode(bytes('Some Data From GitHub', 'utf-8'))
expected_result = 'Some Data From GitHub'
actual_result = decode_readme(dummy_data)
self.assertEqual(actual_result, expected_result)
if __name__ == '__main__':
if __package__ is None:
@ -70,9 +97,9 @@ if __name__ == '__main__':
# python test/test_main.py
sys.path.append(os.path.dirname(
os.path.dirname(os.path.abspath(__file__))))
from main import make_graph
from main import make_graph, generate_new_readme, decode_readme
else:
# Later on if WakaReadme is implemetaion as package
# python -m tests/test_main
from ..main import make_graph
from ..main import make_graph, generate_new_readme, decode_readme
unittest.main()