#!/usr/bin/env python3

"""
test.modtest_d.subseed: subseed unit tests for the MMGen suite
"""

from mmgen.util import msg
from mmgen.seed import Seed
from mmgen.subseed import SubSeedList, SubSeedIdxRange

from ..include.common import cfg, vmsg

nSubseeds = SubSeedList.dfl_len

class unit_tests:

	def ops(self, name, ut, desc='basic ops...'):
		for a, b, c, d, e, f, h in (
				(8, '4710FBF0', '0C1B0615', '803B165C', '2669AC64', 256, '10L'),
				(6, '9D07ABBD', 'EBA9C33F', '20787E6A', '192E2AA2', 192, '10L'),
				(4, '43670520', '04A4CCB3', 'B5F21D7B', 'C1934CFF', 128, '10L'),
			):

			seed_bin = bytes.fromhex('deadbeef' * a)
			seed = Seed(cfg, seed_bin=seed_bin)
			assert seed.sid == b, seed.sid

			subseed = seed.subseed('2s')
			assert subseed.sid == c, subseed.sid

			subseed = seed.subseed('3')
			assert subseed.sid == d, subseed.sid

			subseed = seed.subseed_by_seed_id(e)
			assert subseed.bitlen == f, subseed.bitlen
			assert subseed.sid == e, subseed.sid
			assert subseed.idx == 10, subseed.idx
			assert subseed.ss_idx == h, subseed.ss_idx

			seed2 = Seed(cfg, seed_bin=seed_bin)
			ss2_list = seed2.subseeds

			seed2.subseeds._generate(1)
			assert len(ss2_list) == 1, len(ss2_list)

			seed2.subseeds._generate(1) # do nothing
			seed2.subseeds._generate(2) # append one item

			seed2.subseeds._generate(5)
			assert len(ss2_list) == 5, len(ss2_list)

			seed2.subseeds._generate(3) # do nothing
			assert len(ss2_list) == 5, len(ss2_list)

			seed2.subseeds._generate(10)
			assert len(ss2_list) == 10, len(ss2_list)

			assert seed.pfmt() == seed2.pfmt()
			assert seed.subseeds.pfmt() == seed2.subseeds.pfmt()

			s = seed.subseeds.format(1, nSubseeds)
			s_lines = s.strip().split('\n')
			assert len(s_lines) == nSubseeds + 4, s

			a = seed.subseed('2L').sid
			b = [e for e in s_lines if ' 2L:' in e][0].strip().split()[1]
			assert a == b, b

			c = seed.subseed('2').sid
			assert c == a, c

			a = seed.subseed('5S').sid
			b = [e for e in s_lines if ' 5S:' in e][0].strip().split()[3]
			assert a == b, b

			s = seed.subseeds.format(nSubseeds+1, nSubseeds+2)
			s_lines = s.strip().split('\n')
			assert len(s_lines) == 6, s

			ss_idx = str(nSubseeds+2) + 'S'
			a = seed.subseed(ss_idx).sid
			b = [e for e in s_lines if f' {ss_idx}:' in e][0].strip().split()[3]
			assert a == b, b

			s = seed.subseeds.format(1, 10)
			s_lines = s.strip().split('\n')
			assert len(s_lines) == 14, s

			vmsg(f'{s}')

		return True

	def limits(self, name, ut, desc='limits and ranges'):

		seed_bin = bytes.fromhex('deadbeef' * 8)

		vmsg('Testing limits')

		seed = Seed(cfg, seed_bin=seed_bin, nSubseeds=11)
		seed.subseeds._generate()
		ss = seed.subseeds
		assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
		assert len(ss) == 11, len(ss)

		seed = Seed(cfg, seed_bin=seed_bin)
		seed.subseeds._generate()
		ss = seed.subseeds
		assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
		assert len(ss) == nSubseeds, len(ss)

		seed = Seed(cfg, seed_bin=seed_bin)
		seed.subseed_by_seed_id('EEEEEEEE')
		ss = seed.subseeds
		assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
		assert len(ss) == nSubseeds, len(ss)

		seed = Seed(cfg, seed_bin=seed_bin)
		subseed = seed.subseed_by_seed_id('803B165C')
		assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
		assert subseed.sid == '803B165C', subseed.sid
		assert subseed.idx == 3, subseed.idx

		seed = Seed(cfg, seed_bin=seed_bin)
		subseed = seed.subseed_by_seed_id('803B165C', last_idx=1)
		assert len(ss.data['long']) == len(ss.data['short']), len(ss.data['short'])
		assert subseed is None, subseed

		vmsg('Testing SubSeedIdxRange()')

		r = SubSeedIdxRange('1-5')
		r2 = SubSeedIdxRange(1, 5)
		assert r2 == r, r2
		assert r == (r.first, r.last), r
		assert r.first == 1, r.first
		assert r.last == 5, r.last
		assert r.items == [1, 2, 3, 4, 5], r.items
		assert list(r.iterate()) == r.items, list(r.iterate())

		r = SubSeedIdxRange('22')
		r2 = SubSeedIdxRange(22, 22)
		assert r2 == r, r2
		assert r == (r.first, r.last), r
		assert r.first == 22, r.first
		assert r.last == 22, r.last
		assert r.items == [22], r
		assert list(r.iterate()) == r.items, list(r.iterate())

		r = SubSeedIdxRange('3-3')
		assert r.items == [3], r.items

		r = SubSeedIdxRange(f'{nSubseeds-1}-{nSubseeds}')
		assert r.items == [nSubseeds-1, nSubseeds], r.items

		for n, e in enumerate(SubSeedIdxRange('1-5').iterate(), 1):
			assert n == e, e

		assert n == 5, n

		return True

	def collisions(self, name, ut, desc='Seed ID collisions'):

		ss_count, ltr, last_sid, collisions_chk, mode_desc = (
			(SubSeedIdxRange.max_idx, 'S', '2788F26B', 470, 'normal'),
			(49509, 'L', '8D1FE500', 2, 'fast')
		)[bool(cfg.fast)]

		last_idx = str(ss_count) + ltr

		vmsg(f'Using {ss_count} subseed pairs ({mode_desc} mode)')

		seed_bin = bytes.fromhex('12abcdef' * 8) # 95B3D78D
		seed = Seed(cfg, seed_bin=seed_bin)

		seed.subseeds._generate(ss_count)
		ss = seed.subseeds

		assert seed.subseed(last_idx).sid == last_sid, seed.subseed(last_idx).sid

		for sid in ss.data['long']:
			# msg(sid)
			assert sid not in ss.data['short']

		collisions = 0
		for k in ('short', 'long'):
			for sid in ss.data[k]:
				collisions += ss.data[k][sid][1]

		assert collisions == collisions_chk, collisions
		vmsg(f'{collisions} collisions, last_sid {last_sid}')

		return True
