summary refs log blame commit diff stats
path: root/test/tc_ext.py
blob: b80942336313257679735e58411fb2c6c44fadba (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "ht
# Copyright (C) 2009, 2010  Roman Zimbelmann <romanz@lavabit.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

if __name__ == '__main__': from __init__ import init; init()
import unittest
from collections import deque

from ranger.ext.iter_tools import *

class TestCases(unittest.TestCase):
	def test_flatten(self):
		def f(x):
			return list(flatten(x))

		self.assertEqual(
			[1,2,3,4,5],
			f([1,2,3,4,5]))
		self.assertEqual(
			[1,2,3,4,5],
			f([1,[2,3],4,5]))
		self.assertEqual(
			[1,2,3,4,5],
			f([[1,[2,3]],4,5]))
		self.assertEqual(
			[],
			f([[[[]]]]))
		self.assertEqual(
			['a', 'b', 'fskldfjl'],
			f(['a', ('b', 'fskldfjl')]))
		self.assertEqual(
			['a', 'b', 'fskldfjl'],
			f(['a', deque(['b', 'fskldfjl'])]))
		self.assertEqual(
			set([3.5, 4.3, 5.2, 6.0]),
			set(f([6.0, set((3.5, 4.3)), (5.2, )])))

	def test_unique(self):
		def u(x):
			return list(unique(x))

		self.assertEqual(
			[1,2,3],
			u([1,2,3]))
		self.assertEqual(
			[1,2,3],
			u([1,2,3,2,1]))
		self.assertEqual(
			[1,2,3],
			u([1,2,3,1,2,3,2,2,3,1,2,3,1,2,3,2,3,2,1]))
		self.assertEqual(
			[1,[2,3]],
			u([1,[2,3],1,[2,3],[2,3],1,[2,3],1,[2,3],[2,3],1]))

	def test_unique_keeps_type(self):
		def u(x):
			return unique(x)

		self.assertEqual(
			[1,2,3],
			u([1,2,3,1]))
		self.assertEqual(
			(1,2,3),
			u((1,2,3,1)))
		self.assertEqual(
			set((1,2,3)),
			u(set((1,2,3,1))))
		self.assertEqual(
			deque((1,2,3)),
			u(deque((1,2,3,1))))

	def test_mount_path(self):
		# assuming ismount() is used

		def my_ismount(path):
			depth = path.count('/')
			if path.startswith('/media'):
				return depth == 0 or depth == 2
			return depth <= 1

		from ranger.ext import mount_path
		original_ismount = mount_path.ismount
		mount_path.ismount = my_ismount
		try:
			mp = mount_path.mount_path

			self.assertEqual('/home', mp('/home/hut/porn/bondage'))
			self.assertEqual('/', mp('/'))
			self.assertEqual('/media/sdb1', mp('/media/sdb1/foo/bar'))
			self.assertEqual('/media/sdc2', mp('/media/sdc2/a/b/c/d/e'))
		finally:
			mount_path.ismount = original_ismount

		# TODO: links are not tested but I don't see how its possible
		# without messing around with mounts.
		# self.assertEqual('/media/foo',
		#     mount_path('/media/bar/some_link_to_a_foo_subdirectory'))

	def test_openstruct(self):
		from ranger.ext.openstruct import OpenStruct
		from random import randint, choice
		from string import ascii_letters

		os = OpenStruct(a='a')
		self.assertEqual(os.a, 'a')
		self.assertRaises(AttributeError, getattr, os, 'b')

		dictionary = {'foo': 'bar', 'zoo': 'zar'}
		os = OpenStruct(dictionary)
		self.assertEqual(os.foo, 'bar')
		self.assertEqual(os.zoo, 'zar')
		self.assertRaises(AttributeError, getattr, os, 'sdklfj')

		for i in range(100):
			attr_name = ''.join(choice(ascii_letters) \
				for x in range(randint(3,9)))
			value = randint(100,999)
			if not attr_name in os:
				self.assertRaises(AttributeError, getattr, os, attr_name)
			setattr(os, attr_name, value)
			value2 = randint(100,999)
			setattr(os, attr_name, value2)
			self.assertEqual(value2, getattr(os, attr_name))

	def test_shell_escape(self):
		from ranger.ext.shell_escape import shell_escape, shell_quote
		self.assertEqual(r"'luigi'\''s pizza'", shell_quote("luigi's pizza"))
		self.assertEqual(r"luigi\'s\ pizza", shell_escape("luigi's pizza"))
		self.assertEqual(r"\$lol/foo\\xyz\|\>\<\]\[",
				shell_escape(r"$lol/foo\xyz|><]["))


if __name__ == '__main__':
	unittest.main()
n> <span id="L34" class="LineNr"> 34 </span> 8b/-&gt; *(ebp+0x10) 2/r32/edx <span id="L35" class="LineNr"> 35 </span> { <span id="L36" class="LineNr"> 36 </span> <span class="subxComment"># if (i &gt;= size) break</span> <span id="L37" class="LineNr"> 37 </span> 39/compare %ecx 2/r32/edx <span id="L38" class="LineNr"> 38 </span> 7d/jump-if-&gt;= <span class="Constant">break</span>/disp8 <span id="L39" class="LineNr"> 39 </span> <span class="subxComment"># *curr-dest = *curr-src</span> <span id="L40" class="LineNr"> 40 </span> 8a/byte-&gt; *esi 0/r32/AL <span id="L41" class="LineNr"> 41 </span> 88/byte&lt;- *edi 0/r32/AL <span id="L42" class="LineNr"> 42 </span> <span class="subxComment"># update</span> <span id="L43" class="LineNr"> 43 </span> 46/increment-esi <span id="L44" class="LineNr"> 44 </span> 47/increment-edi <span id="L45" class="LineNr"> 45 </span> 41/increment-ecx <span id="L46" class="LineNr"> 46 </span> eb/jump <span class="Constant">loop</span>/disp8 <span id="L47" class="LineNr"> 47 </span> } <span id="L48" class="LineNr"> 48 </span><span class="Constant">$copy-bytes:end</span>: <span id="L49" class="LineNr"> 49 </span> <span class="subxS1Comment"># . restore registers</span> <span id="L50" class="LineNr"> 50 </span> 5f/pop-to-edi <span id="L51" class="LineNr"> 51 </span> 5e/pop-to-esi <span id="L52" class="LineNr"> 52 </span> 5a/pop-to-edx <span id="L53" class="LineNr"> 53 </span> 59/pop-to-ecx <span id="L54" class="LineNr"> 54 </span> 58/pop-to-eax <span id="L55" class="LineNr"> 55 </span> <span class="subxS1Comment"># . epilogue</span> <span id="L56" class="LineNr"> 56 </span> 89/&lt;- %esp 5/r32/ebp <span id="L57" class="LineNr"> 57 </span> 5d/pop-to-ebp <span id="L58" class="LineNr"> 58 </span> c3/return <span id="L59" class="LineNr"> 59 </span> <span id="L60" class="LineNr"> 60 </span><span class="subxFunction">stream-to-array</span>: <span class="subxComment"># in: (addr stream _), out: (addr handle array _)</span> <span id="L61" class="LineNr"> 61 </span> <span class="subxS1Comment"># . prologue</span> <span id="L62" class="LineNr"> 62 </span> 55/push-ebp <span id="L63" class="LineNr"> 63 </span> 89/&lt;- %ebp 4/r32/esp <span id="L64" class="LineNr"> 64 </span> <span class="subxS1Comment"># . save registers</span> <span id="L65" class="LineNr"> 65 </span> 50/push-eax <span id="L66" class="LineNr"> 66 </span> 51/push-ecx <span id="L67" class="LineNr"> 67 </span> 52/push-edx <span id="L68" class="LineNr"> 68 </span> 56/push-esi <span id="L69" class="LineNr"> 69 </span> <span class="subxComment"># esi = s</span> <span id="L70" class="LineNr"> 70 </span> 8b/-&gt; *(ebp+8) 6/r32/esi <span id="L71" class="LineNr"> 71 </span> <span class="subxComment"># var len/ecx: int = s-&gt;write - s-&gt;read</span> <span id="L72" class="LineNr"> 72 </span> 8b/-&gt; *esi 1/r32/ecx <span id="L73" class="LineNr"> 73 </span> 2b/subtract *(esi+4) 1/r32/ecx <span id="L74" class="LineNr"> 74 </span> <span class="subxComment"># allocate</span> <span id="L75" class="LineNr"> 75 </span> (<a href='120allocate.subx.html#L576'>allocate-array</a> <span class="SpecialChar"><a href='120allocate.subx.html#L27'>Heap</a></span> %ecx *(ebp+0xc)) <span id="L76" class="LineNr"> 76 </span> <span class="subxComment"># var in/edx: (addr byte) = s-&gt;data + s-&gt;read</span> <span id="L77" class="LineNr"> 77 </span> 8b/-&gt; *(esi+4) 2/r32/edx <span id="L78" class="LineNr"> 78 </span> 8d/copy-address *(esi+edx+0xc) 2/r32/edx <span id="L79" class="LineNr"> 79 </span> <span class="subxComment"># var dest/eax: (addr byte) = data for out</span> <span id="L80" class="LineNr"> 80 </span> 8b/-&gt; *(ebp+0xc) 0/r32/eax <span id="L81" class="LineNr"> 81 </span> (<a href='120allocate.subx.html#L256'>lookup</a> *eax *(eax+4)) <span class="subxComment"># =&gt; eax</span> <span id="L82" class="LineNr"> 82 </span> 8d/copy-address *(eax+4) 0/r32/eax <span id="L83" class="LineNr"> 83 </span> <span class="subxComment">#</span> <span id="L84" class="LineNr"> 84 </span> (<a href='310copy-bytes.subx.html#L6'>copy-bytes</a> %edx %eax %ecx) <span id="L85" class="LineNr"> 85 </span><span class="Constant">$stream-to-array:end</span>: <span id="L86" class="LineNr"> 86 </span> <span class="subxS1Comment"># . restore registers</span> <span id="L87" class="LineNr"> 87 </span> 5e/pop-to-esi <span id="L88" class="LineNr"> 88 </span> 5a/pop-to-edx <span id="L89" class="LineNr"> 89 </span> 59/pop-to-ecx <span id="L90" class="LineNr"> 90 </span> 58/pop-to-eax <span id="L91" class="LineNr"> 91 </span> <span class="subxS1Comment"># . epilogue</span> <span id="L92" class="LineNr"> 92 </span> 89/&lt;- %esp 5/r32/ebp <span id="L93" class="LineNr"> 93 </span> 5d/pop-to-ebp <span id="L94" class="LineNr"> 94 </span> c3/return <span id="L95" class="LineNr"> 95 </span> <span id="L96" class="LineNr"> 96 </span><span class="subxTest">test-stream-to-array</span>: <span id="L97" class="LineNr"> 97 </span> <span class="subxS1Comment"># . prologue</span> <span id="L98" class="LineNr"> 98 </span> 55/push-ebp <span id="L99" class="LineNr"> 99 </span> 89/&lt;- %ebp 4/r32/esp <span id="L100" class="LineNr">100 </span> <span class="subxComment"># setup</span> <span id="L101" class="LineNr">101 </span> (<a href='106stream.subx.html#L17'>clear-stream</a> <a href='112read-byte.subx.html#L331'>_test-input-stream</a>) <span id="L102" class="LineNr">102 </span> (<a href='108write.subx.html#L24'>write</a> <a href='112read-byte.subx.html#L331'>_test-input-stream</a> <span class="Constant">&quot;abc&quot;</span>) <span id="L103" class="LineNr">103 </span> <span class="subxComment"># skip something</span> <span id="L104" class="LineNr">104 </span> (<a href='112read-byte.subx.html#L273'>read-byte</a> <a href='112read-byte.subx.html#L331'>_test-input-stream</a>) <span class="subxComment"># =&gt; eax</span> <span id="L105" class="LineNr">105 </span> <span class="subxComment"># var out/ecx: (handle array byte)</span> <span id="L106" class="LineNr">106 </span> 68/push 0/imm32 <span id="L107" class="LineNr">107 </span> 68/push 0/imm32 <span id="L108" class="LineNr">108 </span> 89/&lt;- %ecx 4/r32/esp <span id="L109" class="LineNr">109 </span> <span class="subxComment">#</span> <span id="L110" class="LineNr">110 </span> (<a href='310copy-bytes.subx.html#L60'>stream-to-array</a> <a href='112read-byte.subx.html#L331'>_test-input-stream</a> %ecx) <span id="L111" class="LineNr">111 </span> (<a href='120allocate.subx.html#L256'>lookup</a> *ecx *(ecx+4)) <span class="subxComment"># =&gt; eax</span> <span id="L112" class="LineNr">112 </span> (<a href='105string-equal.subx.html#L220'>check-strings-equal</a> %eax <span class="Constant">&quot;bc&quot;</span>) <span id="L113" class="LineNr">113 </span> <span class="subxS1Comment"># . epilogue</span> <span id="L114" class="LineNr">114 </span> 89/&lt;- %esp 5/r32/ebp <span id="L115" class="LineNr">115 </span> 5d/pop-to-ebp <span id="L116" class="LineNr">116 </span> c3/return <span id="L117" class="LineNr">117 </span> <span id="L118" class="LineNr">118 </span><span class="subxComment"># like stream-to-array but ignore surrounding quotes</span> <span id="L119" class="LineNr">119 </span><span class="subxComment"># we might do other stuff here later</span> <span id="L120" class="LineNr">120 </span><span class="subxFunction">unquote-stream-to-array</span>: <span class="subxComment"># in: (addr stream _), out: (addr handle array _)</span> <span id="L121" class="LineNr">121 </span> <span class="subxS1Comment"># . prologue</span> <span id="L122" class="LineNr">122 </span> 55/push-ebp <span id="L123" class="LineNr">123 </span> 89/&lt;- %ebp 4/r32/esp <span id="L124" class="LineNr">124 </span> <span class="subxS1Comment"># . save registers</span> <span id="L125" class="LineNr">125 </span> 50/push-eax <span id="L126" class="LineNr">126 </span> 51/push-ecx <span id="L127" class="LineNr">127 </span> 52/push-edx <span id="L128" class="LineNr">128 </span> 56/push-esi <span id="L129" class="LineNr">129 </span> <span class="subxComment"># esi = s</span> <span id="L130" class="LineNr">130 </span> 8b/-&gt; *(ebp+8) 6/r32/esi <span id="L131" class="LineNr">131 </span> <span class="subxComment"># var len/ecx: int = s-&gt;write - s-&gt;read - 2</span> <span id="L132" class="LineNr">132 </span> 8b/-&gt; *esi 1/r32/ecx <span id="L133" class="LineNr">133 </span> 2b/subtract *(esi+4) 1/r32/ecx <span id="L134" class="LineNr">134 </span> 81 7/subop/compare %ecx 2/imm32 <span id="L135" class="LineNr">135 </span> 7c/jump-if-&lt; $unquote-stream-to-array:end/disp8 <span id="L136" class="LineNr">136 </span> 81 5/subop/subtract %ecx 2/imm32 <span id="L137" class="LineNr">137 </span> <span class="subxComment"># allocate</span> <span id="L138" class="LineNr">138 </span> (<a href='120allocate.subx.html#L576'>allocate-array</a> <span class="SpecialChar"><a href='120allocate.subx.html#L27'>Heap</a></span> %ecx *(ebp+0xc)) <span id="L139" class="LineNr">139 </span> <span class="subxComment"># var in/edx: (addr byte) = s-&gt;data + s-&gt;read + 1</span> <span id="L140" class="LineNr">140 </span> 8b/-&gt; *(esi+4) 2/r32/edx <span id="L141" class="LineNr">141 </span> 8d/copy-address *(esi+edx+0xd) 2/r32/edx <span class="subxComment"># Stream-data + 1</span> <span id="L142" class="LineNr">142 </span> <span class="subxComment"># var dest/eax: (addr byte) = data for out</span> <span id="L143" class="LineNr">143 </span> 8b/-&gt; *(ebp+0xc) 0/r32/eax <span id="L144" class="LineNr">144 </span> (<a href='120allocate.subx.html#L256'>lookup</a> *eax *(eax+4)) <span class="subxComment"># =&gt; eax</span> <span id="L145" class="LineNr">145 </span> 8d/copy-address *(eax+4) 0/r32/eax <span id="L146" class="LineNr">146 </span> <span class="subxComment">#</span> <span id="L147" class="LineNr">147 </span> (<a href='310copy-bytes.subx.html#L6'>copy-bytes</a> %edx %eax %ecx) <span id="L148" class="LineNr">148 </span><span class="Constant">$unquote-stream-to-array:end</span>: <span id="L149" class="LineNr">149 </span> <span class="subxS1Comment"># . restore registers</span> <span id="L150" class="LineNr">150 </span> 5e/pop-to-esi <span id="L151" class="LineNr">151 </span> 5a/pop-to-edx <span id="L152" class="LineNr">152 </span> 59/pop-to-ecx <span id="L153" class="LineNr">153 </span> 58/pop-to-eax <span id="L154" class="LineNr">154 </span> <span class="subxS1Comment"># . epilogue</span> <span id="L155" class="LineNr">155 </span> 89/&lt;- %esp 5/r32/ebp <span id="L156" class="LineNr">156 </span> 5d/pop-to-ebp <span id="L157" class="LineNr">157 </span> c3/return </pre> </body> </html> <!-- vim: set foldmethod=manual : -->