Exemplo n.º 1
0
    def test_remove_existing_user(self):
        d = StubDataDriver()
        d.set_file_exists(True)
        d.set_text("a;a\nb;b\nc;c\n")

        p = Passwd(name="test.txt", driver=d)
        p.remove_user("b", "b")
        p.commit()
        self.assertEqual(d.get_text(), "a;a\nc;c\n")
Exemplo n.º 2
0
    def test_add_new_user(self):
        d = StubDataDriver()
        d.set_file_exists(False)

        p = Passwd(name="test.txt", driver=d)
        p.add_user("admin", "admin")
        p.commit()

        self.assertEqual(d.get_text(), "admin;admin\n")