def setUp(self):
super(RelatedFieldTests, self).setUp()
class Artist(Model):
has_one = ('Bio',)
belongs_to = ('Person',)
has_many = ('Song',)
has_and_belongs_to_many = ('Taste',)
self.Artist = Artist
class Bio(Model):
pass
self.Bio = Bio
class Person(Model):
pass
self.Person = Person
class Song(Model):
pass
self.Song = Song
class Taste(Model):
pass
self.Taste = Taste
create_tables()
create_indexes()
def test_new_index(self):
class Order(Model):
belongs_to = ('Customer',)
create_tables()
create_indexes()
self.assert_indexes_created('orders', ['customer_id'])
def setUp(self):
super(DropTablesTests, self).setUp()
class Artist(Model):
pass
create_tables()
def test_existing_table(self):
class Artist(Model):
pass
create_tables()
create_tables()
self.assert_table_created('artists')
def setUp(self):
super(UpdateTests, self).setUp()
class Artist(Model):
pass
self.Artist = Artist
create_tables()
create_indexes()
def setUp(self):
super(UpdateTests, self).setUp()
class Artist(Model):
pass
self.Artist = Artist
create_tables()
create_indexes()
def test_after_delete(self):
class Artist(Model):
def after_delete(self):
self['deleted'] = True
create_tables()
a = Artist.create()
a.delete()
assert a['deleted'] is True
self.assert_deleted(a._table, a.fields.as_dict())
def test_after_save(self):
class Artist(Model):
def after_save(self):
self['confirmed'] = False
create_tables()
a = Artist()
a.save()
assert a['confirmed'] is False
self.assert_not_saved(a._table, a.fields.as_dict())
def test_before_save(self):
class Artist(Model):
def before_save(self):
self['verified'] = True
create_tables()
a = Artist()
a.save()
assert a['verified'] is True
self.assert_saved(a._table, a.fields.as_dict())
def test_after_save(self):
class Artist(Model):
def after_save(self):
self['confirmed'] = False
create_tables()
a = Artist()
a.save()
assert a['confirmed'] is False
self.assert_not_saved(a._table, a.fields.as_dict())
def test_after_delete(self):
class Artist(Model):
def after_delete(self):
self['deleted'] = True
create_tables()
a = Artist.create()
a.delete()
assert a['deleted'] is True
self.assert_deleted(a._table, a.fields.as_dict())
def test_before_save(self):
class Artist(Model):
def before_save(self):
self['verified'] = True
create_tables()
a = Artist()
a.save()
assert a['verified'] is True
self.assert_saved(a._table, a.fields.as_dict())
def init_db():
logger.info("making db")
mk_db()
logger.info("making tables")
create_tables()
logger.info("creating indexes")
create_indexes()
#
# if __name__ == '__main__':
# init()
def setUp(self):
super(HasManyDescriptorTests, self).setUp()
class Artist(Model):
has_many = ('Song',)
self.Artist = Artist
class Song(Model):
pass
self.Song = Song
create_tables()
create_indexes()
def setUp(self):
super(RelatedM2MSetTests, self).setUp()
class Artist(Model):
has_and_belongs_to_many = ('Taste',)
self.Artist = Artist
class Taste(Model):
pass
self.Taste = Taste
create_tables()
create_indexes()
def setUp(self):
super(RelatedSetTests, self).setUp()
class Artist(Model):
has_many = ('Song',)
self.Artist = Artist
class Song(Model):
pass
self.Song = Song
create_tables()
create_indexes()
def setUp(self):
super(HasAndBelongsToManyDescriptorTests, self).setUp()
class Artist(Model):
has_and_belongs_to_many = ('Taste',)
self.Artist = Artist
class Taste(Model):
pass
self.Taste = Taste
create_tables()
create_indexes()
def setUp(self):
super(HasOneDescriptorTests, self).setUp()
class Artist(Model):
has_one = ('Bio',)
self.Artist = Artist
class Bio(Model):
pass
self.Bio = Bio
create_tables()
create_indexes()
def setUp(self):
super(BelongsToDescriptorTests, self).setUp()
class Artist(Model):
belongs_to = ('Person',)
self.Artist = Artist
class Person(Model):
pass
self.Person = Person
create_tables()
create_indexes()
def setUp(self):
super(RelatedSetTests, self).setUp()
class Artist(Model):
has_many = ('Song', )
self.Artist = Artist
class Song(Model):
pass
self.Song = Song
create_tables()
create_indexes()
def setUp(self):
super(RelatedM2MSetTests, self).setUp()
class Artist(Model):
has_and_belongs_to_many = ('Taste', )
self.Artist = Artist
class Taste(Model):
pass
self.Taste = Taste
create_tables()
create_indexes()
def setUp(self):
super(BelongsToDescriptorTests, self).setUp()
class Artist(Model):
belongs_to = ('Person', )
self.Artist = Artist
class Person(Model):
pass
self.Person = Person
create_tables()
create_indexes()
def setUp(self):
super(HasOneDescriptorTests, self).setUp()
class Artist(Model):
has_one = ('Bio', )
self.Artist = Artist
class Bio(Model):
pass
self.Bio = Bio
create_tables()
create_indexes()
def setUp(self):
super(HasAndBelongsToManyDescriptorTests, self).setUp()
class Artist(Model):
has_and_belongs_to_many = ('Taste', )
self.Artist = Artist
class Taste(Model):
pass
self.Taste = Taste
create_tables()
create_indexes()
def setUp(self):
super(HasManyDescriptorTests, self).setUp()
class Artist(Model):
has_many = ('Song', )
self.Artist = Artist
class Song(Model):
pass
self.Song = Song
create_tables()
create_indexes()
def setUp(self):
super(SaveTests, self).setUp()
class Artist(Model):
has_one = ('Bio',)
belongs_to = ('Person',)
self.Artist = Artist
class Bio(Model):
pass
self.Bio = Bio
class Person(Model):
pass
self.Person = Person
create_tables()
create_indexes()
def setUp(self):
super(SaveTests, self).setUp()
class Artist(Model):
has_one = ('Bio',)
belongs_to = ('Person',)
self.Artist = Artist
class Bio(Model):
pass
self.Bio = Bio
class Person(Model):
pass
self.Person = Person
create_tables()
create_indexes()
pass
class CompanyTechnology(Model):
pass
class CompanyTweet(Model):
pass
class CompanyEmailPattern(Model):
pass
class CompanyHiring(Model):
pass
# Other
class CompanySimilar(Model):
pass
class UserContact(Model):
pass
class CompanyEmailPattern(Model):
pass
class CompanyEmployee(Model):
pass
create_tables()
create_indexes()
def test_new_table(self):
class Artist(Model):
pass
create_tables()
self.assert_table_created('artists')
from remodel.helpers import create_tables, create_indexes
from remodel.models import Model
__author__ = 'amitassaraf'
class Lead(Model):
has_one = ('User', )
class User(Model):
belongs_to = ('Lead', )
class Team(Model):
pass
create_tables()
create_indexes()
