summaryrefslogtreecommitdiff
path: root/src/test/java/com/amazon/carbonado/stored/StorableTestBasic.java
blob: 17ed2cf51577839c8bc71c13d2179e74d0e1113a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
/*
 * Copyright 2006-2012 Amazon Technologies, Inc. or its affiliates.
 * Amazon, Amazon.com and Carbonado are trademarks or registered trademarks
 * of Amazon Technologies, Inc. or its affiliates.  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.amazon.carbonado.stored;

import java.util.Random;

import org.joda.time.DateTime;

import com.amazon.carbonado.Nullable;
import com.amazon.carbonado.Storable;
import com.amazon.carbonado.PrimaryKey;
import com.amazon.carbonado.Storage;
import com.amazon.carbonado.Repository;
import com.amazon.carbonado.RepositoryException;

/**
 * StorableTestBasic
 *
 * @author Don Schneider
 */
@PrimaryKey("id")
public abstract class StorableTestBasic implements Storable {
    public abstract int getId();
    public abstract void setId(int id);

    // Basic coverage of the primitives
    public abstract String getStringProp();
    public abstract void setStringProp(String aStringThing);

    public abstract int getIntProp();
    public abstract void setIntProp(int anInt);

    public abstract long getLongProp();
    public abstract void setLongProp(long aLong);

    public abstract double getDoubleProp();
    public abstract void setDoubleProp(double aDouble);

    @Nullable
    public abstract DateTime getDate();
    public abstract void setDate(DateTime aDate);

    public void initPrimaryKeyProperties() {
        setId(10);
    }

    public void initBasicProperties() {
        setStringProp("foo");
        setIntProp(10);
        setLongProp(120);
        setDoubleProp(1.2);
    }

    public void initPropertiesRandomly(int id) {
        setId(id);

        Random random = new Random(1000);

        setIntProp(random.nextInt());
        setLongProp(random.nextLong());
        setDoubleProp(random.nextDouble());
        setStringProp("imaString_" + id % 10);
    }

    public void initPropertiesPredictably(int id) {
        setId(id);

        setIntProp(id*10);
        setLongProp(id*10);
        setDoubleProp(id/2.0);
        setStringProp("string-" + id % 100);
    }

    public static void insertBunches(Repository repository, int count)
            throws RepositoryException
    {
        insertBunches(repository, count, 0, true);
    }


    public static void insertBunches(Repository repository,
                                     int count, int startId,
                                     boolean doRandom)
        throws RepositoryException
    {
        Storage<StorableTestBasic> storage = repository.storageFor(StorableTestBasic.class);
        StorableTestBasic s;

        for (int i = 0; i < count; i ++) {
            s = storage.prepare();
            if (doRandom) {
                s.initPropertiesRandomly(i);
            } else {
                s.initPropertiesPredictably(i+startId);
            }
            s.insert();
        }
    }
}