def test_get_by_name(self): category = KeysCategoryFactory.create(name='HOLMES') loaded_category = KeysCategory.get_by_name(self.db, 'HOLMES') expect(category).to_equal(loaded_category) expect(loaded_category.name).to_equal('HOLMES')
def test_can_get_most_common_violations(self): self.db.query(Violation).delete() self.db.query(Key).delete() category = KeysCategoryFactory.create(name="SEO") for i in range(3): key = KeyFactory.create(name="some.random.fact.%s" % i, category=category) for j in range(i): ViolationFactory.create(key=key, value="value", points=10 * i + j) violation_definitions = { "some.random.fact.1": {"title": "SEO", "category": "SEO"}, "some.random.fact.2": {"title": "SEO", "category": "SEO"}, } violations = Violation.get_most_common_violations(self.db, violation_definitions) expect(violations).to_be_like( [ {"count": 1, "key": "some.random.fact.1", "category": "SEO", "title": "SEO"}, {"count": 2, "key": "some.random.fact.2", "category": "SEO", "title": "SEO"}, ] ) violations = Violation.get_most_common_violations(self.db, violation_definitions, sample_limit=2) expect(violations).to_be_like([{"count": 2, "key": "some.random.fact.2", "category": "SEO", "title": "SEO"}])
def test_can_convert_keys_category_to_dict(self): category = KeysCategoryFactory.create() category_dict = category.to_dict() expect(category_dict).to_be_like({ 'name': str(category.name) })
def test_get_by_id(self): no_category = KeysCategory.get_by_id(self.db, -1) expect(no_category).to_be_null() category = KeysCategoryFactory.create(name='HOLMES') loaded_category = KeysCategory.get_by_id(self.db, category.id) expect(category).to_equal(loaded_category) expect(loaded_category.name).to_equal('HOLMES') expect(loaded_category.id).to_equal(category.id)
def test_can_create_key_category(self): category = KeysCategoryFactory.create() expect(category.id).not_to_be_null() expect(category.name).not_to_be_null() loaded = self.db.query(KeysCategory).get(category.id) expect(str(loaded)).to_be_like('%s' % category.name) expect(loaded.id).to_equal(category.id) expect(loaded.name).to_equal(category.name)
def test_can_add_category(self): key = KeyFactory.create(name='some.random.key') category = KeysCategoryFactory(name='SEO') key.category = category self.db.flush() loaded_category = self.db.query(Key).get(key.id).category expect(str(loaded_category.name)).to_be_like('%s' % category.name) expect(loaded_category.name).to_equal(category.name)
def test_can_convert_keys_category_to_dict(self): category = KeysCategoryFactory.create() category_dict = category.to_dict() expect(category_dict).to_be_like({'name': str(category.name)})