Skip to content

Commit 48f17ed

Browse files
committed
Implement test case for commit 0e8d18f
1 parent 0e8d18f commit 48f17ed

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed

tests.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -939,6 +939,25 @@ def callback(module_data):
939939
self.assertTrue(data['constants']['one'] == 1)
940940
self.assertTrue(data['constants']['two'] == 2)
941941

942+
def testRulesIterator(self):
943+
944+
rules = yara.compile(
945+
source='''
946+
rule test1 { condition: false }
947+
rule test2 { condition: false }
948+
rule test3 { condition: false }
949+
''')
950+
951+
for i, r in enumerate(rules, start=1):
952+
self.assertTrue(r.identifier == 'test%d' % i)
953+
954+
it = rules.__iter__()
955+
r = it.next()
956+
self.assertTrue(r.identifier == 'test1')
957+
r = it.next()
958+
self.assertTrue(r.identifier == 'test2')
959+
r = it.next()
960+
self.assertTrue(r.identifier == 'test3')
942961

943962

944963
if __name__ == "__main__":

0 commit comments

Comments
 (0)