def setUp(self):
super().setUp()
self.sub = mock.make_submission()
self.sub_id = self.sub.id
self.user = create_user()
self.user_id = self.user.id
self.tag = mock.make_tag()
self.tag_id = self.tag.id
def setUp(self):
super().setUp()
self.sub = mock.make_submission()
self.sub_id = self.sub.id
self.user = create_user()
self.user_id = self.user.id
self.tag = mock.make_tag()
self.tag_id = self.tag.id
def test_returns_used_tags(self):
tag = mock.make_tag("example")
sub = mock.make_submission()
user = create_user()
link = SubmissionTagLink(user=user, content_object=sub, tag=tag)
link.save()
results = TagsService.get_all_used_tag_names()
self.assertEqual(list(results), ["example"])
def setUp(self):
super().setUp()
self.tag = mock.make_tag()
self.sub = mock.make_submission()
self.user = create_user()
link = SubmissionTagLink(
user=self.user, content_object=self.sub, tag=self.tag)
link.save()
def setUp(self):
super().setUp()
self.tag = mock.make_tag("remove me")
self.sub = mock.make_submission()
self.user = create_user()
self.link = models.SubmissionTagLink(content_object=self.sub,
tag=self.tag,
user=self.user)
self.link.save()
def setUp(self):
super().setUp()
self.tag = mock.make_tag("remove me")
self.sub = mock.make_submission()
self.user = create_user()
self.link = models.SubmissionTagLink(
content_object=self.sub,
tag=self.tag,
user=self.user)
self.link.save()
def test_data_importer(self, pg):
from user_accounts.tests.mock import create_user
create_user(
name="Ben Golder",
email='*****@*****.**',
username='******')
from intake.management.data_import import (
DataImporter,
USERS_SQL,
SUBMISSIONS_SQL,
LOGS_SQL
)
importer = DataImporter(
import_from='postgres://*****:*****@dbhost:5432/fake_db_name',
ssl=True
)
uuids = mock.uuids(3)
mock_logs = [
{'user': '******'},
{'user': '******'},
{'user': '******'},
{'user': '******'},
{'user': '******'}]
for mock_log in mock_logs:
mock_log.update(
datetime=mock.fake.date_time_between('-1w', 'now'),
submission_key=random.choice(uuids),
event_type=random.choice(['added','referred','opened']))
mock_db_results = {
USERS_SQL: [
{'email': '*****@*****.**'},
{'email': '*****@*****.**'},
{'email': '*****@*****.**'}],
SUBMISSIONS_SQL: mock.fake_typeseam_submission_dicts(uuids),
LOGS_SQL: mock_logs
}
cursor = MagicMock()
def set_fake_results(sql):
data = mock_db_results[sql]
cursor.__iter__.return_value = data
cursor.execute.side_effect = set_fake_results
importer._cursor = cursor
importer.import_records(delete_existing=True)
expected_report = '''Beginning data import from `fake_db_name` on `dbhost`
--------
Successfully imported 3 users:
\tfound [email protected]
\tcreated [email protected]
\tcreated [email protected]
--------
No FormSubmission instances exist. Not deleting anything.
--------
Successfully imported 3 form submissions from `fake_db_name` on `dbhost`
--------
No ApplicationLogEntry instances exist. Not deleting anything.
--------
Successfully imported 5 event logs from `fake_db_name` on `dbhost`'''
self.assertEqual(importer.report(), expected_report)
def test_data_importer(self, pg):
from user_accounts.tests.mock import create_user
create_user(name="Ben Golder",
email='*****@*****.**',
username='******')
from intake.management.data_import import (DataImporter, USERS_SQL,
SUBMISSIONS_SQL, LOGS_SQL)
importer = DataImporter(
import_from=
'postgres://*****:*****@dbhost:5432/fake_db_name',
ssl=True)
uuids = mock.uuids(3)
mock_logs = [{
'user': '******'
}, {
'user': '******'
}, {
'user': '******'
}, {
'user': '******'
}, {
'user': '******'
}]
for mock_log in mock_logs:
mock_log.update(datetime=mock.fake.date_time_between('-1w', 'now'),
submission_key=random.choice(uuids),
event_type=random.choice(
['added', 'referred', 'opened']))
mock_db_results = {
USERS_SQL: [{
'email': '*****@*****.**'
}, {
'email': '*****@*****.**'
}, {
'email': '*****@*****.**'
}],
SUBMISSIONS_SQL:
mock.fake_typeseam_submission_dicts(uuids),
LOGS_SQL:
mock_logs
}
cursor = MagicMock()
def set_fake_results(sql):
data = mock_db_results[sql]
cursor.__iter__.return_value = data
cursor.execute.side_effect = set_fake_results
importer._cursor = cursor
importer.import_records(delete_existing=True)
expected_report = '''Beginning data import from `fake_db_name` on `dbhost`
--------
Successfully imported 3 users:
\tfound [email protected]
\tcreated [email protected]
\tcreated [email protected]
--------
No FormSubmission instances exist. Not deleting anything.
--------
Successfully imported 3 form submissions from `fake_db_name` on `dbhost`
--------
No ApplicationLogEntry instances exist. Not deleting anything.
--------
Successfully imported 5 event logs from `fake_db_name` on `dbhost`'''
self.assertEqual(importer.report(), expected_report)