def setUp(self): if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session = db_storage.DBStorage() self.session.reload() self.amenity = Amenity(name="kitchen") if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session.new(self.amenity) self.session.save()
def setUp(self): if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session = db_storage.DBStorage() self.session.reload() self.state = State(name="A State") if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session.new(self.state) self.session.save()
def setUp(self): if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session = db_storage.DBStorage() self.session.reload() self.state = State(name="Another State") self.city = City(name="Another City", state_id=self.state.id) if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session.new(self.state) self.session.new(self.city) self.session.save()
def setUp(self): if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session = db_storage.DBStorage() self.session.reload() self.state = State(name="A State") self.city = City(name="A City", state_id=self.state.id) self.user = User(email="*****@*****.**", password="******", first_name="", last_name="") if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session.new(self.state) self.session.new(self.city) self.session.new(self.user) self.session.save()
def setUp(self): if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session = db_storage.DBStorage() self.session.reload() self.state = State(name="A State") self.city = City(name="A City", state_id=self.state.id) self.user = User(email="*****@*****.**", password="******") self.place = Place(city_id=self.city.id, user_id=self.user.id, name="", description="") if (os.getenv('HBNB_TYPE_STORAGE') == 'db'): self.session.new(self.state) self.session.new(self.city) self.session.new(self.user) self.session.new(self.place) self.session.save()
#!/usr/bin/python3 from os import getenv from models.base_model import BaseModel from models.user import User from models.amenity import Amenity from models.city import City from models.place import Place from models.state import State from models.review import Review from models.engine import file_storage from models.engine import db_storage if getenv('HBNB_TYPE_STORAGE') == "db": storage = db_storage.DBStorage() else: storage = file_storage.FileStorage() storage.reload()
def setUp(self): self.session = db_storage.DBStorage() self.session.reload()