from __future__ import unicode_literals
from tiny_test_fw import Utility
import datetime
import re
import ttfw_idf


@ttfw_idf.idf_example_test(env_tag='Example_WIFI')
def test_examples_sntp(env, extra_data):

    dut = env.get_dut('sntp', 'examples/protocols/sntp')
    dut.start_app()

    dut.expect_all('Time is not set yet. Connecting to WiFi and getting time over NTP.',
                   'Initializing SNTP',
                   'Waiting for system time to be set... (1/10)',
                   'Notification of a time synchronization event',
                   timeout=60)

    TIME_FORMAT = '%a %b %d %H:%M:%S %Y'
    TIME_FORMAT_REGEX = r'\w+ \w+ \d{1,2} \d{2}:\d{2}:\d{2} \d{4}'
    TIME_DIFF = datetime.timedelta(seconds=10 + 2)  # cpu spends 10 seconds in deep sleep
    NY_time = None
    SH_time = None

    def check_time(prev_NY_time, prev_SH_time):
        NY_str = dut.expect(re.compile(r'The current date/time in New York is: ({})'.format(TIME_FORMAT_REGEX)))[0]
        SH_str = dut.expect(re.compile(r'The current date/time in Shanghai is: ({})'.format(TIME_FORMAT_REGEX)))[0]
        Utility.console_log('New York: "{}"'.format(NY_str))
        Utility.console_log('Shanghai: "{}"'.format(SH_str))
        dut.expect('Entering deep sleep for 10 seconds')
        Utility.console_log('Sleeping...')
        new_NY_time = datetime.datetime.strptime(NY_str, TIME_FORMAT)
        new_SH_time = datetime.datetime.strptime(SH_str, TIME_FORMAT)

        # The initial time is not checked because datetime has problems with timezones
        assert prev_NY_time is None or new_NY_time - prev_NY_time < TIME_DIFF
        assert prev_SH_time is None or new_SH_time - prev_SH_time < TIME_DIFF

        return (new_NY_time, new_SH_time)

    NY_time, SH_time = check_time(NY_time, SH_time)
    for i in range(2, 4):
        dut.expect('example: Boot count: {}'.format(i), timeout=30)
        NY_time, SH_time = check_time(NY_time, SH_time)


if __name__ == '__main__':
    test_examples_sntp()