Provide JoinFilesWithPrefix case with Gochannel

This commit is contained in:
Harshavardhana
2014-12-05 09:49:07 -08:00
parent 1b283441a4
commit 9a5003dd25
5 changed files with 104 additions and 18 deletions

View File

@@ -40,7 +40,7 @@ func (s *MySuite) TestSplitStream(c *C) {
}
bytesWriter.Flush()
log.Println(strconv.Itoa(bytesBuffer.Len()))
ch := make(chan ByteMessage)
ch := make(chan SplitMessage)
reader := bytes.NewReader(bytesBuffer.Bytes())
go SplitStream(reader, 25, ch)
var resultsBuffer bytes.Buffer
@@ -52,9 +52,14 @@ func (s *MySuite) TestSplitStream(c *C) {
c.Assert(bytes.Compare(bytesBuffer.Bytes(), resultsBuffer.Bytes()), Equals, 0)
}
func (s *MySuite) TestFileSplit2(c *C) {
func (s *MySuite) TestFileSplitJoin(c *C) {
err := SplitFilesWithPrefix("TESTFILE", "1KB", "TESTPREFIX")
c.Assert(err, IsNil)
err = SplitFilesWithPrefix("TESTFILE", "1KB", "")
c.Assert(err, Not(IsNil))
err = JoinFilesWithPrefix(".", "TESTPREFIX", "")
c.Assert(err, Not(IsNil))
err = JoinFilesWithPrefix(".", "TESTPREFIX", "NEWFILE")
c.Assert(err, IsNil)
}