# produce mozlog-compatible log messages, following the spec at https://mozbase.readthedocs.io/en/latest/mozlog.html

import json
import os

from time import time

class TestLogger(object):
    def __init__(self, source, threadname='main'):
        self.template = {
            'source': source,
            'thread': threadname,
            'pid': os.getpid(),
        }

    def _record(self, **kwargs):
        record = self.template.copy()
        record.update(**kwargs)
        if 'time' not in record:
            record['time'] = time()
        return record

    def _log_obj(self, obj):
        print(json.dumps(obj, sort_keys=True))

    def _log(self, **kwargs):
        self._log_obj(self._record(**kwargs))

    def suite_start(self):
        self._log(action='suite_start', tests=[])

    def suite_end(self):
        self._log(action='suite_end')

    def test_start(self, testname):
        self._log(action='test_start', test=testname)

    def test_end(self, testname, status):
        self._log(action='test_end', test=testname, status=status)

    def test(self, testname, status, duration, **details):
        record = self._record(action='test_start', test=testname)
        end_time = record['time']
        record['time'] -= duration
        self._log_obj(record)

        record['action'] = 'test_end'
        record['time'] = end_time
        record['status'] = status
        record.update(**details)
        self._log_obj(record)