def test_to_dict(self):
"""Test output of to_dict() method"""
org = Organization(name='Example')
self.session.add(org)
dom1 = Domain(domain='example.com',
is_top_domain=True,
organization=org)
dom2 = Domain(domain='us.example.net',
is_top_domain=False,
organization=org)
self.session.add(dom1)
self.session.add(dom2)
self.session.commit()
# Tests
d = org.to_dict()
self.assertIsInstance(d, dict)
self.assertEqual(d['name'], 'Example')
doms = d['domains']
self.assertEqual(len(doms), 2)
d0 = doms[0]
self.assertEqual(d0['domain'], 'example.com')
self.assertEqual(d0['top_domain'], True)
self.assertEqual(d0['organization'], 'Example')
d1 = doms[1]
self.assertEqual(d1['domain'], 'us.example.net')
self.assertEqual(d1['top_domain'], False)
self.assertEqual(d1['organization'], 'Example')
def test_is_top_domain_invalid_type(self):
"""Check invalid values on top_domain bool collumn"""
with self.assertRaisesRegexp(StatementError, INVALID_DATATYPE_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain(domain='example.com', is_top_domain='True')
dom1.organization = org1
self.session.add(dom1)
self.session.commit()
def test_is_top_domain_invalid_type(self):
"""Check invalid values on top_domain bool collumn"""
with self.assertRaisesRegexp(StatementError, INVALID_DATATYPE_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain(domain='example.com', is_top_domain='True')
dom1.organization = org1
self.session.add(dom1)
self.session.commit()
def test_none_name_domains(self):
"""Check whether domains without name can be stored"""
with self.assertRaisesRegex(IntegrityError, NULL_CHECK_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain()
dom1.organization = org1
self.session.add(dom1)
self.session.commit()
def test_none_name_domains(self):
"""Check whether domains without name can be stored"""
with self.assertRaisesRegex(IntegrityError, NULL_CHECK_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain()
dom1.organization = org1
self.session.add(dom1)
self.session.commit()
def test_unique_domains(self):
"""Check whether domains are unique"""
with self.assertRaisesRegexp(IntegrityError, DUP_CHECK_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain(domain='example.com')
dom1.organization = org1
dom2 = Domain(domain='example.com')
dom2.organization = org1
self.session.add(dom1)
self.session.add(dom2)
self.session.commit()
def test_not_null_organizations(self):
"""Check whether every domain is assigned to an organization"""
with self.assertRaisesRegex(IntegrityError, NULL_CHECK_ERROR):
dom1 = Domain(domain='example.com')
self.session.add(dom1)
self.session.commit()
def test_none_name_domains(self):
"""Check whether domains without name can be stored"""
if sys.version_info[0] >= 3: # Python 3
expected = IntegrityError
else: # Python 2
expected = OperationalError
with self.assertRaisesRegexp(expected, NULL_CHECK_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain()
dom1.organization = org1
self.session.add(dom1)
self.session.commit()
def test_none_name_domains(self):
"""Check whether domains without name can be stored"""
if sys.version_info[0] >= 3: # Python 3
expected = IntegrityError
else: # Python 2
expected = OperationalError
with self.assertRaisesRegexp(expected, NULL_CHECK_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain()
dom1.organization = org1
self.session.add(dom1)
self.session.commit()
def test_to_dict(self):
"""Test output of to_dict() method"""
org = Organization(name='Example')
self.session.add(org)
dom = Domain(domain='example.com',
is_top_domain=True,
organization=org)
self.session.add(dom)
self.session.commit()
# Tests
d = dom.to_dict()
self.assertIsInstance(d, dict)
self.assertEqual(d['domain'], 'example.com')
self.assertEqual(d['top_domain'], True)
self.assertEqual(d['organization'], 'Example')
def test_to_dict(self):
"""Test output of to_dict() method"""
org = Organization(name='Example')
self.session.add(org)
dom = Domain(domain='example.com',
is_top_domain=True,
organization=org)
self.session.add(dom)
self.session.commit()
# Tests
d = dom.to_dict()
self.assertIsInstance(d, dict)
self.assertEqual(d['domain'], 'example.com')
self.assertEqual(d['top_domain'], True)
self.assertEqual(d['organization'], 'Example')
def test_not_null_organizations(self):
"""Check whether every domain is assigned to an organization"""
if sys.version_info[0] >= 3: # Python 3
expected = IntegrityError
else: # Python 2
expected = OperationalError
with self.assertRaisesRegexp(expected, NULL_CHECK_ERROR):
dom1 = Domain(domain='example.com')
self.session.add(dom1)
self.session.commit()
def test_unique_domains(self):
"""Check whether domains are unique"""
with self.assertRaisesRegexp(IntegrityError, DUP_CHECK_ERROR):
org1 = Organization(name='Example')
self.session.add(org1)
dom1 = Domain(domain='example.com')
dom1.organization = org1
dom2 = Domain(domain='example.com')
dom2.organization = org1
self.session.add(dom1)
self.session.add(dom2)
self.session.commit()
