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