tools/fileinfo/features: Add a test for limiting memory (--max-memory).

Author: s3rvac

tools/fileinfo/features/max-memory/ack.ex

@@ -0,0 +1,64 @@
+from regression_tests import *
+
+
+class TestAnalysisWhenNoLimit(Test):
+    """Checks that fileinfo correctly analyzes file when there is no memory
+    limit.
+
+    Test for https://github.com/avast-tl/retdec/issues/270
+    """
+
+    settings = TestSettings(
+        tool='fileinfo',
+        input='ack.ex',
+    )
+
+    def test_correctly_analyzes_file(self):
+        assert self.fileinfo.succeeded
+
+        self.assertEqual(
+            self.fileinfo.output['SHA256'],
+            '35f7b82373b66579d782fa61732da86717631eff95876b799a9549661709fd8b'
+        )
+
+
+class TestAnalysisWhenSufficient(Test):
+    """Checks that fileinfo correctly analyzes file when there is a sufficient
+    memory limit.
+
+    Test for https://github.com/avast-tl/retdec/issues/270
+    """
+
+    settings = TestSettings(
+        tool='fileinfo',
+        args=[
+            '--max-memory=104857600',  # 100 MB
+            '--max-memory-half-ram',
+        ],
+        input='ack.ex',
+    )
+
+    def test_correctly_analyzes_file(self):
+        assert self.fileinfo.succeeded
+
+        self.assertEqual(
+            self.fileinfo.output['SHA256'],
+            '35f7b82373b66579d782fa61732da86717631eff95876b799a9549661709fd8b'
+        )
+
+
+class TestAnalysisWhenInsufficientLimit(Test):
+    """Checks that fileinfo fails to analyze the file when the memory limit is
+    not sufficient.
+
+    Test for https://github.com/avast-tl/retdec/issues/270
+    """
+
+    settings = TestSettings(
+        tool='fileinfo',
+        args='--max-memory=100',  # 100 bytes
+        input='ack.ex',
+    )
+
+    def test_fails_to_analyze_file(self):
+        assert not self.fileinfo.succeeded, 'fileinfo succeeded but should have failed'