Fix predicate

AndMu · Dec 7, 2017 · ea0440a · ea0440a
1 parent e08d890
commit ea0440a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/Wikiled.YiScanner.Tests/Client/Predicates/NewFilesPredicateTests.cs b/src/Wikiled.YiScanner.Tests/Client/Predicates/NewFilesPredicateTests.cs
@@ -21,11 +21,8 @@ public void CanDownload()
             var result = instance.CanDownload(null, "Test.file", DateTime.Now);
             Assert.IsFalse(result);
             result = instance.CanDownload(DateTime.Now, "Test.file", DateTime.Now);
-            Assert.IsTrue(result);
-            result = instance.CanDownload(null, "Test.file", DateTime.Now);
             Assert.IsFalse(result);
-            instance.Downloaded("Test.file");
-            result = instance.CanDownload(DateTime.Now, "Test.file", DateTime.Now);
+            result = instance.CanDownload(null, "Test.file", DateTime.Now);
             Assert.IsFalse(result);
 
             result = instance.CanDownload(DateTime.Now,  "Test2.file", DateTime.Now);

diff --git a/src/YiScanner/Client/Predicates/NewFilesPredicate.cs b/src/YiScanner/Client/Predicates/NewFilesPredicate.cs
@@ -14,6 +14,11 @@ public bool CanDownload(DateTime? lastScan, string file, DateTime modified)
                 return false;
             }
 
+            if (lastScan == null)
+            {
+                table[file] = true;
+            }
+
             return lastScan != null;
         }