python/jc
1
0
Fork 0
mirror of https://github.com/kellyjonbrazil/jc.git synced 2025-06-19 00:17:51 +02:00
Code Issues Releases Activity

add_timestamp_to tests in cli.py

Browse Source
This commit is contained in:
Kelly Brazil
2022-08-04 17:49:18 -07:00
parent cb3484e1ab
commit 6d66557b11
1 changed files with 37 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
tests/test_cli.py
View File

@ -1,4 +1,5 @@
import unittest import unittest
from datetime import datetime, timezone
import pygments import pygments
from pygments.token import (Name, Number, String, Keyword) from pygments.token import (Name, Number, String, Keyword)
import jc.cli import jc.cli
@ -254,5 +255,41 @@ class MyTests(unittest.TestCase):
self.assertGreaterEqual(jc.cli.about_jc()['parser_count'], 55) self.assertGreaterEqual(jc.cli.about_jc()['parser_count'], 55)
self.assertEqual(jc.cli.about_jc()['parser_count'], len(jc.cli.about_jc()['parsers'])) self.assertEqual(jc.cli.about_jc()['parser_count'], len(jc.cli.about_jc()['parsers']))
def test_add_timestamp_to_simple_dict(self):
list_or_dict = {'a': 1, 'b': 2}
runtime = datetime(2022, 8, 5, 0, 37, 9, 273349, tzinfo=timezone.utc)
magic_exit_code = 0
expected = {'a': 1, 'b': 2, '_jc_meta': {'timestamp': 1659659829.273349}}
jc.cli.add_timestamp_to(list_or_dict, runtime, magic_exit_code)
self.assertEqual(list_or_dict, expected)
def test_add_timestamp_to_simple_list(self):
list_or_dict = [{'a': 1, 'b': 2},{'a': 3, 'b': 4}]
runtime = datetime(2022, 8, 5, 0, 37, 9, 273349, tzinfo=timezone.utc)
magic_exit_code = 0
expected = [{'a': 1, 'b': 2, '_jc_meta': {'timestamp': 1659659829.273349}}, {'a': 3, 'b': 4, '_jc_meta': {'timestamp': 1659659829.273349}}]
jc.cli.add_timestamp_to(list_or_dict, runtime, magic_exit_code)
self.assertEqual(list_or_dict, expected)
def test_add_timestamp_to_dict_existing_meta(self):
list_or_dict = {'a': 1, 'b': 2, '_jc_meta': {'foo': 'bar'}}
runtime = datetime(2022, 8, 5, 0, 37, 9, 273349, tzinfo=timezone.utc)
magic_exit_code = 0
expected = {'a': 1, 'b': 2, '_jc_meta': {'foo': 'bar', 'timestamp': 1659659829.273349}}
jc.cli.add_timestamp_to(list_or_dict, runtime, magic_exit_code)
self.assertEqual(list_or_dict, expected)
def test_add_timestamp_to_list_existing_meta(self):
list_or_dict = [{'a': 1, 'b': 2, '_jc_meta': {'foo': 'bar'}},{'a': 3, 'b': 4, '_jc_meta': {'foo': 'bar'}}]
runtime = datetime(2022, 8, 5, 0, 37, 9, 273349, tzinfo=timezone.utc)
magic_exit_code = 0
expected = [{'a': 1, 'b': 2, '_jc_meta': {'foo': 'bar', 'timestamp': 1659659829.273349}}, {'a': 3, 'b': 4, '_jc_meta': {'foo': 'bar', 'timestamp': 1659659829.273349}}]
jc.cli.add_timestamp_to(list_or_dict, runtime, magic_exit_code)
self.assertEqual(list_or_dict, expected)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()