added some toplevel git-annex subcommand tests

note that test coverage doesn't work for those yet

1 files changed, 102 insertions, 6 deletions

diff --git a/test.hs b/test.hs
index 77a8606b3..b6b0c2740 100644
--- a/test.hs
+++ b/test.hs
@@ -1,14 +1,38 @@
+{- git-annex test suite
+ -
+ - Copyright 2010 Joey Hess <joey@kitenet.net>
+ -
+ - Licensed under the GNU GPL version 3 or higher.
+ -}
+
 import Test.HUnit
 import Test.HUnit.Tools
+import System.Directory
+import System.Posix.Directory (changeWorkingDirectory)
+import System.Posix.Files
+import System.Posix.Env
+import IO (bracket_, bracket)
+import Control.Monad (unless, when)
+import Data.List
+import System.IO.Error
 
-import GitRepo
+import qualified GitRepo as Git
 import Locations
 import Utility
 import TypeInternals
 
-alltests :: [Test]
-alltests = 
-	[ qctest "prop_idempotent_deencode" prop_idempotent_deencode
+main :: IO (Counts, Int)
+main = do
+	-- Add current directory to the from of PATH, so git-annex etc will
+	-- be used, no matter where it is run from.
+	cwd <- getCurrentDirectory
+	p <- getEnvDefault "PATH" ""
+	setEnv "PATH" (cwd++":"++p) True
+	runVerboseTests $ TestList [quickchecks, toplevels]
+
+quickchecks :: Test
+quickchecks = TestLabel "quickchecks" $ TestList
+	[ qctest "prop_idempotent_deencode" Git.prop_idempotent_deencode
 	, qctest "prop_idempotent_fileKey" prop_idempotent_fileKey
 	, qctest "prop_idempotent_key_read_show" prop_idempotent_key_read_show
 	, qctest "prop_idempotent_shellEscape" prop_idempotent_shellEscape
@@ -17,5 +41,77 @@ alltests =
 	, qctest "prop_relPathDirToDir_basics" prop_relPathDirToDir_basics
 	]
 
-main :: IO (Counts, Int)
-main = runVerboseTests (TestList alltests)
+toplevels :: Test
+toplevels = TestLabel "toplevel" $ TestList
+	[ test_init
+	, test_add
+	]
+
+test_init :: Test
+test_init = TestLabel "git-annex init" $ TestCase $ ingitrepo $ do
+	git_annex "init" ["-q", reponame] @? "init failed"
+	e <- doesFileExist annexlog
+	unless e $
+		assertFailure $ annexlog ++ " not created"
+	c <- readFile annexlog
+	unless (isInfixOf reponame c) $
+		assertFailure $ annexlog ++ " does not contain repo name"
+	where
+		annexlog = ".git-annex/uuid.log"
+		reponame = "test repo"
+
+test_add :: Test
+test_add = TestLabel "git-annex add" $ TestCase $ inannex $ do
+	writeFile file content
+	git_annex "add" ["-q", "foo"] @? "add failed"
+	s <- getSymbolicLinkStatus file
+        unless (isSymbolicLink s) $
+		assertFailure "git-annex add did not create symlink"
+	c <- readFile file
+	unless (c == content) $
+		assertFailure "file content changed during git-annex add"
+	r <- try (writeFile file $ content++"bar")
+	case r of
+		Left _ -> return () -- expected permission error
+		Right _ -> assertFailure "was able to modify annexed file content"
+	where
+		file = "foo"
+		content = "foo file content"
+
+git_annex :: String -> [String] -> IO Bool
+git_annex command params = boolSystem "git-annex" (command:params)
+
+inannex :: Assertion -> Assertion
+inannex a = ingitrepo $ do
+	git_annex "init" ["-q", reponame] @? "init failed"
+	a
+	where
+		reponame = "test repo"
+
+ingitrepo :: Assertion -> Assertion
+ingitrepo a = withgitrepo $ \r -> do
+	cwd <- getCurrentDirectory
+	bracket_ (changeWorkingDirectory $ Git.workTree r)
+		(\_ -> changeWorkingDirectory cwd)
+		a
+
+withgitrepo :: (Git.Repo -> Assertion) -> Assertion
+withgitrepo = bracket setup cleanup
+	where
+		tmpdir = ".t"
+		repodir = tmpdir ++ "/repo"
+		setup = do
+			cleanup True
+			createDirectory tmpdir
+			ok <- boolSystem "git" ["init", "-q", repodir]
+			unless ok $
+				assertFailure "git init failed"
+			return $ Git.repoFromPath repodir
+		cleanup _ = do
+			e <- doesDirectoryExist tmpdir
+			when e $ do
+				-- git-annex prevents annexed file content
+				-- from being removed with permissions
+				-- bits; undo
+				_ <- boolSystem "chmod" ["+rw", "-R", tmpdir]
+				removeDirectoryRecursive tmpdir