// Copyright (c) 2022, 2024 blackshirt. All rights reserved.
// Use of this source code is governed by a MIT License
// that can be found in the LICENSE file.
module asn1

import os
import time

fn test_serialize_utctime_basic() ! {
	inp := '191215190210Z'

	exp := [u8(0x17), 0x0D, 49, 57, 49, 50, 49, 53, 49, 57, 48, 50, 49, 48, 90]

	ut := UtcTime.new(inp)!
	out := encode(ut)!

	assert out == exp

	// back
	back, pos := UtcTime.decode(out)!
	assert back.tag().tag_number() == int(TagType.utctime)
	assert back == ut
	assert back.value == inp
}

fn test_create_utctime_from_std_time() ! {
	now := time.new(year: 2024, month: 11, day: 13, hour: 17, minute: 45, second: 50)

	utb := UtcTime.from_time(now)!
	utc := UtcTime.from_time(now)!

	assert utb.value == utc.value

	tt := utc.into_utctime()!
	assert now == tt

	inp := '191215190210Z'
	t_inp := UtcTime.new(inp)!
	t_inp_utc := t_inp.into_utctime()!

	// time to UtcTime back
	tinp_back := UtcTime.from_time(t_inp_utc)!
	assert tinp_back.value == inp
}

fn test_create_utctime_from_std_time_with_negative_offset() ! {
	tz := os.getenv('TZ')
	os.setenv('TZ', 'utc 1', true)

	defer {
		os.setenv('TZ', tz, true)
	}

	now := time.new(year: 2024, month: 11, day: 13, hour: 17, minute: 45, second: 50)
	UtcTime.from_time(now)!
}

fn test_serialize_utctime_error_without_z() ! {
	// this input does not contains zulu 'Z' part
	inp := '191215190210'

	exp := [u8(0x17), 0x0D, 49, 57, 49, 50, 49, 53, 49, 57, 48, 50, 49, 48]

	ut := UtcTime.new(inp) or {
		assert err == error('UtcTime: fail on validate utctime')
		return
	}
}

fn test_serialize_utctime_error_month() ! {
	// the month part is > 12
	inp := '191815190210Z'

	exp := [u8(0x17), 0x0D, 49, 57, 49, 56, 49, 53, 49, 57, 48, 50, 49, 48]

	ut := UtcTime.new(inp) or {
		assert err == error('UtcTime: fail on validate utctime')
		return
	}
}

fn test_serialize_utctime_error_day() ! {
	// the day part is > 30
	inp := '191235190210Z'

	exp := [u8(0x17), 0x0D, 0x31, 0x39, 0x31, 0x32, 0x31, 0x32, 0x31, 0x39, 0x30, 0x32, 0x31, 0x30,
		0x5A]

	ut := UtcTime.new(inp) or {
		assert err == error('UtcTime: fail on validate utctime')
		return
	}
}

fn test_serialize_decode_generalizedtime() ! {
	s := '20100102030405Z'

	exp := [u8(0x18), 0x0f, 50, 48, 49, 48, 48, 49, 48, 50, 48, 51, 48, 52, 48, 53, 90]

	gt := GeneralizedTime.new(s)!
	out := encode(gt)!
	assert out == exp

	// back
	back, pos := GeneralizedTime.decode(out)!

	assert back == gt
	assert back.value == s
	assert back.tag().tag_number() == int(TagType.generalizedtime)
}

fn test_create_generalizedtime_from_std_time() ! {
	now := time.new(year: 2024, month: 11, day: 13, hour: 17, minute: 45, second: 50)

	gtb := GeneralizedTime.from_time(now)!
	gtc := GeneralizedTime.from_time(now)!

	assert gtb.value == gtc.value
	assert gtb.value == '20241113174550Z'
	assert gtc.value == '20241113174550Z'

	tt := gtc.into_utctime()!
	assert now == tt

	s := '20100102030405Z'
	gt := GeneralizedTime.new(s)!

	g_utc := gt.into_utctime()!
	g_utc_back := GeneralizedTime.from_time(g_utc)!

	assert g_utc_back.value == s
}

fn test_create_generalizedtime_from_std_time_with_negative_offset() ! {
	tz := os.getenv('TZ')
	os.setenv('TZ', 'utc 1', true)

	defer {
		os.setenv('TZ', tz, true)
	}

	now := time.new(year: 2024, month: 11, day: 13, hour: 17, minute: 45, second: 50)
	GeneralizedTime.from_time(now)!
}