class TestSubscribers(unittest.TestCase):
def __init__(self, *args, **kwargs):
super(TestSubscribers, self).__init__(*args, **kwargs)
if os.path.exists('subscribers') and os.path.isfile('subscribers'):
os.remove('subscribers')
self.subscribers = Subscribers()
def test_check_if_email(self):
self.subscribers.__check_if_email__('*****@*****.**')
with self.assertRaises(ValueError):
self.subscribers.__check_if_email__('Hello world!')
with self.assertRaises(ValueError):
self.subscribers.__check_if_email__('a@b')
with self.assertRaises(ValueError):
self.subscribers.__check_if_email__('@.com')
with self.assertRaises(ValueError):
self.subscribers.__check_if_email__(42)
def test_add(self):
self.subscribers.add('*****@*****.**')
self.subscribers.add('*****@*****.**')
with self.assertRaises(Exception):
self.subscribers.add('*****@*****.**')
s = Subscribers()
with self.assertRaises(Exception):
s.add('*****@*****.**')
s.add('*****@*****.**')
with self.assertRaises(Exception):
s.add('*****@*****.**')
def test_add(self):
self.subscribers.add('*****@*****.**')
self.subscribers.add('*****@*****.**')
with self.assertRaises(Exception):
self.subscribers.add('*****@*****.**')
s = Subscribers()
with self.assertRaises(Exception):
s.add('*****@*****.**')
s.add('*****@*****.**')
with self.assertRaises(Exception):
s.add('*****@*****.**')
import sys
from Subscribers import Subscribers
if len(sys.argv) < 2:
sys.exit('Usage: {0} e-mail'.format(sys.argv[0]))
s = Subscribers()
try:
s.add(sys.argv[1])
except Exception as e:
sys.exit(e.message)
print 'Registration succeeded!'
