| /* |
| * Copyright (C) 2009 Marc Blank |
| * Licensed to The Android Open Source Project. |
| * |
| * 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.android.exchange; |
| |
| import com.android.email.provider.EmailContent.Account; |
| |
| import org.apache.http.Header; |
| import org.apache.http.client.methods.HttpPost; |
| import org.apache.http.client.methods.HttpRequestBase; |
| |
| import android.content.Context; |
| import android.test.AndroidTestCase; |
| |
| import java.io.File; |
| import java.io.IOException; |
| |
| /** |
| * You can run this entire test case with: |
| * runtest -c com.android.exchange.EasSyncServiceTests email |
| */ |
| public class EasSyncServiceTests extends AndroidTestCase { |
| Context mMockContext; |
| |
| @Override |
| public void setUp() throws Exception { |
| super.setUp(); |
| mMockContext = getContext(); |
| } |
| |
| /** |
| * Test that our unique file name algorithm works as expected. |
| * @throws IOException |
| */ |
| public void testCreateUniqueFile() throws IOException { |
| // Delete existing files, if they exist |
| EasSyncService svc = new EasSyncService(); |
| svc.mContext = mMockContext; |
| try { |
| String fileName = "A11achm3n1.doc"; |
| File uniqueFile = svc.createUniqueFileInternal(null, fileName); |
| assertEquals(fileName, uniqueFile.getName()); |
| if (uniqueFile.createNewFile()) { |
| uniqueFile = svc.createUniqueFileInternal(null, fileName); |
| assertEquals("A11achm3n1-2.doc", uniqueFile.getName()); |
| if (uniqueFile.createNewFile()) { |
| uniqueFile = svc.createUniqueFileInternal(null, fileName); |
| assertEquals("A11achm3n1-3.doc", uniqueFile.getName()); |
| } |
| } |
| fileName = "A11achm3n1"; |
| uniqueFile = svc.createUniqueFileInternal(null, fileName); |
| assertEquals(fileName, uniqueFile.getName()); |
| if (uniqueFile.createNewFile()) { |
| uniqueFile = svc.createUniqueFileInternal(null, fileName); |
| assertEquals("A11achm3n1-2", uniqueFile.getName()); |
| } |
| } finally { |
| // These are the files that should be created earlier in the test. Make sure |
| // they are deleted for the next go-around |
| File directory = getContext().getFilesDir(); |
| String[] fileNames = new String[] {"A11achm3n1.doc", "A11achm3n1-2.doc", "A11achm3n1"}; |
| int length = fileNames.length; |
| for (int i = 0; i < length; i++) { |
| File file = new File(directory, fileNames[i]); |
| if (file.exists()) { |
| file.delete(); |
| } |
| } |
| } |
| } |
| |
| public void testResetHeartbeats() { |
| EasSyncService svc = new EasSyncService(); |
| // Test case in which the minimum and force heartbeats need to come up |
| svc.mPingMaxHeartbeat = 1000; |
| svc.mPingMinHeartbeat = 200; |
| svc.mPingHeartbeat = 300; |
| svc.mPingForceHeartbeat = 100; |
| svc.mPingHeartbeatDropped = true; |
| svc.resetHeartbeats(400); |
| assertEquals(400, svc.mPingMinHeartbeat); |
| assertEquals(1000, svc.mPingMaxHeartbeat); |
| assertEquals(400, svc.mPingHeartbeat); |
| assertEquals(400, svc.mPingForceHeartbeat); |
| assertFalse(svc.mPingHeartbeatDropped); |
| |
| // Test case in which the force heartbeat needs to come up |
| svc.mPingMaxHeartbeat = 1000; |
| svc.mPingMinHeartbeat = 200; |
| svc.mPingHeartbeat = 100; |
| svc.mPingForceHeartbeat = 100; |
| svc.mPingHeartbeatDropped = true; |
| svc.resetHeartbeats(150); |
| assertEquals(200, svc.mPingMinHeartbeat); |
| assertEquals(1000, svc.mPingMaxHeartbeat); |
| assertEquals(150, svc.mPingHeartbeat); |
| assertEquals(150, svc.mPingForceHeartbeat); |
| assertFalse(svc.mPingHeartbeatDropped); |
| |
| // Test case in which the maximum needs to come down |
| svc.mPingMaxHeartbeat = 1000; |
| svc.mPingMinHeartbeat = 200; |
| svc.mPingHeartbeat = 800; |
| svc.mPingForceHeartbeat = 100; |
| svc.mPingHeartbeatDropped = true; |
| svc.resetHeartbeats(600); |
| assertEquals(200, svc.mPingMinHeartbeat); |
| assertEquals(600, svc.mPingMaxHeartbeat); |
| assertEquals(600, svc.mPingHeartbeat); |
| assertEquals(100, svc.mPingForceHeartbeat); |
| assertFalse(svc.mPingHeartbeatDropped); |
| } |
| |
| public void testAddHeaders() { |
| HttpRequestBase method = new HttpPost(); |
| EasSyncService svc = new EasSyncService(); |
| svc.mAuthString = "auth"; |
| svc.mProtocolVersion = "12.1"; |
| svc.mDeviceType = "android"; |
| svc.mAccount = null; |
| // With second argument false, there should be no header |
| svc.setHeaders(method, false); |
| Header[] headers = method.getHeaders("X-MS-PolicyKey"); |
| assertEquals(0, headers.length); |
| // With second argument true, there should always be a header |
| // The value will be "0" without an account |
| method.removeHeaders("X-MS-PolicyKey"); |
| svc.setHeaders(method, true); |
| headers = method.getHeaders("X-MS-PolicyKey"); |
| assertEquals(1, headers.length); |
| assertEquals("0", headers[0].getValue()); |
| // With an account, but null security key, the header's value should be "0" |
| Account account = new Account(); |
| account.mSecuritySyncKey = null; |
| svc.mAccount = account; |
| method.removeHeaders("X-MS-PolicyKey"); |
| svc.setHeaders(method, true); |
| headers = method.getHeaders("X-MS-PolicyKey"); |
| assertEquals(1, headers.length); |
| assertEquals("0", headers[0].getValue()); |
| // With an account and security key, the header's value should be the security key |
| account.mSecuritySyncKey = "key"; |
| svc.mAccount = account; |
| method.removeHeaders("X-MS-PolicyKey"); |
| svc.setHeaders(method, true); |
| headers = method.getHeaders("X-MS-PolicyKey"); |
| assertEquals(1, headers.length); |
| assertEquals("key", headers[0].getValue()); |
| } |
| } |