def delete_password(self, service, username):
'''
Delete the password for the username of the service
'''
inv = Inventory.get((Inventory.account_name) == username
& (Inventory.hostname == service))
inv.delete_instance()
class TestOrder(unittest.TestCase):
def setUp(self):
self.inventory = Inventory()
self.inventory.add("item #1", 50)
def test_order_filled(self):
order = Order("item #1", 50)
order.fill(self.inventory)
self.assertTrue(order.filled)
self.assertEquals(0, self.inventory.get("item #1"))
def test_order_not_filled(self):
order = Order("item #1", 51)
order.fill(self.inventory)
self.assertFalse(order.filled)
self.assertEquals(50, self.inventory.get("item #1"))
def delete_password(self, service, username):
'''
Delete the password for the username of the service
'''
inv = Inventory.get(
(Inventory.account_name) == username &
(Inventory.hostname == service)
)
inv.delete_instance()
def grant(hostname, user, account_name, verbose=False):
'''
Grant access to some credentials
'''
inv = Inventory.get(hostname=hostname, account_name=account_name)
ar = AccessRights.create(created_by=user.identifier,
user=user.identifier, host=inv.identifier)
plaintext = crypt.rsa_dec(user, inv.account_password, verbose)
cipher = crypt.rsa_enc(user, plaintext)
Credentials.create(access=ar.identifier, password=cipher)
def test_add_item_into_inventory():
inventory = Inventory()
inventory.add("item #1", 1)
assert inventory.get("item #1") == 1
def test_remove_then_add_item_in_inventory():
inventory = Inventory()
inventory.add("item #1", 5)
inventory.remove("item #1", 3).add("item #1", 4)
assert inventory.get("item #1") == 6
def test_get_item_in_inventory():
inventory = Inventory()
assert inventory.get("item #1") == 0
def test_remove_invalid_quantity_inventory():
inventory = Inventory()
inventory.add("item #1", 3).remove("item #1", 5)
assert inventory.get("item #1") == 3
def test_remove_item_from_inventory():
inventory = Inventory()
inventory.add("item #1", 5).remove("item #1", 3)
assert inventory.get("item #1") == 2
