fix(CornerShot): updated total_shots earlier

zeronetworks · Feb 8, 2021 · 2ac1d17 · 2ac1d17
1 parent 1da0efe
commit 2ac1d17
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/cornershot/cornershot.py b/cornershot/cornershot.py
@@ -83,6 +83,8 @@ def add_many_shot_pairs(self, carrier_target_pairs, target_ports=None, destinati
                 for cls in tport_shot_class_pair[1]:
                     self.shot_list.append(cls(self.username, self.password, self.domain, carrier, target,target_port=target_port))
 
+        self.total_shots += len(self.shot_list)
+
     def _shots_generator(self, destinations, targets, target_ports, destination_ports=None):
         for destination in destinations:
             for target in targets:

diff --git a/setup.py b/setup.py
@@ -15,7 +15,7 @@
 setup(
     name='cornershot',
     python_requires='>=3',
-    version='0.2.5',
+    version='0.2.6',
     description='Library to test network connectivity',
     long_description_content_type='text/markdown',
     long_description=long_description,