def test_can_get_or_create(self):
# Create
key1 = Key.get_or_create(self.db, 'some.random.key')
expect(key1.name).to_equal('some.random.key')
# Get
key2 = Key.get_or_create(self.db, 'some.random.key')
expect(key1.id).to_equal(key2.id)
def test_can_get_or_create(self):
# Create
key1 = Key.get_or_create(self.db, 'some.random.key')
expect(key1.name).to_equal('some.random.key')
# Get
key2 = Key.get_or_create(self.db, 'some.random.key')
expect(key1.id).to_equal(key2.id)
def _adjust_kwargs(cls, **kwargs):
if 'page' in kwargs:
kwargs['domain'] = kwargs['page'].domain
if 'page' in kwargs and 'uuid' in kwargs:
kwargs['page'].last_review_uuid = kwargs['uuid']
if 'number_of_violations' in kwargs:
number_of_violations = kwargs['number_of_violations']
del kwargs['number_of_violations']
if 'page' in kwargs:
kwargs['page'].violations_count = number_of_violations
violations = []
for i in range(number_of_violations):
db = cls.FACTORY_SESSION
key = Key.get_or_create(db, "violation.%d" % i)
violations.append(
Violation(
key=key,
value="value %d" % i,
points=i,
domain=kwargs['page'].domain
)
)
kwargs['violations'] = violations
return kwargs
def test_can_get_domains_full_data(self):
domains = []
for i in range(3):
domains.append(DomainFactory.create(name='domain-%d.com' % i))
pages = []
for i, domain in enumerate(domains):
pages.append([])
for j in range(3):
pages[i].append(PageFactory.create(domain=domain))
requests = reviews = []
for i, (domain, page) in enumerate(zip(domains, pages)):
for j in range(i + 1):
reviews.append(
ReviewFactory.create(domain=domain,
page=page[j],
is_active=True,
number_of_violations=(5 + 2 * j)))
requests.append(
RequestFactory.create(status_code=200 if j %
2 == 0 else 404,
domain_name=domain.name,
response_time=0.25 * (i + 1)))
self.server.application.violation_definitions = {
'key.%s' % i: {
'title':
'title.%s' % i,
'category':
'category.%s' % (i % 3),
'key':
Key.get_or_create(self.db, 'key.%d' % i,
'category.%d' % (i % 3))
}
for i in range(9)
}
response = yield self.authenticated_fetch('/domains-details')
expect(response.code).to_equal(200)
full_data = loads(response.body)
expect(full_data).to_length(3)
expect(full_data[0].keys()).to_length(10)
expect(map(lambda d: d['name'], full_data)).to_be_like(
['domain-0.com', 'domain-1.com', 'domain-2.com'])
expect(map(lambda d: d['pageCount'], full_data)).to_be_like([3, 3, 3])
expect(map(lambda d: d['reviewCount'],
full_data)).to_be_like([1, 2, 3])
expect(map(lambda d: d['violationCount'],
full_data)).to_be_like([5, 12, 21])
expect(map(lambda d: d['reviewPercentage'],
full_data)).to_be_like([33.33, 66.67, 100.0])
expect(map(lambda d: d['errorPercentage'],
full_data)).to_be_like([0.0, 50.0, 33.33])
expect(map(lambda d: d['averageResponseTime'],
full_data)).to_be_like([0.25, 0.5, 0.75])
def test_can_get_domain_grouped_violations(self):
domain1 = DomainFactory.create()
domain2 = DomainFactory.create()
page1 = PageFactory.create(domain=domain1)
page2 = PageFactory.create(domain=domain1)
page3 = PageFactory.create(domain=domain2)
ReviewFactory.create(domain=domain1, page=page1, is_active=True, number_of_violations=5)
ReviewFactory.create(domain=domain1, page=page2, is_active=True, number_of_violations=7)
ReviewFactory.create(domain=domain2, page=page3, is_active=True, number_of_violations=9)
self.server.application.violation_definitions = {
'key.%s' % i: {
'title': 'title.%s' % i,
'category': 'category.%s' % (i % 3),
'key': Key.get_or_create(self.db, 'key.%d' % i, 'category.%d' % (i % 3))
} for i in xrange(9)
}
response = yield self.http_client.fetch(
self.get_url('/domains/%s/violations' % domain1.name)
)
expect(response.code).to_equal(200)
domain_violations = loads(response.body)
expect(domain_violations).to_length(5)
expect(domain_violations.keys()).to_be_like(['domainName', 'violations', 'total', 'domainURL', 'domainId'])
expect(domain_violations['total']).to_equal(12)
expect(domain_violations['violations']).to_length(3)
counts = map(lambda v: v['count'], domain_violations['violations'])
expect(counts).to_be_like([5, 4, 3])
def _insert_keys(self, keys):
from holmes.models import Key
for name in keys.keys():
self.db.begin(subtransactions=True)
key = Key.get_or_create(self.db, name)
keys[name]['key'] = key
self.db.add(key)
self.db.commit()
def test_can_get_most_common_violations(self):
self.db.query(Violation).delete()
review = ReviewFactory.create()
for i in range(5):
key = Key.get_or_create(self.db, 'violation.0')
review.add_violation(key, 'value', 100, review.domain)
for j in range(2):
key = Key.get_or_create(self.db, 'violation.1')
review.add_violation(key, 'value', 300, review.domain)
self.server.application.violation_definitions = {
'violation.%d' % i: {
'title': 'title.%s' % i,
'category': 'category.%s' % i,
'key': Key.get_or_create(self.db, 'violation.%d' % i, 'category.%d' % i)
} for i in range(3)
}
self.db.flush()
response = yield self.authenticated_fetch('/most-common-violations/')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_be_like([
{'count': 5, 'name': 'title.0', 'category': 'category.0', 'key': 'violation.0'},
{'count': 2, 'name': 'title.1', 'category': 'category.1', 'key': 'violation.1'},
{'count': 0, 'name': 'title.2', 'category': 'category.2', 'key': 'violation.2'},
])
self.db.query(Violation).delete()
response = yield self.authenticated_fetch('/most-common-violations/')
violations_from_cache = loads(response.body)
expect(response.code).to_equal(200)
expect(violations_from_cache).to_be_like(violations)
def _adjust_kwargs(cls, **kwargs):
if 'page' in kwargs:
kwargs['domain'] = kwargs['page'].domain
if 'page' in kwargs and 'uuid' in kwargs:
kwargs['page'].last_review_uuid = kwargs['uuid']
if 'number_of_violations' in kwargs:
number_of_violations = kwargs['number_of_violations']
del kwargs['number_of_violations']
if 'page' in kwargs:
kwargs['page'].violations_count = number_of_violations
violations = []
for i in range(number_of_violations):
key = Key.get_or_create(db, 'key.%d' % i, 'category.%d' % (i % 3))
violations.append(
Violation(
key=key,
value="value %d" % i,
points=i,
domain=kwargs['page'].domain,
review_is_active=kwargs['is_active']
)
)
kwargs['violations'] = violations
if 'number_of_facts' in kwargs:
number_of_facts = kwargs['number_of_facts']
del kwargs['number_of_facts']
facts = []
for i in range(number_of_facts):
key = Key.get_or_create(db, 'key.%d' % i, 'category.%d' % (i % 3))
facts.append(Fact(key=key, value="value %d" % i))
kwargs['facts'] = facts
return kwargs
def _adjust_kwargs(cls, **kwargs):
if 'page' in kwargs:
kwargs['domain'] = kwargs['page'].domain
if 'page' in kwargs and 'uuid' in kwargs:
kwargs['page'].last_review_uuid = kwargs['uuid']
if 'number_of_violations' in kwargs:
number_of_violations = kwargs['number_of_violations']
del kwargs['number_of_violations']
if 'page' in kwargs:
kwargs['page'].violations_count = number_of_violations
violations = []
for i in range(number_of_violations):
key = Key.get_or_create(db, 'key.%d' % i,
'category.%d' % (i % 3))
violations.append(
Violation(key=key,
value="value %d" % i,
points=i,
domain=kwargs['page'].domain,
review_is_active=kwargs['is_active']))
kwargs['violations'] = violations
if 'number_of_facts' in kwargs:
number_of_facts = kwargs['number_of_facts']
del kwargs['number_of_facts']
facts = []
for i in range(number_of_facts):
key = Key.get_or_create(db, 'key.%d' % i,
'category.%d' % (i % 3))
facts.append(Fact(key=key, value="value %d" % i))
kwargs['facts'] = facts
return kwargs
def test_can_get_domains_full_data(self):
domains = []
for i in xrange(3):
domains.append(DomainFactory.create(name='domain-%d.com' % i))
pages = []
for i, domain in enumerate(domains):
pages.append([])
for j in xrange(3):
pages[i].append(PageFactory.create(domain=domain))
requests = reviews = []
for i, (domain, page) in enumerate(zip(domains, pages)):
for j in xrange(i + 1):
reviews.append(ReviewFactory.create(
domain=domain,
page=page[j],
is_active=True,
number_of_violations=(5 + 2 * j)
))
requests.append(RequestFactory.create(
status_code=200 if j % 2 == 0 else 404,
domain_name=domain.name,
response_time=0.25 * (i + 1)
))
self.server.application.violation_definitions = {
'key.%s' % i: {
'title': 'title.%s' % i,
'category': 'category.%s' % (i % 3),
'key': Key.get_or_create(self.db, 'key.%d' % i, 'category.%d' % (i % 3))
} for i in xrange(9)
}
response = yield self.http_client.fetch(
self.get_url('/domains-details')
)
expect(response.code).to_equal(200)
full_data = loads(response.body)
expect(full_data).to_length(3)
expect(full_data[0].keys()).to_length(10)
expect(map(lambda d: d['name'], full_data)).to_be_like(['domain-0.com', 'domain-1.com', 'domain-2.com'])
expect(map(lambda d: d['pageCount'], full_data)).to_be_like([3, 3, 3])
expect(map(lambda d: d['reviewCount'], full_data)).to_be_like([1, 2, 3])
expect(map(lambda d: d['violationCount'], full_data)).to_be_like([5, 12, 21])
expect(map(lambda d: d['reviewPercentage'], full_data)).to_be_like([33.33, 66.67, 100.0])
expect(map(lambda d: d['errorPercentage'], full_data)).to_be_like([0.0, 50.0, 33.33])
expect(map(lambda d: d['averageResponseTime'], full_data)).to_be_like([0.25, 0.5, 0.75])
def _insert_keys(self, keys):
for name in keys.keys():
key = Key.get_or_create(self.application.db, name)
keys[name]['key'] = key
category_name = keys[name].get('category', None)
if category_name:
category = self._insert_key_category(key, category_name)
key.category = category
self.application.db.add(key)
self.application.db.flush()
self.application.db.commit()
def test_count_by_violation_key_name(self):
self.db.query(Review).delete()
self.db.query(Violation).delete()
review = ReviewFactory.create(is_active=True)
for i in range(3):
key = Key.get_or_create(self.db, 'violation.%d' % i)
review.add_violation(key, 'value', 100, review.domain)
self.db.flush()
key_id = review.violations[0].key_id
count = Review.count_by_violation_key_name(self.db, key_id)
expect(count).to_equal(1)
for i in range(3):
key = Key.get_or_create(self.db, 'violation.0')
review.add_violation(key, 'value', 100, review.domain)
key_id = review.violations[0].key_id
count = Review.count_by_violation_key_name(self.db, key_id)
expect(count).to_equal(4)
def test_can_get_domain_top_category_violations(self):
domain1 = DomainFactory.create()
domain2 = DomainFactory.create()
page1 = PageFactory.create(domain=domain1)
page2 = PageFactory.create(domain=domain1)
page3 = PageFactory.create(domain=domain2)
ReviewFactory.create(domain=domain1,
page=page1,
is_active=True,
number_of_violations=5)
ReviewFactory.create(domain=domain1,
page=page2,
is_active=True,
number_of_violations=7)
ReviewFactory.create(domain=domain2,
page=page3,
is_active=True,
number_of_violations=9)
self.server.application.violation_definitions = {
'key.%s' % i: {
'title':
'title.%s' % i,
'category':
'category.%s' % (i % 3),
'key':
Key.get_or_create(self.db, 'key.%d' % i,
'category.%d' % (i % 3))
}
for i in range(9)
}
key = Key.get_by_name(self.db, 'key.0')
response = yield self.authenticated_fetch(
'/domains/%s/violations/%d/' % (domain1.name, key.category_id))
expect(response.code).to_equal(200)
domain_top_category = loads(response.body)
expect(domain_top_category).to_length(5)
expect(domain_top_category.keys()).to_be_like([
'violations', 'domainId', 'categoryId', 'domainURL', 'domainName'
])
expect(domain_top_category['violations']).to_length(3)
counts = map(lambda v: v['count'], domain_top_category['violations'])
expect(counts).to_be_like([2, 1, 1])
def test_can_get_most_common_violations(self):
self.db.query(Violation).delete()
self.clean_cache(cache_keys=['most-common-violations'])
review = ReviewFactory.create()
for i in range(5):
key = Key.get_or_create(self.db, 'violation1')
review.add_violation(key, 'value', 100, review.domain)
for j in range(2):
key = Key.get_or_create(self.db, 'violation2')
review.add_violation(key, 'value', 300, review.domain)
self.db.flush()
response = yield self.http_client.fetch(
self.get_url('/most-common-violations/')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_be_like([
{'count': 5, 'name': 'undefined', 'category': 'undefined', 'key': 'violation1'},
{'count': 2, 'name': 'undefined', 'category': 'undefined', 'key': 'violation2'}
])
self.db.query(Violation).delete()
response = yield self.http_client.fetch(
self.get_url('/most-common-violations/')
)
violations_from_cache = loads(response.body)
expect(response.code).to_equal(200)
expect(violations_from_cache).to_be_like(violations)
def test_get_by_violation_key_name(self):
self.db.query(Review).delete()
self.db.query(Violation).delete()
review = ReviewFactory.create(is_active=True)
for i in range(3):
key = Key.get_or_create(self.db, 'violation.%d' % i)
review.add_violation(key, 'value', 100, review.domain)
review.page.last_review_id = review.id
review.page.last_review_uuid = review.uuid
review.page.last_review_date = review.completed_date
self.db.flush()
key_id = review.violations[0].key_id
reviews = Review.get_by_violation_key_name(self.db, key_id)
expect(reviews).to_length(1)
def test_get_by_violation_key_name(self):
self.db.query(Review).delete()
self.db.query(Violation).delete()
review = ReviewFactory.create(is_active=True)
for i in range(3):
key = Key.get_or_create(self.db, "violation.%d" % i)
review.add_violation(key, "value", 100, review.domain)
review.page.last_review_id = review.id
review.page.last_review_uuid = review.uuid
review.page.last_review_date = review.completed_date
self.db.flush()
key_id = review.violations[0].key_id
reviews = Review.get_by_violation_key_name(self.db, key_id)
expect(reviews).to_length(1)
def test_can_get_domain_top_category_violations(self):
domain1 = DomainFactory.create()
domain2 = DomainFactory.create()
page1 = PageFactory.create(domain=domain1)
page2 = PageFactory.create(domain=domain1)
page3 = PageFactory.create(domain=domain2)
ReviewFactory.create(domain=domain1, page=page1, is_active=True, number_of_violations=5)
ReviewFactory.create(domain=domain1, page=page2, is_active=True, number_of_violations=7)
ReviewFactory.create(domain=domain2, page=page3, is_active=True, number_of_violations=9)
self.server.application.violation_definitions = {
'key.%s' % i: {
'title': 'title.%s' % i,
'category': 'category.%s' % (i % 3),
'key': Key.get_or_create(self.db, 'key.%d' % i, 'category.%d' % (i % 3))
} for i in range(9)
}
key = Key.get_by_name(self.db, 'key.0')
response = yield self.authenticated_fetch(
'/domains/%s/violations/%d/' % (domain1.name, key.category_id)
)
expect(response.code).to_equal(200)
domain_top_category = loads(response.body)
expect(domain_top_category).to_length(5)
expect(domain_top_category.keys()).to_be_like(['violations', 'domainId', 'categoryId', 'domainURL', 'domainName'])
expect(domain_top_category['violations']).to_length(3)
counts = map(lambda v: v['count'], domain_top_category['violations'])
expect(counts).to_be_like([2, 1, 1])
def test_can_get_most_common_violations(self):
self.db.query(Violation).delete()
review = ReviewFactory.create()
for i in range(5):
key = Key.get_or_create(self.db, 'violation.0')
review.add_violation(key, 'value', 100, review.domain)
for j in range(2):
key = Key.get_or_create(self.db, 'violation.1')
review.add_violation(key, 'value', 300, review.domain)
self.server.application.violation_definitions = {
'violation.%d' % i: {
'title':
'title.%s' % i,
'category':
'category.%s' % i,
'key':
Key.get_or_create(self.db, 'violation.%d' % i,
'category.%d' % i)
}
for i in range(3)
}
self.db.flush()
response = yield self.authenticated_fetch('/most-common-violations/')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_be_like([
{
'count': 5,
'name': 'title.0',
'category': 'category.0',
'key': 'violation.0'
},
{
'count': 2,
'name': 'title.1',
'category': 'category.1',
'key': 'violation.1'
},
{
'count': 0,
'name': 'title.2',
'category': 'category.2',
'key': 'violation.2'
},
])
self.db.query(Violation).delete()
response = yield self.authenticated_fetch('/most-common-violations/')
violations_from_cache = loads(response.body)
expect(response.code).to_equal(200)
expect(violations_from_cache).to_be_like(violations)
def test_can_get_violation_by_key_name(self):
domains = [DomainFactory.create(
name='g%s.com' % chr(i),
url='http://g%s.com/' % chr(i)
) for i in xrange(ord('a'), ord('d'))]
pages = [PageFactory.create(
domain=domains[i % 3],
url='%s%d' % (domains[i % 3].url, i % 2)
) for i in xrange(6)]
for i, page in enumerate(pages):
ReviewFactory.create(page=page, is_active=True, number_of_violations=i)
self.db.flush()
self.server.application.violation_definitions = {
'key.%s' % i: {
'title': 'title.%s' % i,
'category': 'category.%s' % (i % 3),
'key': Key.get_or_create(self.db, 'key.%d' % i, 'category.%d' % (i % 3))
} for i in xrange(6)
}
response = yield self.http_client.fetch(
self.get_url('/violation/key.1')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(4)
expect(violations['reviewsCount']).to_equal(4)
response = yield self.http_client.fetch(
self.get_url('/violation/key.1?page_size=2¤t_page=1')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_equal(4)
response = yield self.http_client.fetch(
self.get_url('/violation/key.1?page_filter=1')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_equal(2)
response = yield self.http_client.fetch(
self.get_url('/violation/key.1?domain_filter=gc.com')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_equal(8)
response = yield self.http_client.fetch(
self.get_url('/violation/key.1?domain_filter=foobar')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(4)
expect(violations['reviewsCount']).to_equal(4)
response = yield self.http_client.fetch(
self.get_url('/violation/key.1?domain_filter=gc.com&page_filter=1')
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(1)
expect(violations['reviewsCount']).to_equal(1)
def test_can_get_violation_by_key_name_using_no_external_search_provider(
self):
self.use_no_external_search_provider()
domains = [
DomainFactory.create(name='g%s.com' % chr(i),
url='http://g%s.com' % chr(i))
for i in range(ord('a'), ord('d'))
]
pages = [
PageFactory.create(domain=domains[i % 3],
url='%s/%d' % (domains[i % 3].url, i % 2))
for i in range(6)
]
for i, page in enumerate(pages):
review = ReviewFactory.create(
page=page,
is_active=True,
number_of_violations=i,
created_date=datetime(2014, 04, 15, 11, 44, i),
completed_date=datetime(2014, 04, 15, 11, 44, i * 2))
review.page.last_review_id = review.id
review.page.last_review_uuid = review.uuid
review.page.last_review_date = review.completed_date
self.db.flush()
self.server.application.violation_definitions = {
'key.%s' % i: {
'title':
'title.%s' % i,
'category':
'category.%s' % (i % 3),
'generic_description':
'description.%s' % (i % 3),
'key':
Key.get_or_create(self.db, 'key.%d' % i,
'category.%d' % (i % 3))
}
for i in range(6)
}
dt = datetime(2014, 04, 15, 11, 44, 4)
dt_timestamp = calendar.timegm(dt.utctimetuple())
response = yield self.authenticated_fetch('/violation/key.1')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(4)
expect(violations['reviewsCount']).to_equal(4)
expect(violations['reviews'][3]['domain']).to_equal('gc.com')
expect(violations['reviews'][3]['page']['url']).to_equal(
'http://gc.com/0')
expect(violations['reviews'][3]['page']['completedAt']).to_equal(
dt_timestamp)
response = yield self.authenticated_fetch(
'/violation/key.1?page_size=2¤t_page=1')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_equal(4)
response = yield self.authenticated_fetch(
'/violation/key.1?page_filter=1')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(4)
expect(violations['reviewsCount']).to_be_null()
response = yield self.authenticated_fetch(
'/violation/key.1?domain_filter=gc.com')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_be_null()
response = yield self.authenticated_fetch(
'/violation/key.1?domain_filter=gc.com&page_filter=1')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(1)
expect(violations['reviewsCount']).to_be_null()
try:
response = yield self.authenticated_fetch('/violation/foobar')
except HTTPError:
err = sys.exc_info()[1]
expect(err).not_to_be_null()
expect(err.code).to_equal(404)
expect(
err.response.reason).to_be_like('Invalid violation key foobar')
else:
assert False, 'Should not get this far'
try:
response = yield self.authenticated_fetch(
'/violation/key.1?domain_filter=foobar')
except HTTPError:
err = sys.exc_info()[1]
expect(err).not_to_be_null()
expect(err.code).to_equal(404)
expect(err.response.reason).to_be_like('Domain foobar not found')
else:
assert False, 'Should not get this far'
def test_can_get_violation_by_key_name_using_no_external_search_provider(self):
self.use_no_external_search_provider()
domains = [DomainFactory.create(
name='g%s.com' % chr(i),
url='http://g%s.com' % chr(i)
) for i in range(ord('a'), ord('d'))]
pages = [PageFactory.create(
domain=domains[i % 3],
url='%s/%d' % (domains[i % 3].url, i % 2)
) for i in range(6)]
for i, page in enumerate(pages):
review = ReviewFactory.create(
page=page,
is_active=True,
number_of_violations=i,
created_date=datetime(2014, 04, 15, 11, 44, i),
completed_date=datetime(2014, 04, 15, 11, 44, i * 2)
)
review.page.last_review_id = review.id
review.page.last_review_uuid = review.uuid
review.page.last_review_date = review.completed_date
self.db.flush()
self.server.application.violation_definitions = {
'key.%s' % i: {
'title': 'title.%s' % i,
'category': 'category.%s' % (i % 3),
'generic_description': 'description.%s' % (i % 3),
'key': Key.get_or_create(self.db, 'key.%d' % i, 'category.%d' % (i % 3))
} for i in range(6)
}
dt = datetime(2014, 04, 15, 11, 44, 4)
dt_timestamp = calendar.timegm(dt.utctimetuple())
response = yield self.authenticated_fetch('/violation/key.1')
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(4)
expect(violations['reviewsCount']).to_equal(4)
expect(violations['reviews'][3]['domain']).to_equal('gc.com')
expect(violations['reviews'][3]['page']['url']).to_equal('http://gc.com/0')
expect(violations['reviews'][3]['page']['completedAt']).to_equal(dt_timestamp)
response = yield self.authenticated_fetch(
'/violation/key.1?page_size=2¤t_page=1'
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_equal(4)
response = yield self.authenticated_fetch(
'/violation/key.1?page_filter=1'
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(4)
expect(violations['reviewsCount']).to_be_null()
response = yield self.authenticated_fetch(
'/violation/key.1?domain_filter=gc.com'
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(2)
expect(violations['reviewsCount']).to_be_null()
response = yield self.authenticated_fetch(
'/violation/key.1?domain_filter=gc.com&page_filter=1'
)
violations = loads(response.body)
expect(response.code).to_equal(200)
expect(violations).to_length(3)
expect(violations['title']).to_equal('title.1')
expect(violations['reviews']).to_length(1)
expect(violations['reviewsCount']).to_be_null()
try:
response = yield self.authenticated_fetch(
'/violation/foobar'
)
except HTTPError:
err = sys.exc_info()[1]
expect(err).not_to_be_null()
expect(err.code).to_equal(404)
expect(err.response.reason).to_be_like('Invalid violation key foobar')
else:
assert False, 'Should not get this far'
try:
response = yield self.authenticated_fetch(
'/violation/key.1?domain_filter=foobar'
)
except HTTPError:
err = sys.exc_info()[1]
expect(err).not_to_be_null()
expect(err.code).to_equal(404)
expect(err.response.reason).to_be_like('Domain foobar not found')
else:
assert False, 'Should not get this far'
