def ReadAllGRRUsers(self):
for u in self.users.values():
yield objects.GRRUser(username=u["username"],
password=u.get("password"),
ui_mode=u.get("ui_mode"),
canary_mode=u.get("canary_mode"),
user_type=u.get("user_type"))
def testEmptyGRRUserReadWrite(self):
d = self.db
d.WriteGRRUser("foo")
u = d.ReadGRRUser("foo")
u_expected = objects.GRRUser(username="******")
self.assertEqual(u_expected, u)
def testReadingMultipleGRRUsersEntriesWorks(self):
d = self.db
u_foo = objects.GRRUser(
username="******",
ui_mode="ADVANCED",
canary_mode=True,
user_type=objects.GRRUser.UserType.USER_TYPE_ADMIN)
d.WriteGRRUser(u_foo.username,
ui_mode=u_foo.ui_mode,
canary_mode=u_foo.canary_mode,
user_type=u_foo.user_type)
u_bar = objects.GRRUser(username="******")
d.WriteGRRUser(u_bar.username)
users = sorted(d.ReadGRRUsers(), key=lambda x: x.username)
self.assertEqual(users[0], u_bar)
self.assertEqual(users[1], u_foo)
def ReadGRRUser(self, username):
try:
u = self.users[username]
return objects.GRRUser(password=u.get("password"),
ui_mode=u.get("ui_mode"),
canary_mode=u.get("canary_mode"))
except KeyError:
raise db.UnknownGRRUserError("Can't find user with name: %s" %
username)
def _RowToGRRUser(self, row):
username, password, ui_mode, canary_mode, user_type = row
result = objects.GRRUser(username=username,
ui_mode=ui_mode,
canary_mode=canary_mode,
user_type=user_type)
if password:
result.password.ParseFromString(password)
return result
def testFilledGRRUserReadWrite(self):
d = self.db
u_expected = objects.GRRUser(ui_mode="ADVANCED", canary_mode=True)
u_expected.password.SetPassword("blah")
d.WriteGRRUser("foo",
password=u_expected.password,
ui_mode=u_expected.ui_mode,
canary_mode=u_expected.canary_mode)
u = d.ReadGRRUser("foo")
self.assertEqual(u_expected, u)
def testFilledGRRUserReadWrite(self):
d = self.db
u_expected = objects.GRRUser(
username="******",
ui_mode="ADVANCED",
canary_mode=True,
user_type=objects.GRRUser.UserType.USER_TYPE_ADMIN)
u_expected.password.SetPassword("blah")
d.WriteGRRUser("foo",
password=u_expected.password,
ui_mode=u_expected.ui_mode,
canary_mode=u_expected.canary_mode,
user_type=u_expected.user_type)
u = d.ReadGRRUser("foo")
self.assertEqual(u_expected, u)
