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