60 lines
2.0 KiB
Ruby
60 lines
2.0 KiB
Ruby
require 'test/unit'
|
|
require 'rubygems'
|
|
require 'shadow'
|
|
require 'stringio'
|
|
class RubyShadowTest < Test::Unit::TestCase
|
|
STRUCT_METHODS = %w( sp_namp sp_pwdp sp_lstchg sp_min sp_max sp_warn sp_inact sp_expire sp_loginclass )
|
|
# --------------------------------------------------- test_smoke_test_getspent
|
|
def test_smoke_test_getspent
|
|
x = Shadow::Passwd.getspent
|
|
assert( x, "ensure you have permissions to check this (sudo?)" )
|
|
check_struct( x )
|
|
end
|
|
# ----------------------------------------------------- test_getspnam_for_user
|
|
def test_getspnam_for_user
|
|
user = `whoami`.strip
|
|
x = Shadow::Passwd.getspnam( user )
|
|
assert( x, "ensure you have permissions to check this (sudo?)" )
|
|
check_struct( x )
|
|
end
|
|
# ---------------------------------------- test_getspnam_for_non_existent_user
|
|
def test_getspnam_for_non_existent_user
|
|
assert_nil( Shadow::Passwd.getspnam( 'somebadusername' ) )
|
|
end
|
|
# -------------------------------------------------------------- test_putspent
|
|
def test_putspent
|
|
omit_if( Shadow::IMPLEMENTATION != 'SHADOW' ) if respond_to?( :omit_if )
|
|
#result = StringIO.open( '', 'w' ) do |fh|
|
|
File.open( 'test_password', 'w' ) do |fh|
|
|
Shadow::Passwd.add_password_entry( sample_entry, fh )
|
|
end
|
|
assert( File.read( 'test_password' ).match( sample_entry.first ) )
|
|
end
|
|
# --------------------------------------------------------------- check_struct
|
|
def check_struct( s )
|
|
STRUCT_METHODS.each do |m|
|
|
s.send( m )
|
|
end
|
|
end
|
|
# ----------------------------------------------------------------- test_entry
|
|
def sample_entry( &block )
|
|
return ['test_user', # sp_namp
|
|
'new_pass', # sp_pwdp
|
|
0, #sp_lastchg
|
|
0, #sp_min
|
|
0, #sp_max
|
|
0, #sp_warn
|
|
0, #sp_inact
|
|
0, #????
|
|
0, #sp_expire
|
|
0 # sp_flag
|
|
]
|
|
e = Shadow::Passwd::Entry.new
|
|
e.sp_namp = 'test_user'
|
|
e.sp_pwdp = 'password'
|
|
yield e if block_given?
|
|
return e
|
|
end
|
|
end
|
|
|