mirror of
https://github.com/google/nomulus.git
synced 2025-07-02 17:23:32 +02:00
mv com/google/domain/registry google/registry
This change renames directories in preparation for the great package rename. The repository is now in a broken state because the code itself hasn't been updated. However this should ensure that git correctly preserves history for each file.
This commit is contained in:
parent
a41677aea1
commit
5012893c1d
2396 changed files with 0 additions and 0 deletions
247
javatests/google/registry/storage/drive/DriveConnectionTest.java
Normal file
247
javatests/google/registry/storage/drive/DriveConnectionTest.java
Normal file
|
@ -0,0 +1,247 @@
|
|||
// Copyright 2016 The Domain Registry Authors. All Rights Reserved.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package com.google.domain.registry.storage.drive;
|
||||
|
||||
import static com.google.common.io.ByteStreams.toByteArray;
|
||||
import static com.google.common.truth.Truth.assertThat;
|
||||
import static org.mockito.Matchers.anyString;
|
||||
import static org.mockito.Matchers.argThat;
|
||||
import static org.mockito.Matchers.eq;
|
||||
import static org.mockito.Mockito.times;
|
||||
import static org.mockito.Mockito.verify;
|
||||
import static org.mockito.Mockito.when;
|
||||
|
||||
import com.google.api.client.http.ByteArrayContent;
|
||||
import com.google.api.services.drive.Drive;
|
||||
import com.google.api.services.drive.Drive.Children;
|
||||
import com.google.api.services.drive.Drive.Files;
|
||||
import com.google.api.services.drive.model.ChildList;
|
||||
import com.google.api.services.drive.model.ChildReference;
|
||||
import com.google.api.services.drive.model.File;
|
||||
import com.google.api.services.drive.model.ParentReference;
|
||||
import com.google.common.collect.ImmutableList;
|
||||
import com.google.common.net.MediaType;
|
||||
import com.google.domain.registry.testing.ExceptionRule;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Rule;
|
||||
import org.junit.Test;
|
||||
import org.junit.runner.RunWith;
|
||||
import org.mockito.ArgumentMatcher;
|
||||
import org.mockito.Mock;
|
||||
import org.mockito.runners.MockitoJUnitRunner;
|
||||
|
||||
import java.util.Arrays;
|
||||
import java.util.List;
|
||||
|
||||
/** Tests for {@link DriveConnection}.*/
|
||||
@RunWith(MockitoJUnitRunner.class)
|
||||
public class DriveConnectionTest {
|
||||
|
||||
@Rule
|
||||
public final ExceptionRule thrown = new ExceptionRule();
|
||||
|
||||
@Mock
|
||||
Drive drive;
|
||||
|
||||
@Mock
|
||||
Files files;
|
||||
|
||||
@Mock
|
||||
Children children;
|
||||
|
||||
@Mock
|
||||
Files.Insert insert;
|
||||
|
||||
@Mock
|
||||
Files.Update update;
|
||||
|
||||
@Mock
|
||||
Children.List childrenList;
|
||||
|
||||
private static final byte[] DATA = {1, 2, 3};
|
||||
ChildList childList;
|
||||
DriveConnection driveConnection;
|
||||
List<String> allChildren;
|
||||
|
||||
private ArgumentMatcher<ByteArrayContent> hasByteArrayContent(final byte[] data) {
|
||||
return new ArgumentMatcher<ByteArrayContent>() {
|
||||
@Override
|
||||
public boolean matches(Object arg) {
|
||||
try {
|
||||
return Arrays.equals(data, toByteArray(((ByteArrayContent) arg).getInputStream()));
|
||||
} catch (Exception e) {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
@Before
|
||||
public void init() throws Exception {
|
||||
driveConnection = new DriveConnection();
|
||||
driveConnection.drive = drive;
|
||||
when(drive.files()).thenReturn(files);
|
||||
when(drive.children()).thenReturn(children);
|
||||
when(insert.execute()).thenReturn(new File().setId("id"));
|
||||
when(update.execute()).thenReturn(new File().setId("id"));
|
||||
|
||||
// Mocking required for listFiles.
|
||||
ChildReference child1 = new ChildReference().setId("child1");
|
||||
ChildReference child2 = new ChildReference().setId("child2");
|
||||
ChildReference child3 = new ChildReference().setId("child3");
|
||||
ChildReference child4 = new ChildReference().setId("child4");
|
||||
List<ChildReference> children1 = ImmutableList.of(child1, child2);
|
||||
List<ChildReference> children2 = ImmutableList.of(child3, child4);
|
||||
allChildren = ImmutableList.of(child1.getId(), child2.getId(), child3.getId(), child4.getId());
|
||||
ChildList childList1 = new ChildList();
|
||||
childList1.setItems(children1);
|
||||
childList1.setNextPageToken("page2");
|
||||
ChildList childList2 = new ChildList();
|
||||
childList2.setItems(children2);
|
||||
childList2.setNextPageToken(null);
|
||||
when(childrenList.execute()).thenReturn(childList1, childList2);
|
||||
when(childrenList.setQ(anyString())).thenReturn(childrenList);
|
||||
when(childrenList.getPageToken()).thenCallRealMethod();
|
||||
when(childrenList.setPageToken(anyString())).thenCallRealMethod();
|
||||
when(children.list("driveFolderId")).thenReturn(childrenList);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateFileAtRoot() throws Exception {
|
||||
when(files.insert(
|
||||
eq(new File().setTitle("title").setMimeType("image/gif")),
|
||||
argThat(hasByteArrayContent(DATA))))
|
||||
.thenReturn(insert);
|
||||
assertThat(driveConnection.createFile("title", MediaType.GIF, null, DATA)).isEqualTo("id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateFileInFolder() throws Exception {
|
||||
when(files.insert(
|
||||
eq(new File()
|
||||
.setTitle("title")
|
||||
.setMimeType("image/gif")
|
||||
.setParents(ImmutableList.of(new ParentReference().setId("parent")))),
|
||||
argThat(hasByteArrayContent(DATA))))
|
||||
.thenReturn(insert);
|
||||
assertThat(driveConnection.createFile("title", MediaType.GIF, "parent", DATA)).isEqualTo("id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateFolderAtRoot() throws Exception {
|
||||
when(files.insert(new File()
|
||||
.setTitle("title")
|
||||
.setMimeType("application/vnd.google-apps.folder")))
|
||||
.thenReturn(insert);
|
||||
assertThat(driveConnection.createFolder("title", null)).isEqualTo("id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateFolderInFolder() throws Exception {
|
||||
when(files.insert(new File()
|
||||
.setTitle("title")
|
||||
.setMimeType("application/vnd.google-apps.folder")
|
||||
.setParents(ImmutableList.of(new ParentReference().setId("parent")))))
|
||||
.thenReturn(insert);
|
||||
assertThat(driveConnection.createFolder("title", "parent")).isEqualTo("id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testListFiles_noQueryWithPagination() throws Exception {
|
||||
assertThat(driveConnection.listFiles("driveFolderId"))
|
||||
.containsExactlyElementsIn(allChildren);
|
||||
verify(childrenList).setPageToken("page2");
|
||||
verify(childrenList).setPageToken(null);
|
||||
verify(childrenList, times(0)).setQ(anyString());
|
||||
verify(childrenList, times(2)).getPageToken();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testListFiles_withQueryAndPagination() throws Exception {
|
||||
assertThat(driveConnection.listFiles("driveFolderId", "sampleQuery"))
|
||||
.containsExactlyElementsIn(allChildren);
|
||||
verify(childrenList).setPageToken("page2");
|
||||
verify(childrenList).setPageToken(null);
|
||||
verify(childrenList, times(1)).setQ("sampleQuery");
|
||||
verify(childrenList, times(2)).getPageToken();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateOrUpdateFile_succeedsForNewFile() throws Exception {
|
||||
when(files.insert(
|
||||
eq(new File()
|
||||
.setTitle("title")
|
||||
.setMimeType("video/webm")
|
||||
.setParents(ImmutableList.of(new ParentReference().setId("driveFolderId")))),
|
||||
argThat(hasByteArrayContent(DATA))))
|
||||
.thenReturn(insert);
|
||||
ChildList emptyChildList = new ChildList()
|
||||
.setItems(ImmutableList.<ChildReference> of())
|
||||
.setNextPageToken(null);
|
||||
when(childrenList.execute()).thenReturn(emptyChildList);
|
||||
assertThat(driveConnection.createOrUpdateFile(
|
||||
"title",
|
||||
MediaType.WEBM_VIDEO,
|
||||
"driveFolderId",
|
||||
DATA))
|
||||
.isEqualTo("id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateOrUpdateFile_succeedsForUpdatingFile() throws Exception {
|
||||
when(files.update(
|
||||
eq("id"),
|
||||
eq(new File().setTitle("title")),
|
||||
argThat(hasByteArrayContent(DATA))))
|
||||
.thenReturn(update);
|
||||
ChildList childList = new ChildList()
|
||||
.setItems(ImmutableList.of(new ChildReference().setId("id")))
|
||||
.setNextPageToken(null);
|
||||
when(childrenList.execute()).thenReturn(childList);
|
||||
assertThat(driveConnection.createOrUpdateFile(
|
||||
"title",
|
||||
MediaType.WEBM_VIDEO,
|
||||
"driveFolderId",
|
||||
DATA))
|
||||
.isEqualTo("id");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testCreateOrUpdateFile_throwsExceptionWhenMultipleFilesWithNameAlreadyExist()
|
||||
throws Exception {
|
||||
thrown.expect(IllegalStateException.class,
|
||||
"Could not update file 'title' in Drive folder id 'driveFolderId' "
|
||||
+ "because multiple files with that name already exist.");
|
||||
ChildList childList = new ChildList()
|
||||
.setItems(ImmutableList.of(
|
||||
new ChildReference().setId("id1"),
|
||||
new ChildReference().setId("id2")))
|
||||
.setNextPageToken(null);
|
||||
when(childrenList.execute()).thenReturn(childList);
|
||||
driveConnection.createOrUpdateFile("title", MediaType.WEBM_VIDEO, "driveFolderId", DATA);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUpdateFile_succeeds() throws Exception {
|
||||
when(files.update(
|
||||
eq("id"),
|
||||
eq(new File().setTitle("title")),
|
||||
argThat(hasByteArrayContent(DATA))))
|
||||
.thenReturn(update);
|
||||
assertThat(driveConnection.updateFile("id", "title", MediaType.WEBM_VIDEO, DATA))
|
||||
.isEqualTo("id");
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue