#! /usr/bin/env python
"""Unit tests for SCardConnect/SCardGetAttrib/SCardDisconnect.
This test case can be executed individually, or with all other test cases
thru testsuite_scard.py.
__author__ = "http://www.gemalto.com"
Copyright 2001-2010 gemalto
Author: Jean-Daniel Aussel, mailto:jean-daniel.aussel@gemalto.com
This file is part of pyscard.
pyscard is free software; you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as published by
the Free Software Foundation; either version 2.1 of the License, or
(at your option) any later version.
pyscard is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public License
along with pyscard; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
"""
import struct
import unittest
from smartcard.scard import *
# import local_config for reader/card configuration
# configcheck.py is generating local_config.py in
# the test suite.
import sys
sys.path += ['..']
try:
from local_config import expectedATRs,expectedReaders,expectedReaderGroups,expectedATRinReader
except:
print 'execute test suite first to generate the local_config.py file'
sys.exit()
class testcase_getAttrib(unittest.TestCase):
"""Test scard API for SCardGetAttrib"""
def setUp(self):
hresult, self.hcontext = SCardEstablishContext( SCARD_SCOPE_USER )
self.assertEquals(hresult, 0)
hresult, self.readers = SCardListReaders( self.hcontext, [] )
self.assertEquals(hresult, 0)
def tearDown(self):
hresult = SCardReleaseContext( self.hcontext )
self.assertEquals(hresult, 0)
def _getAttrib(self, r):
if r<len(expectedATRs) and []!=expectedATRs[r]:
hresult, hcard, dwActiveProtocol = SCardConnect(
self.hcontext, self.readers[r], SCARD_SHARE_SHARED, SCARD_PROTOCOL_T0 | SCARD_PROTOCOL_T1)
self.assertEquals(hresult, 0)
try:
hresult, reader, state, protocol, atr = SCardStatus( hcard )
self.assertEquals(hresult, 0)
self.assertEquals( reader, expectedReaders[r] )
self.assertEquals( atr, expectedATRs[r] )
if scard.__dict__.has_key( 'SCARD_ATTR_ATR_STRING' ):
hresult, attrib = SCardGetAttrib( hcard, SCARD_ATTR_ATR_STRING )
self.assertEquals(hresult, 0)
self.assertEquals( expectedATRs[r], attrib )
if 'winscard'==resourceManager:
hresult, attrib = SCardGetAttrib( hcard, SCARD_ATTR_DEVICE_SYSTEM_NAME_A )
self.assertEquals(hresult, 0)
trimmedAttrib=attrib[:-1]
self.assertEquals( expectedReaders[r], apply( struct.pack, [ '<' + 'B' * len(trimmedAttrib)] + trimmedAttrib ) )
finally:
hresult = SCardDisconnect( hcard, SCARD_UNPOWER_CARD )
self.assertEquals(hresult, 0)
def test_getATR0(self):
testcase_getAttrib._getAttrib(self,0)
def test_getATR1(self):
testcase_getAttrib._getAttrib(self,1)
def test_getATR3(self):
testcase_getAttrib._getAttrib(self,2)
def test_getATR4(self):
testcase_getAttrib._getAttrib(self,3)
def suite():
suite1 = unittest.makeSuite(testcase_getAttrib)
return unittest.TestSuite((suite1))
if __name__ == '__main__':
unittest.main()
|