def test_notice_insert():
"""We can insert and replace a notice"""
dmn = DMNotices()
doc = {
"some": "structure",
'effective_on': '2011-01-01',
'fr_url': 'http://example.com',
'publication_date': '2010-02-02',
'cfr_parts': ['222']
}
dmn.insert('docdoc', doc)
expected = {
"document_number": "docdoc",
"effective_on": date(2011, 1, 1),
"fr_url": "http://example.com",
"publication_date": date(2010, 2, 2),
"noticecfrpart__cfr_part": '222',
"notice": doc
}
assert list(Notice.objects.all().values(*expected.keys())) == [expected]
doc['fr_url'] = 'url2'
dmn.delete('docdoc')
dmn.insert('docdoc', doc)
expected['fr_url'] = 'url2'
assert list(Notice.objects.all().values(*expected.keys())) == [expected]
def test_notice_listing():
dmn = DMNotices()
n = Notice.objects.create(document_number='22',
fr_url='fr1',
notice={},
effective_on=date(2005, 5, 5),
publication_date=date(2001, 3, 3))
n.noticecfrpart_set.create(cfr_part='876')
n = Notice.objects.create(document_number='9',
fr_url='fr2',
notice={},
publication_date=date(1999, 1, 1))
n.noticecfrpart_set.create(cfr_part='876')
n.noticecfrpart_set.create(cfr_part='111')
assert dmn.listing() == [{
'document_number': '22',
'fr_url': 'fr1',
'publication_date': '2001-03-03',
'effective_on': '2005-05-05'
}, {
'document_number': '9',
'fr_url': 'fr2',
'publication_date': '1999-01-01'
}]
assert dmn.listing() == dmn.listing('876')
assert dmn.listing('888') == []
def setUp(self):
self.dmn = DMNotices()
def __init__(self):
self.dm = DMNotices()
self.es = ESNotices()
self.get = self.dm.get
self.listing = self.dm.listing
def test_notice_get_success():
Notice.objects.create(document_number='docdoc',
fr_url='frfr',
publication_date=date.today(),
notice={"some": "body"})
assert DMNotices().get('docdoc') == {'some': 'body'}
def test_notice_get_404():
assert DMNotices().get('docdoc') is None
