swarm/api: Added encrypted test cases

This commit is contained in:
Balint Gabor 2018-04-10 14:09:05 +02:00
parent c342a8c376
commit b74f35b078
2 changed files with 26 additions and 3 deletions

View file

@ -61,6 +61,14 @@ func TestClientUploadDownloadRaw(t *testing.T) {
// TestClientUploadDownloadFiles test uploading and downloading files to swarm // TestClientUploadDownloadFiles test uploading and downloading files to swarm
// manifests // manifests
func TestClientUploadDownloadFiles(t *testing.T) { func TestClientUploadDownloadFiles(t *testing.T) {
testClientUploadDownloadFiles(false, t)
}
func TestClientUploadDownloadFilesEncrypted(t *testing.T) {
testClientUploadDownloadFiles(true, t)
}
func testClientUploadDownloadFiles(toEncrypt bool, t *testing.T) {
srv := testutil.NewTestSwarmServer(t) srv := testutil.NewTestSwarmServer(t)
defer srv.Close() defer srv.Close()
@ -74,7 +82,7 @@ func TestClientUploadDownloadFiles(t *testing.T) {
Size: int64(len(data)), Size: int64(len(data)),
}, },
} }
hash, err := client.Upload(file, manifest, false) hash, err := client.Upload(file, manifest, toEncrypt)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -217,6 +225,14 @@ func TestClientUploadDownloadDirectory(t *testing.T) {
// TestClientFileList tests listing files in a swarm manifest // TestClientFileList tests listing files in a swarm manifest
func TestClientFileList(t *testing.T) { func TestClientFileList(t *testing.T) {
testClientFileList(false, t)
}
func TestClientFileListEncrypted(t *testing.T) {
testClientFileList(true, t)
}
func testClientFileList(toEncrypt bool, t *testing.T) {
srv := testutil.NewTestSwarmServer(t) srv := testutil.NewTestSwarmServer(t)
defer srv.Close() defer srv.Close()
@ -224,7 +240,7 @@ func TestClientFileList(t *testing.T) {
defer os.RemoveAll(dir) defer os.RemoveAll(dir)
client := NewClient(srv.URL) client := NewClient(srv.URL)
hash, err := client.UploadDirectory(dir, "", "", false) hash, err := client.UploadDirectory(dir, "", "", toEncrypt)
if err != nil { if err != nil {
t.Fatalf("error uploading directory: %s", err) t.Fatalf("error uploading directory: %s", err)
} }

View file

@ -458,6 +458,13 @@ func testBzzGetPath(encrypted bool, t *testing.T) {
// a trailing slash gets redirected to include the trailing slash so that // a trailing slash gets redirected to include the trailing slash so that
// relative URLs work as expected. // relative URLs work as expected.
func TestBzzRootRedirect(t *testing.T) { func TestBzzRootRedirect(t *testing.T) {
testBzzRootRedirect(false, t)
}
func TestBzzRootRedirectEncrypted(t *testing.T) {
testBzzRootRedirect(true, t)
}
func testBzzRootRedirect(toEncrypt bool, t *testing.T) {
srv := testutil.NewTestSwarmServer(t) srv := testutil.NewTestSwarmServer(t)
defer srv.Close() defer srv.Close()
@ -472,7 +479,7 @@ func TestBzzRootRedirect(t *testing.T) {
Size: int64(len(data)), Size: int64(len(data)),
}, },
} }
hash, err := client.Upload(file, "", false) hash, err := client.Upload(file, "", toEncrypt)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }