Newer
Older
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
package com.ic.er;
import com.ic.er.bean.entity.AttributeDO;
import com.ic.er.common.DataType;
import org.apache.log4j.LogManager;
import org.junit.After;
import org.junit.Assert;
import org.junit.Before;
import org.junit.Test;
import java.sql.SQLException;
import java.util.Date;
import java.util.List;
public class TestAttribute {
private View testView;
private Entity testEntity;
@Before
public void init() {
try {
ER.connectDB();
ER.createTables();
} catch (SQLException e) {
System.out.println(e);
throw new RuntimeException(e);
}
testView = ER.createView("testView", "wt22");
testEntity = testView.addEntity("teacher");
}
@Test
public void addAttributeTest() {
Attribute a1 = testEntity.addAttribute("teacher_id", DataType.VARCHAR, 1, 0);
Attribute a2 = testEntity.addAttribute("name", DataType.VARCHAR, 0, 0);
Attribute a3 = testEntity.addAttribute("age", DataType.INTEGER, 0, 0);
System.out.printf("a1 ID: %d\n", a1.getID());
System.out.printf("a2 ID: %d\n", a2.getID());
System.out.printf("a3 ID: %d\n", a3.getID());
}
@Test
public void updateTest() {
Attribute a1 = testEntity.addAttribute("teacher_id", DataType.VARCHAR, 1, 0);
String newName = "new_teacher_id";
a1.setName(newName);
a1.updateDB();
List<Attribute> attributeList = Attribute.queryByAttribute(new AttributeDO(a1.getID()));
Assert.assertEquals(attributeList.size(), 1);
Assert.assertEquals(attributeList.get(0).getName(), newName);
}
@Test
public void selectByIdTest(){
Attribute a1 = testEntity.addAttribute("teacher_id", DataType.VARCHAR, 1, 0);
AttributeDO aDo = ER.attributeMapper.selectById(a1.getID());
Assert.assertNotNull(aDo);
}
@Test
public void deleteByIdTest(){
Attribute a1 = testEntity.addAttribute("teacher_id", DataType.VARCHAR, 1, 0);
// delete
a1.deleteDB();
Attribute attribute = Attribute.queryByID(a1.getID());
Assert.assertNull(attribute);
}
}