def get_autocomplete_data(group, attribute):
groups = dict(
users=User.get_all(),
products=Product.get_all(),
customers = Customer.get_all(),
expenses=Expense.get_all(),
warehouses=Warehouse.get_all()
)
data = []
for obj in groups[group]:
value = getattr(obj, attribute)
if value not in set(data):
data.append(value)
return data
def test_should_return_all_expenses(self):
expenses = Expense.get_all()
self.assertEqual(expenses, [self.expense])