def test_merge_note(cls):
cmd = Command(
'mssqlclient.py #{DOMAIN}/#{USER}@#{RHOST} -windows-auth')
cmd.add_note('desc: this is description')
cmd.add_note('refer: this is refer')
cmd.add_note('refer: this is refer 2')
cmd.add_note('link: this is link')
cmd.add_note('link: this is link 2')
cmd.add_note('this is note')
cmd.add_note('this is note 2')
notes = [
'desc: this is merge description', 'refer: this is merge refer',
'link: this is merge link', 'this is merge note'
]
cmd.merge_notes(notes)
assert len(cmd.notes) == 3
assert len(cmd.links) == 3
assert len(cmd.refer) == 3
assert cmd.desc == 'this is merge description'
assert cmd.notes[0] == 'this is merge note'
# 只有 note 会翻转
assert cmd.links[2] == 'this is merge link'
assert cmd.refer[2] == 'this is merge refer'
def test_add_note(cls):
cmd = Command(
'mssqlclient.py #{DOMAIN}/#{USER}@#{RHOST} -windows-auth')
cmd.add_note('desc: this is description')
cmd.add_note('refer: this is refer')
cmd.add_note('refer: this is refer 2')
cmd.add_note('link: this is link')
cmd.add_note('link: this is link 2')
cmd.add_note('this is note')
cmd.add_note('this is note 2')
assert len(cmd.notes) == 2
assert len(cmd.refer) == 2
assert len(cmd.links) == 2
assert cmd.desc == 'this is description'
assert cmd.refer[0] == 'this is refer'
assert cmd.links[0] == 'this is link'
assert cmd.notes[0] == 'this is note'