def test_find(self): testDataContent = [ '648:abcdefghi', '517:QjhWkizN5t', '323:dwcIY99gTN1', '653:UlfY2UvMfeuX' ] data1 = DAO(testDataContent) found = data1.find('abcdefghi') assert found.tasktime == '648' and found.taskstring == 'abcdefghi' notFound = data1.find('nonexistent') assert notFound is None
def test_remove_entry(self): testDataContent = [ '648:abcdefghi', '517:QjhWkizN5t', '323:dwcIY99gTN1', '653:UlfY2UvMfeuX' ] data1 = DAO(testDataContent) assert data1.find('abcdefghi') data1.remove_entry('abcdefghi') assert data1.find('abcdefghi') is None with pytest.raises(Exception): data1.remove_entry('abcdefghi')
def get(self): ''' return all the certs ''' cert_col = DAO().get_collection("certificates") certs = cert_col.find({}, {'_id': 0}) return list(certs)
def test_add_entry(self): data1 = DAO([]) assert data1.is_empty() data1.add_entry('648', 'abcdefghi') assert not data1.is_empty() found = data1.find('abcdefghi') assert found.tasktime == '648' and found.taskstring == 'abcdefghi' with pytest.raises(Exception): data1.add_entry('648', 'abcdefghi') data1.add_entry('000', 'abcdefghi')
def test_advance_time(self): testDataContent = [ '648:abcdefghi', '517:QjhWkizN5t', '323:dwcIY99gTN1', '653:UlfY2UvMfeuX' ] data1 = DAO(testDataContent) with pytest.raises(Exception): data1.advance_time('nonexistent', '30') data1.advance_time('abcdefghi', '30') assert data1.find('abcdefghi').tasktime == '678'