test_base_ui.py 6.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194
  1. # ============================================================================
  2. # This file is part of Pwman3.
  3. #
  4. # Pwman3 is free software; you can redistribute it and/or modify
  5. # it under the terms of the GNU General Public License, version 2
  6. # as published by the Free Software Foundation;
  7. #
  8. # Pwman3 is distributed in the hope that it will be useful,
  9. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  11. # GNU General Public License for more details.
  12. #
  13. # You should have received a copy of the GNU General Public License
  14. # along with Pwman3; if not, write to the Free Software
  15. # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
  16. # ============================================================================
  17. # Copyright (C) 2014 Oz Nahum Tiram <nahumoz@gmail.com>
  18. # ============================================================================
  19. import os
  20. import unittest
  21. from io import StringIO, BytesIO
  22. import sys
  23. from pwman.util.crypto_engine import CryptoEngine
  24. from .test_crypto_engine import give_key, DummyCallback
  25. from pwman.data.database import __DB_FORMAT__
  26. from .test_tools import (SetupTester)
  27. from pwman.data import factory
  28. from pwman.data.nodes import Node
  29. testdb = os.path.abspath(os.path.join(os.path.dirname(__file__),
  30. "test-baseui.pwman.db"))
  31. print(testdb)
  32. class dummy_stdin(object):
  33. def __init__(self):
  34. self.idx = -1
  35. self.ans = ['4', 'some fucking notes', 'X']
  36. def __call__(self, msg):
  37. self.idx += 1
  38. return self.ans[self.idx]
  39. class TestBaseUI(unittest.TestCase):
  40. @classmethod
  41. def tearDownClass(cls):
  42. for item in (testdb, 'foo.csv', 'pwman-export.csv'):
  43. try:
  44. os.unlink(item)
  45. except OSError:
  46. continue
  47. SetupTester().clean()
  48. def setUp(self):
  49. "test that the right db instance was created"
  50. dbver = __DB_FORMAT__
  51. self.db = factory.createdb('sqlite://' + testdb, dbver)
  52. self.tester = SetupTester(dbver, testdb)
  53. self.tester.create()
  54. def tearDown(self):
  55. self.tester.cli.do_exit("")
  56. def test_get_tags(self):
  57. sys.stdin = StringIO("foo bar baz\n")
  58. tags = self.tester.cli._get_tags(reader=lambda: "foo bar baz")
  59. self.assertListEqual(['foo', 'bar', 'baz'], tags)
  60. sys.stdin = sys.__stdin__
  61. def test_1_do_new(self):
  62. sys.stdin = BytesIO((b"alice\nsecret\nexample.com\nsome notes"
  63. b"\nfoo bar baz"))
  64. _node = self.tester.cli._do_new('')
  65. sys.stdin = sys.__stdin__
  66. self.assertListEqual([b'foo', b'bar', b'baz'], [t for t
  67. in _node.tags])
  68. nodeid = self.tester.cli._db.listnodes()
  69. self.assertListEqual([1], nodeid)
  70. nodes = self.tester.cli._db.getnodes(nodeid)
  71. ce = CryptoEngine.get()
  72. user = ce.decrypt(nodes[0][1])
  73. self.assertTrue(user, 'alice')
  74. tags = nodes[0][5:]
  75. for idx, t in enumerate(['foo', 'bar', 'baz']):
  76. self.assertTrue(t, tags[idx])
  77. def test_2_do_list(self):
  78. self.output = StringIO()
  79. sys.stdout = self.output
  80. self.tester.cli.do_list('')
  81. self.tester.cli.do_list('foo')
  82. self.tester.cli.do_list('bar')
  83. sys.stdout = sys.__stdout__
  84. self.output.getvalue()
  85. def test_3_do_export(self):
  86. self.tester.cli.do_export("{'filename':'foo.csv'}")
  87. with open('foo.csv') as f:
  88. l = f.readlines()
  89. self.assertIn('alice;example.com;secret;some notes;foo,bar,baz', l[1])
  90. def test_3a_do_export(self):
  91. self.tester.cli.do_export("f")
  92. with open('pwman-export.csv') as f:
  93. l = f.readlines()
  94. self.assertIn('alice;example.com;secret;some notes;foo,bar,baz', l[1])
  95. def test_4_do_forget(self):
  96. self.tester.cli.do_forget('')
  97. ce = CryptoEngine.get()
  98. self.assertIsNone(ce._cipher)
  99. def test_5_do_print(self):
  100. v = StringIO()
  101. sys.stdout = v
  102. self.tester.cli.do_print('1')
  103. self.assertIn('\x1b[31mUsername:\x1b[0m alice', v.getvalue())
  104. self.tester.cli.do_print('a')
  105. self.assertIn("print accepts only a single ID ...", v.getvalue())
  106. sys.stdout = sys.__stdout__
  107. def test_6_do_tags(self):
  108. v = StringIO()
  109. sys.stdout = v
  110. self.tester.cli.do_tags('1')
  111. v = v.getvalue()
  112. for t in ['foo', 'bar', 'baz']:
  113. t in v
  114. sys.stdout = sys.__stdout__
  115. print(v)
  116. def test_7_get_ids(self):
  117. # used by do_cp or do_open,
  118. # this spits many time could not understand your input
  119. self.assertEqual([1], self.tester.cli._get_ids('1'))
  120. self.assertListEqual([1, 2, 3, 4, 5], self.tester.cli._get_ids('1-5'))
  121. self.assertListEqual([], self.tester.cli._get_ids('5-1'))
  122. self.assertListEqual([], self.tester.cli._get_ids('5x-1'))
  123. self.assertListEqual([], self.tester.cli._get_ids('5x'))
  124. self.assertListEqual([], self.tester.cli._get_ids('5\\'))
  125. def test_8_do_edit_1(self):
  126. node = self.tester.cli._db.getnodes([1])[0]
  127. node = node[1:5] + [node[5:]]
  128. node = Node.from_encrypted_entries(*node)
  129. sys.stdin = StringIO(("1\nfoo\nx\n"))
  130. self.tester.cli.do_edit('1')
  131. v = StringIO()
  132. sys.stdin = sys.__stdin__
  133. sys.stdout = v
  134. self.tester.cli.do_print('1')
  135. self.assertIn('\x1b[31mUsername:\x1b[0m foo', v.getvalue())
  136. def test_8_do_edit_2(self):
  137. node = self.tester.cli._db.getnodes([1])[0]
  138. node = node[1:5] + [node[5:]]
  139. node = Node.from_encrypted_entries(*node)
  140. sys.stdin = StringIO(("2\ns3kr3t\nx\n"))
  141. self.tester.cli.do_edit('1')
  142. v = StringIO()
  143. sys.stdin = sys.__stdin__
  144. sys.stdout = v
  145. self.tester.cli.do_print('1')
  146. self.assertIn('\x1b[31mPassword:\x1b[0m s3kr3t', v.getvalue())
  147. def test_9_do_delete(self):
  148. self.assertIsNone(self.tester.cli._do_rm('x'))
  149. sys.stdin = StringIO("y\n")
  150. self.tester.cli.do_rm('1')
  151. sys.stdin = sys.__stdin__
  152. sys.stdout = StringIO()
  153. self.tester.cli.do_ls('')
  154. self.assertNotIn('alice', sys.stdout.getvalue())
  155. sys.stdout = sys.__stdout__
  156. def test_10_do_info(self):
  157. self.output = StringIO()
  158. sys.stdout = self.output
  159. self.tester.cli.do_info(b'')
  160. self.assertIn("test.pwman.db", sys.stdout.getvalue())
  161. if __name__ == '__main__':
  162. ce = CryptoEngine.get()
  163. ce.callback = DummyCallback()
  164. ce.changepassword(reader=give_key)
  165. unittest.main(verbosity=2, failfast=True)