def test_create_relationship(self):
with self.app.app_context():
from src.database.db import init_db, distroy_db, get_db
from src.models.question_model import Question, PreQuestion
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_question_1 = Question(
question="Test Question 1",
slug="test_question_1",
language="en"
)
test_question_1.save()
test_prequestion_1 = PreQuestion(
text = "This is an example prequestion",
slug = "test_prequestion_1",
language = "en"
)
test_prequestion_1.save()
rel = test_prequestion_1.questions.connect(test_question_1)
pytest.test_question_1 = test_question_1
pytest.test_prequestion_1 = test_prequestion_1
pytest.test_prequestion_question_rel_1 = rel
def test_application_database_initialisation(self):
"""testing initialising the application initialisation"""
from src import create_app
app = create_app(mode='development',
static_path='../static',
templates_path='../templates',
instance_path='../instance')
with app.app_context():
from src.database.db import init_db
init_db(app)
from neomodel import config
assert config.DATABASE_URL == app.config['NEOMODEL_DATABASE_URI']
from neomodel import db
from src.models.survey_model import Survey, SurveyVersion
from src.models.conducted_survey_model import ConductedSurvey
from src.models.conducted_survey_question_model import ConductedSurveyQuestion
from src.models.question_model import Question, PreQuestion
from src.models.answers_model import Answer
assert Survey in db._NODE_CLASS_REGISTRY.values()
assert Question in db._NODE_CLASS_REGISTRY.values()
assert ConductedSurvey in db._NODE_CLASS_REGISTRY.values()
assert ConductedSurveyQuestion in db._NODE_CLASS_REGISTRY.values()
assert Answer in db._NODE_CLASS_REGISTRY.values()
assert SurveyVersion in db._NODE_CLASS_REGISTRY.values()
assert PreQuestion in db._NODE_CLASS_REGISTRY.values()
def test_create_relationship(self):
with self.app.app_context():
from src.database.db import init_db, get_db, distroy_db
from src.models.survey_model import Survey
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_survey_1 = Survey(
language="en",
slug = "test_survey_1"
)
test_survey_1.save()
test_survey_2 = Survey(
language="fr",
slug = "test_survey_2"
)
test_survey_2.save()
rel = test_survey_1.related_surveys.connect(
test_survey_2,
{
'reason': 'language',
'description': "These surveys are related"
}
)
pytest.test_survey_1 = test_survey_1
pytest.test_survey_2 = test_survey_2
pytest.test_survey_survey_rel_1 = rel
def test_create_relationship(self):
with self.app.app_context():
from src.database.db import get_db, init_db, distroy_db
from src.models.question_model import PreQuestion
from src.models.survey_model import SurveyVersion
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_prequestion_1 = PreQuestion(
slug="test_prequestion_1",
text="This is and example PreQuestion 1",
language="en")
test_prequestion_1.save()
test_surveyversion_1 = SurveyVersion(
title="Test SurveyVersion 1")
test_surveyversion_1.save()
rel = test_surveyversion_1.prequestions.connect(
test_prequestion_1)
pytest.test_prequestion_1 = test_prequestion_1
pytest.test_surveyversion_1 = test_surveyversion_1
pytest.test_surveyversion_prequestion_rel_1 = rel
def test_OPP_validate_Brands_Empty_NonZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Brands import Brands
b = Brands("Diefei",1)
resp = OPP.validate_brands(b)
assert resp == True
def test_OPP_validate_bankaccout_validCheck_ZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.BankAccounts import Bankaccount
p = Bankaccount("HBL","Danish","PK-0123455",1)
resp = OPP.validate_bankaccount(p)
assert resp == True
def test_OPP_validate_order_transported_point_Empty_NonZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Orders import Orders
o = Orders(1,1,None,None,None,None,None,None)
resp = OPP.validate_order(o)
resp = json.loads(resp)
assert resp['status'] != 0
def test_OPP_validate_order_validCheck_ZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Orders import Orders
p = Orders(1,1,'abc',2000,1000,'lahore',12,1)
resp = OPP.validate_order(p)
assert resp == True
def test_OPP_validate_product_emptyCheck_nonZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Products import Products
p = Products()
resp = OPP.validate_product(p)
resp = json.loads(resp)
assert resp['status'] != 0
def test_OPP_validate_product_nameEmpty_nonZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Products import Products
p = Products(None,bytearray('abc','utf-8'),1234,None,None,None,1)
resp = OPP.validate_product(p)
resp = json.loads(resp)
assert resp['status'] != 0
def test_OPP_validate_Brands__shopkeeper_id_Empty_NonZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Brands import Brands
b = Brands("Diefei",None)
resp = OPP.validate_brands(b)
resp = json.loads(resp)
assert resp['status'] != 0
def test_OPP_validate_product_validCheck_ZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.Products import Products
p = Products("abc",bytearray('abc','utf-8'),None,None,121,None,1)
resp = OPP.validate_product(p)
assert resp == True
def test_OPP_validate_bankaccount_bank_name_Empty_NonZeroStatus():
from src.logic_processor.order_products_processor import OPP
from src.database.db import init_db
init_db()
from src.models.BankAccounts import Bankaccount
b = Bankaccount(None,"Danish",12345,1)
resp = OPP.validate_bankaccount(b)
resp = json.loads(resp)
assert resp['status'] != 0
def test_create_node(self):
with self.app.app_context():
from src.database.db import get_db, distroy_db, init_db
from src.models.survey_model import SurveyVersion
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_survey_version_1 = SurveyVersion(
title="Survey Version 1"
)
test_survey_version_1.save()
pytest.test_survey_version_1 = test_survey_version_1
def test_set_up(self):
with self.app.app_context():
from src.database.db import get_db, init_db, distroy_db
from src.models.survey_model import Survey
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_survey_1 = Survey(slug="test_survey_1", language="en")
test_survey_1.save()
pytest.test_survey_1 = test_survey_1 # type: Survey
def test_create_node(self):
with self.app.app_context():
from src.database.db import get_db, distroy_db, init_db
distroy_db(self.app)
init_db(self.app)
from src.models.conducted_survey_model import ConductedSurvey
transaction_factory = get_db()
current_transaction = transaction_factory.transaction
with current_transaction:
from src.models.conducted_survey_model import ConductedSurvey
test_conducted_survey_1 = ConductedSurvey()
test_conducted_survey_1.save()
pytest.test_conducted_survey_1 = test_conducted_survey_1
pytest.nodeId = test_conducted_survey_1.nodeId
def test_create_answer_node(self):
with self.app.app_context():
from src.database.db import get_db, init_db, distroy_db
distroy_db(self.app)
init_db(self.app)
from src.models.answers_model import Answer
current_transaction = get_db().transaction
with current_transaction:
test_answer_1 = Answer(answer="this is an answer 1")
test_answer_1.save()
node = get_db().cypher_query("MATCH (a:Answer) RETURN a")
assert test_answer_1.answer == \
node[0][0][0]._properties['answer']
assert test_answer_1.updatedOn is not None
assert test_answer_1.addedOn is not None
assert test_answer_1.id is not None
def test_create_node(self):
with self.app.app_context():
from src.database.db import get_db, distroy_db, init_db
distroy_db(self.app)
init_db(self.app)
from src.models.survey_model import Survey
transaction_factory = get_db()
current_transaction = transaction_factory.transaction
with current_transaction:
from src.models.survey_model import Survey
test_survey_1 = Survey(
slug="test_survey_1",
language="en"
)
test_survey_1.save()
pytest.test_survey_1 = test_survey_1
def test_create_node(self):
with self.app.app_context():
from src.database.db import get_db, init_db, distroy_db
from src.models.conducted_survey_question_model import ConductedSurveyQuestion
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_conducted_survey_question_1 = ConductedSurveyQuestion()
test_conducted_survey_question_1.save()
assert test_conducted_survey_question_1.nodeId is not None
assert test_conducted_survey_question_1.addedOn is not None
assert isinstance(test_conducted_survey_question_1.addedOn, datetime)
assert test_conducted_survey_question_1.sentimentSet is True
assert test_conducted_survey_question_1.sentimentCalculated is False
def test_schema_dump(self):
with self.app.app_context():
from src.database.db import get_db, init_db, distroy_db
from src.models.survey_model import Survey
from src.utils.marshmallow.survey_schema import SurveySchema
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_survey_1 = Survey(
slug="test_survey_1",
language="en",
)
test_survey_1.save()
test_output_1 = SurveySchema().dump(test_survey_1)
pytest.test_survey_1 = test_survey_1
pytest.test_output_1_data = test_output_1.data
assert bool(test_output_1.errors) is False
def test_schema_dump(self):
with self.app.app_context():
from src.database.db import get_db, init_db, distroy_db
from src.models.survey_model import SurveyVersion
from src.utils.marshmallow.surveyversion_schema import SurveyVersionSchema
distroy_db(self.app)
init_db(self.app)
current_transaction = get_db().transaction
with current_transaction:
test_surveyversion_1 = SurveyVersion(
title="Test SurveyVersion 1", )
test_surveyversion_1.save()
pytest.test_surveyversion_1 = test_surveyversion_1
test_output_1 = SurveyVersionSchema().dump(test_surveyversion_1)
pytest.test_output_1_data = test_output_1.data
pytest.test_output_1_errors = test_output_1.errors
assert bool(pytest.test_output_1_errors) == False
def initialise_database():
init_db(app)
help='Change the file name from table_records to whatever you wish')
parser.add_argument(
'--getpolygondata',
default=False,
action='store_true',
help='Change the file name from table_records to whatever you wish')
args = parser.parse_args()
print("Program Start at ", datetime.now())
if args.getpolygondata:
print("Getting Tickers from Polygon")
get_all_tickers_data()
print("Getting submissions...")
# call reddit api to get results
current_scores, prev_scores = get_submission_generators(
args.interval, args.sub, args.allsub, args.psaw)
print("Populating results...")
results_df = populate_df(current_scores, prev_scores, args.interval)
filter_df(results_df)
print("Program Complete at ", datetime.now())
if __name__ == '__main__':
init_db()
main()
