From c49ff9dc397b03dd71765b79cd82c7c5a7209df8 Mon Sep 17 00:00:00 2001
From: Chen Yufei <cyf@chenyufei.info>
Date: Thu, 31 Oct 2013 22:47:44 +0800
Subject: [PATCH] Fix test as new site is consider as direct now.

---
 sitestat_test.go | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/sitestat_test.go b/sitestat_test.go
index 5d3641f..e744c68 100644
--- a/sitestat_test.go
+++ b/sitestat_test.go
@@ -191,8 +191,11 @@ func TestSiteStatGetVisitCnt(t *testing.T) {
 
 	g, _ := ParseRequestURI("gtemp.com")
 	si := ss.GetVisitCnt(g)
-	if si.AsBlocked() || si.AsDirect() || si.AsTempBlocked() {
-		t.Error("never visited site should not be considered as blocked/direct/temp blocked")
+	if !si.AsDirect() {
+		t.Error("never visited site should be considered as direct")
+	}
+	if si.AsBlocked() || si.AsTempBlocked() {
+		t.Error("never visited site should not be considered as blocked/temp blocked")
 	}
 	si.DirectVisit()
 	gw, _ := ParseRequestURI("www.gtemp.com")
@@ -242,15 +245,6 @@ func TestSiteStatGetVisitCnt(t *testing.T) {
 	}
 
 	g1, _ := ParseRequestURI("www.shoulddirect.com")
-	si = ss.GetVisitCnt(g1)
-	if si.AsBlocked() || si.AsDirect() || si.AsTempBlocked() {
-		t.Errorf("%s not visited, should return unknow visit method\n", g1.Host)
-	}
-	si.DirectVisit()
-	si = ss.GetVisitCnt(g1)
-	if si.AsDirect() || si.AsBlocked() || si.AsTempBlocked() {
-		t.Errorf("%s visited only once, should still return unknow visit method\n", g1.Host)
-	}
 	for i := 0; i < directDelta; i++ {
 		si.DirectVisit()
 	}