added k8s stub adapter for execution environment
This commit is contained in:
@ -1,45 +0,0 @@
|
||||
package nullio
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type responseWriterStub struct {
|
||||
bytes.Buffer
|
||||
header http.Header
|
||||
}
|
||||
|
||||
func (r *responseWriterStub) Header() http.Header {
|
||||
return r.header
|
||||
}
|
||||
func (r *responseWriterStub) WriteHeader(_ int) {
|
||||
}
|
||||
|
||||
func (s *MainTestSuite) TestContentLengthWriter_Write() {
|
||||
header := http.Header(make(map[string][]string))
|
||||
buf := &responseWriterStub{header: header}
|
||||
writer := &ContentLengthWriter{Target: buf}
|
||||
part1 := []byte("-rw-rw-r-- 1 kali ka")
|
||||
contentLength := "42"
|
||||
part2 := []byte("li " + contentLength + " 1660763446 flag\nFL")
|
||||
part3 := []byte("AG")
|
||||
|
||||
count, err := writer.Write(part1)
|
||||
s.Require().NoError(err)
|
||||
s.Equal(len(part1), count)
|
||||
s.Empty(buf.String())
|
||||
s.Equal("", header.Get("Content-Length"))
|
||||
|
||||
count, err = writer.Write(part2)
|
||||
s.Require().NoError(err)
|
||||
s.Equal(len(part2), count)
|
||||
s.Equal("FL", buf.String())
|
||||
s.Equal(contentLength, header.Get("Content-Length"))
|
||||
|
||||
count, err = writer.Write(part3)
|
||||
s.Require().NoError(err)
|
||||
s.Equal(len(part3), count)
|
||||
s.Equal("FLAG", buf.String())
|
||||
s.Equal(contentLength, header.Get("Content-Length"))
|
||||
}
|
@ -1,95 +0,0 @@
|
||||
package nullio
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"github.com/openHPI/poseidon/tests"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestLs2JsonTestSuite(t *testing.T) {
|
||||
suite.Run(t, new(Ls2JsonTestSuite))
|
||||
}
|
||||
|
||||
type Ls2JsonTestSuite struct {
|
||||
tests.MemoryLeakTestSuite
|
||||
buf *bytes.Buffer
|
||||
writer *Ls2JsonWriter
|
||||
}
|
||||
|
||||
func (s *Ls2JsonTestSuite) SetupTest() {
|
||||
s.MemoryLeakTestSuite.SetupTest()
|
||||
s.buf = &bytes.Buffer{}
|
||||
s.writer = &Ls2JsonWriter{Target: s.buf, Ctx: context.Background()}
|
||||
}
|
||||
|
||||
func (s *Ls2JsonTestSuite) TestLs2JsonWriter_WriteCreationAndClose() {
|
||||
count, err := s.writer.Write([]byte(""))
|
||||
s.Zero(count)
|
||||
s.NoError(err)
|
||||
|
||||
s.Equal("{\"files\": [", s.buf.String())
|
||||
|
||||
s.writer.Close()
|
||||
s.Equal("{\"files\": []}", s.buf.String())
|
||||
}
|
||||
|
||||
func (s *Ls2JsonTestSuite) TestLs2JsonWriter_WriteFile() {
|
||||
input := "total 0\n-rw-rw-r-- 1 kali kali 0 1660763446 flag\n"
|
||||
count, err := s.writer.Write([]byte(input))
|
||||
s.Equal(len(input), count)
|
||||
s.NoError(err)
|
||||
s.writer.Close()
|
||||
|
||||
s.Equal("{\"files\": [{\"name\":\"flag\",\"entryType\":\"-\",\"size\":0,\"modificationTime\":1660763446"+
|
||||
",\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"}]}",
|
||||
s.buf.String())
|
||||
}
|
||||
|
||||
func (s *Ls2JsonTestSuite) TestLs2JsonWriter_WriteRecursive() {
|
||||
input := ".:\ntotal 4\ndrwxrwxr-x 2 kali kali 4096 1660764411 dir\n" +
|
||||
"-rw-rw-r-- 1 kali kali 0 1660763446 flag\n" +
|
||||
"\n./dir:\ntotal 4\n-rw-rw-r-- 1 kali kali 3 1660764366 another.txt\n"
|
||||
count, err := s.writer.Write([]byte(input))
|
||||
s.Equal(len(input), count)
|
||||
s.NoError(err)
|
||||
s.writer.Close()
|
||||
|
||||
s.Equal("{\"files\": ["+
|
||||
"{\"name\":\"./dir\",\"entryType\":\"d\",\"size\":4096,\"modificationTime\":1660764411,"+
|
||||
"\"permissions\":\"rwxrwxr-x\",\"owner\":\"kali\",\"group\":\"kali\"},"+
|
||||
"{\"name\":\"./flag\",\"entryType\":\"-\",\"size\":0,\"modificationTime\":1660763446,"+
|
||||
"\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"},"+
|
||||
"{\"name\":\"./dir/another.txt\",\"entryType\":\"-\",\"size\":3,\"modificationTime\":1660764366,"+
|
||||
"\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"}"+
|
||||
"]}",
|
||||
s.buf.String())
|
||||
}
|
||||
|
||||
func (s *Ls2JsonTestSuite) TestLs2JsonWriter_WriteRemaining() {
|
||||
input1 := "total 4\n-rw-rw-r-- 1 kali kali 3 1660764366 an.txt\n-rw-rw-r-- 1 kal"
|
||||
_, err := s.writer.Write([]byte(input1))
|
||||
s.NoError(err)
|
||||
s.Equal("{\"files\": [{\"name\":\"an.txt\",\"entryType\":\"-\",\"size\":3,\"modificationTime\":1660764366,"+
|
||||
"\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"}", s.buf.String())
|
||||
|
||||
input2 := "i kali 0 1660763446 flag\n"
|
||||
_, err = s.writer.Write([]byte(input2))
|
||||
s.NoError(err)
|
||||
s.writer.Close()
|
||||
s.Equal("{\"files\": [{\"name\":\"an.txt\",\"entryType\":\"-\",\"size\":3,\"modificationTime\":1660764366,"+
|
||||
"\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"},"+
|
||||
"{\"name\":\"flag\",\"entryType\":\"-\",\"size\":0,\"modificationTime\":1660763446,"+
|
||||
"\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"}]}", s.buf.String())
|
||||
}
|
||||
|
||||
func (s *Ls2JsonTestSuite) TestLs2JsonWriter_WriteLink() {
|
||||
input1 := "total 4\nlrw-rw-r-- 1 kali kali 3 1660764366 another.txt -> /bin/bash\n"
|
||||
_, err := s.writer.Write([]byte(input1))
|
||||
s.NoError(err)
|
||||
s.writer.Close()
|
||||
s.Equal("{\"files\": [{\"name\":\"another.txt\",\"entryType\":\"l\",\"linkTarget\":\"/bin/bash\",\"size\":3,"+
|
||||
"\"modificationTime\":1660764366,\"permissions\":\"rw-rw-r--\",\"owner\":\"kali\",\"group\":\"kali\"}]}",
|
||||
s.buf.String())
|
||||
}
|
@ -1,60 +0,0 @@
|
||||
package nullio
|
||||
|
||||
import (
|
||||
"context"
|
||||
"github.com/openHPI/poseidon/tests"
|
||||
"github.com/stretchr/testify/suite"
|
||||
"io"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
type MainTestSuite struct {
|
||||
tests.MemoryLeakTestSuite
|
||||
}
|
||||
|
||||
func TestMainTestSuite(t *testing.T) {
|
||||
suite.Run(t, new(MainTestSuite))
|
||||
}
|
||||
|
||||
func (s *MainTestSuite) TestReader_Read() {
|
||||
read := func(reader io.Reader, ret chan<- bool) {
|
||||
p := make([]byte, 0, 5)
|
||||
_, err := reader.Read(p)
|
||||
s.ErrorIs(io.EOF, err)
|
||||
close(ret)
|
||||
}
|
||||
|
||||
s.Run("WithContext_DoesNotReturnImmediately", func() {
|
||||
readingContext, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
readerReturned := make(chan bool)
|
||||
go read(&Reader{readingContext}, readerReturned)
|
||||
|
||||
select {
|
||||
case <-readerReturned:
|
||||
s.Fail("The reader returned before the timeout was reached")
|
||||
case <-time.After(tests.ShortTimeout):
|
||||
}
|
||||
})
|
||||
|
||||
s.Run("WithoutContext_DoesReturnImmediately", func() {
|
||||
readerReturned := make(chan bool)
|
||||
go read(&Reader{}, readerReturned)
|
||||
|
||||
select {
|
||||
case <-readerReturned:
|
||||
case <-time.After(tests.ShortTimeout):
|
||||
s.Fail("The reader returned before the timeout was reached")
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MainTestSuite) TestReadWriterWritesEverything() {
|
||||
readWriter := &ReadWriter{}
|
||||
p := []byte{1, 2, 3}
|
||||
n, err := readWriter.Write(p)
|
||||
s.NoError(err)
|
||||
s.Equal(len(p), n)
|
||||
}
|
Reference in New Issue
Block a user