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)