class TestLog(TestCase):
def setUp(self):
config = getNuauthConf()
config["nuauth_tls_request_cert"] = "0"
config["nuauth_user_logs_module"] = '"syslog"'
config["nuauth_user_session_logs_module"] = '"syslog"'
self.nuauth = Nuauth(config)
def tearDown(self):
self.nuauth.stop()
def findLog(self, match):
warning("Search string >%s< in log" % match)
matched = False
for line in self.nuauth.readlines(total_timeout=2.0):
if match in line:
return True
return False
def testLogin(self):
# Client login
client = createClientWithCerts()
self.assert_(connectClient(client))
# Check log output
self.assert_(
self.findLog("[nuauth] User %s connect on " % client.username))
# Client logout
client.stop()
self.assert_(
self.findLog("[nuauth] User %s disconnect on " % client.username))
class TestScript(TestCase):
def setUp(self):
# Prepare our new scripts
self.script_up = ReplaceFile(SCRIPT_UP, SCRIPT % "UP", MODE)
self.script_down = ReplaceFile(SCRIPT_DOWN, SCRIPT % "DOWN", MODE)
self.script_up.install()
self.script_down.install()
# Create nuauth
config = NuauthConf()
config["nuauth_user_session_logs_module"] = '"script"'
self.nuauth = Nuauth(config)
def tearDown(self):
# Restore scripts and nuauth config
self.script_up.desinstall()
self.script_down.desinstall()
self.nuauth.stop()
def checkScript(self, match):
warning("checkScript(%r)" % match)
for line in self.nuauth.readlines(total_timeout=2.0):
if line == match:
return True
return False
def testLogin(self):
# Client login
client = createClientWithCerts()
self.assert_(connectClient(client))
# Check log output
match = "SCRIPT UP COUNT=2 TEXT >>>%s %s<<<" \
% (client.username, client.ip)
self.assert_(self.checkScript(match))
# Client logout
client.stop()
match = "SCRIPT DOWN COUNT=2 TEXT >>>%s %s<<<" \
% (client.username, client.ip)
self.assert_(self.checkScript(match))
class MysqlLog(TestCase):
def setUp(self):
startNufw(["-s"])
config = NuauthConf()
config["nuauth_log_users"] = '9'
config["mysql_prefix_version"] = '1'
if POSTGRESQL:
config.need_restart = True
self.conn = pgdb.connect(host=DB_SERVER,
user=DB_USER,
password=DB_PASSWORD,
database=DB_DBNAME)
config["nuauth_user_logs_module"] = '"pgsql"'
config["nuauth_user_session_logs_module"] = '"pgsql"'
else:
self.conn = MySQLdb.Connect(host=DB_SERVER,
user=DB_USER,
passwd=DB_PASSWORD,
db=DB_DBNAME)
config["nuauth_user_logs_module"] = '"mysql"'
config["nuauth_user_session_logs_module"] = '"mysql"'
self.users = USERDB
self.user = self.users[0]
self.acls = PlaintextAcl()
self.acls.addAcl("web",
VALID_PORT,
self.user.gid,
log_prefix=LOG_PREFIX)
self.users.install(config)
self.acls.install(config)
self.nuauth = Nuauth(config)
self.start_time = int(time() - 1.1)
def query(self, sql):
if POSTGRESQL:
prefix = "PostgreSQL"
else:
prefix = "MySQL"
info("%s query: %s" % (prefix, sql))
cursor = self.conn.cursor()
cursor.execute(sql)
info("%s result: %s rows" % (prefix, cursor.rowcount))
return cursor
def fetchone(self, cursor):
row = cursor.fetchone()
if POSTGRESQL:
info("PostgreSQL fetchone(): %s" % repr(row))
else:
info("MySQL fetchone(): %s" % repr(row))
return row
def tearDown(self):
# Stop nuauth
self.nuauth.stop()
self.conn.close()
self.users.desinstall()
self.acls.desinstall()
def _login(self, sql):
# Client login
client = self.user.createClientWithCerts()
self.assert_(connectClient(client))
# Check number of rows
for when in retry(timeout=QUERY_TIMEOUT):
cursor = self.query(sql)
for line in self.nuauth.readlines():
pass
if cursor.rowcount:
break
self.assertEqual(cursor.rowcount, 1)
# Read row columns
(ip_saddr, user_id, username, os_sysname, os_release, os_version,
end_time) = self.fetchone(cursor)
if not POSTGRESQL:
ip_saddr = ntohl(ip_saddr) & 0xFFFFFFFF
# Check values
self.assertEqual(IP(ip_saddr), client.ip)
self.assertEqual(user_id, self.user.uid)
self.assertEqual(username, client.username)
self.assertEqual(os_sysname, OS_SYSNAME)
self.assertEqual(os_release, OS_RELEASE)
self.assertEqual(os_version, OS_VERSION)
return client
def _logout(self, sql, client):
# Client logout
# Use datetime.fromtimestamp() with int(time()) to have microsecond=0
logout_before = datetime_before()
client.stop()
for when in retry(timeout=QUERY_TIMEOUT):
# Get last MySQL row
cursor = self.query(sql)
# Check number of rows
if not cursor.rowcount:
continue
self.assertEqual(cursor.rowcount, 1)
# Read row columns
(ip_saddr, user_id, username, os_sysname, os_release, os_version,
end_time) = self.fetchone(cursor)
if not end_time:
continue
break
# Check values
if not POSTGRESQL:
# FIXME: Convert string to datetime for PostgreSQL
logout_after = datetime_after()
self.assert_(logout_before <= end_time <= logout_after)
