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_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_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()